PixivFE/session/aux.go

package session

import (
	"log"
	"net/url"
	"strings"

	config "codeberg.org/vnpower/pixivfe/v2/config"
	"github.com/gofiber/fiber/v2"
)

func GetPixivToken(c *fiber.Ctx) string {
	return GetCookie(c, Cookie_Token)
}

func GetImageProxy(c *fiber.Ctx) url.URL {
	value := GetCookie(c, Cookie_ImageProxy)
	if value == "" {
		// fall through to default case
	} else {
		proxyUrl, err := url.Parse(value)
		if err != nil {
			// fall through to default case
		} else {
			return *proxyUrl
		}
	}
	return config.GlobalServerConfig.ProxyServer
}

func ProxyImageUrl(c *fiber.Ctx, s string) string {
	proxyOrigin := GetImageProxyPrefix(c)
	s = strings.ReplaceAll(s, `https:\/\/i.pximg.net`, proxyOrigin)
	// s = strings.ReplaceAll(s, `https:\/\/i.pximg.net`, "/proxy/i.pximg.net")
	s = strings.ReplaceAll(s, `https:\/\/s.pximg.net`, "/proxy/s.pximg.net")
	return s
}

func ProxyImageUrlNoEscape(c *fiber.Ctx, s string) string {
	proxyOrigin := GetImageProxyPrefix(c)
	s = strings.ReplaceAll(s, `https://i.pximg.net`, proxyOrigin)
	// s = strings.ReplaceAll(s, `https:\/\/i.pximg.net`, "/proxy/i.pximg.net")
	s = strings.ReplaceAll(s, `https://s.pximg.net`, "/proxy/s.pximg.net")
	return s
}

func GetImageProxyOrigin(c *fiber.Ctx) string {
	url := GetImageProxy(c)
	return urlAuthority(url)
}

func GetImageProxyPrefix(c *fiber.Ctx) string {
	url := GetImageProxy(c)
	return urlAuthority(url) + url.Path
	// note: not sure if url.EscapedPath() is useful here. go's standard library is trash at handling URL (:// should be part of the scheme)
}

// note: still cannot believe Go doesn't have this function built-in
func urlAuthority(url url.URL) string {
	r := ""
	if (url.Scheme != "") != (url.Host != "") {
		log.Panicf("url must have both scheme and authority or neither: %s", url.String())
	}
	if url.Scheme != "" {
		r += url.Scheme + "://"
	}
	r += url.Host
	return r
}