OAuth continue-URL rewriting shares path/locale utilities with SEO, so keep those in locale.go and leave seo.go for SEOFromRequest only. Signed-off-by: Émile Ré <emile@probo.com>
109 lines
3.2 KiB
Go
109 lines
3.2 KiB
Go
// Copyright (c) 2026 Probo Inc <hello@probo.com>.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
// SOFTWARE.
|
|
|
|
package complianceportal_v1
|
|
|
|
import (
|
|
"net/url"
|
|
"slices"
|
|
"strings"
|
|
|
|
"go.probo.inc/probo/pkg/iam"
|
|
)
|
|
|
|
const defaultCompliancePortalLocale = "en"
|
|
|
|
func isCompliancePortalLocale(value string) bool {
|
|
return slices.Contains(iam.SupportedIdentityLocales, value)
|
|
}
|
|
|
|
func splitLocaleFromAppPath(appPath string) (locale, rest string) {
|
|
segments := strings.Split(strings.Trim(appPath, "/"), "/")
|
|
if len(segments) == 0 || segments[0] == "" {
|
|
return defaultCompliancePortalLocale, "/"
|
|
}
|
|
|
|
if isCompliancePortalLocale(segments[0]) {
|
|
locale = segments[0]
|
|
if len(segments) == 1 {
|
|
return locale, "/"
|
|
}
|
|
|
|
return locale, "/" + strings.Join(segments[1:], "/")
|
|
}
|
|
|
|
// Unprefixed path — treat content path as-is; default lang for tags.
|
|
return defaultCompliancePortalLocale, appPath
|
|
}
|
|
|
|
func localizedPageURL(pageBaseURL, locale, rest string) string {
|
|
base := strings.TrimRight(pageBaseURL, "/")
|
|
segments := []string{locale}
|
|
|
|
if rest != "/" && rest != "" {
|
|
trimmed := strings.Trim(rest, "/")
|
|
if trimmed != "" {
|
|
segments = append(segments, strings.Split(trimmed, "/")...)
|
|
}
|
|
}
|
|
|
|
escaped := make([]string, len(segments))
|
|
for i, segment := range segments {
|
|
escaped[i] = url.PathEscape(segment)
|
|
}
|
|
|
|
joined, err := url.JoinPath(base, escaped...)
|
|
if err != nil {
|
|
return base + "/" + strings.Join(escaped, "/")
|
|
}
|
|
|
|
return joined
|
|
}
|
|
|
|
// rewriteContinueURLLocale swaps the leading locale segment of continueURL's
|
|
// path for locale (a supported short tag). Relative and absolute URLs are
|
|
// accepted; query and fragment are preserved. Unprefixed paths get the locale
|
|
// prepended. Returns continueURL unchanged when locale is unsupported or the
|
|
// URL cannot be parsed.
|
|
func rewriteContinueURLLocale(continueURL, locale string) string {
|
|
if !isCompliancePortalLocale(locale) {
|
|
return continueURL
|
|
}
|
|
|
|
u, err := url.Parse(continueURL)
|
|
if err != nil {
|
|
return continueURL
|
|
}
|
|
|
|
path := u.Path
|
|
if path == "" {
|
|
path = "/"
|
|
}
|
|
|
|
_, rest := splitLocaleFromAppPath(path)
|
|
if rest == "/" || rest == "" {
|
|
u.Path = "/" + locale
|
|
} else {
|
|
u.Path = "/" + locale + rest
|
|
}
|
|
|
|
return u.String()
|
|
}
|