113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
// GoToSocial
|
|
// Copyright (C) GoToSocial Authors admin@gotosocial.org
|
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package util
|
|
|
|
import (
|
|
"net/url"
|
|
"strings"
|
|
|
|
"golang.org/x/net/idna"
|
|
)
|
|
|
|
// Punify converts the given domain to lowercase
|
|
// then to punycode (for international domain names).
|
|
//
|
|
// Returns the resulting domain or an error if the
|
|
// punycode conversion fails.
|
|
func Punify(domain string) (string, error) {
|
|
domain = strings.ToLower(domain)
|
|
return idna.ToASCII(domain)
|
|
}
|
|
|
|
// DePunify converts the given punycode string
|
|
// to its original unicode representation (lowercased).
|
|
// Noop if the domain is (already) not puny.
|
|
//
|
|
// Returns an error if conversion fails.
|
|
func DePunify(domain string) (string, error) {
|
|
out, err := idna.ToUnicode(domain)
|
|
return strings.ToLower(out), err
|
|
}
|
|
|
|
// URIMatches returns true if the expected URI matches
|
|
// any of the given URIs, taking account of punycode.
|
|
func URIMatches(expect *url.URL, uris ...*url.URL) (bool, error) {
|
|
// Normalize expect to punycode.
|
|
expectPuny, err := PunifyURI(expect)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
expectStr := expectPuny.String()
|
|
|
|
for _, uri := range uris {
|
|
uriPuny, err := PunifyURI(uri)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if uriPuny.String() == expectStr {
|
|
// Looks good.
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
// Didn't match.
|
|
return false, nil
|
|
}
|
|
|
|
// PunifyURI returns a copy of the given URI
|
|
// with the 'host' part converted to punycode.
|
|
func PunifyURI(in *url.URL) (*url.URL, error) {
|
|
// Take a copy of in.
|
|
out := new(url.URL)
|
|
*out = *in
|
|
|
|
// Normalize host to punycode.
|
|
var err error
|
|
out.Host, err = Punify(in.Host)
|
|
return out, err
|
|
}
|
|
|
|
// PunifyURIStr returns a copy of the given URI
|
|
// string with the 'host' part converted to punycode.
|
|
func PunifyURIStr(in string) (string, error) {
|
|
inURI, err := url.Parse(in)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
outURIPuny, err := Punify(inURI.Host)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if outURIPuny == in {
|
|
// Punify did nothing, so in was
|
|
// already punified, return as-is.
|
|
return in, nil
|
|
}
|
|
|
|
// Take a copy of in.
|
|
outURI := new(url.URL)
|
|
*outURI = *inURI
|
|
|
|
// Normalize host to punycode.
|
|
outURI.Host = outURIPuny
|
|
return outURI.String(), err
|
|
}
|