// 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 }