Batyka/soundcloak
1
0
Fork 0
mirror of https://git.maid.zone/stuff/soundcloak.git synced 2025-12-10 05:39:38 +05:00
Code Issues Packages Projects Releases Wiki Activity

greeting message and some improvements

Browse Source
This commit is contained in:
Laptop
2025-01-02 21:29:18 +02:00
parent c71b890143
commit 7796f81da6
7 changed files with 100 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
lib/misc/init.go
View File

@@ -44,3 +44,40 @@ func Log(what ...any) {
fmt.Println(what...)
}
}
var ImageClient *fasthttp.HostClient
var HlsClient *fasthttp.HostClient
var HlsAacClient *fasthttp.HostClient
func init() {
if cfg.Restream || cfg.ProxyImages {
ImageClient = &fasthttp.HostClient{
Addr: cfg.ImageCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
}
if cfg.Restream || cfg.ProxyStreams {
HlsClient = &fasthttp.HostClient{
Addr: cfg.HLSCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
HlsAacClient = &fasthttp.HostClient{
Addr: cfg.HLSAACCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
}
}

11
lib/proxy_images/init.go
View File

@@ -10,18 +10,9 @@ import (
"github.com/valyala/fasthttp"
)
var httpc *fasthttp.HostClient
var al_httpc *fasthttp.HostClient
func Load(r *fiber.App) {
httpc = &fasthttp.HostClient{
Addr: cfg.ImageCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
al_httpc = &fasthttp.HostClient{
Addr: "al.sndcdn.com:443",
@@ -53,7 +44,7 @@ func Load(r *fiber.App) {
var cl *fasthttp.HostClient
if parsed.Host()[0] == 'i' {
parsed.SetHost(cfg.ImageCDN)
cl = httpc
cl = misc.ImageClient
} else if string(parsed.Host()[:2]) == "al" {
cl = al_httpc
}

29
lib/proxy_streams/init.go
View File

@@ -11,28 +11,7 @@ import (
"github.com/valyala/fasthttp"
)
var httpc *fasthttp.HostClient
var httpc_aac *fasthttp.HostClient
func Load(r *fiber.App) {
httpc = &fasthttp.HostClient{
Addr: cfg.HLSCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
httpc_aac = &fasthttp.HostClient{
Addr: cfg.HLSAACCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
r.Get("/_/proxy/streams", func(c fiber.Ctx) error {
ur := c.Query("url")
if ur == "" {
@@ -61,7 +40,7 @@ func Load(r *fiber.App) {
resp := fasthttp.AcquireResponse()
//defer fasthttp.ReleaseResponse(resp)
err = sc.DoWithRetry(httpc, req, resp)
err = sc.DoWithRetry(misc.HlsClient, req, resp)
if err != nil {
return err
}
@@ -99,7 +78,7 @@ func Load(r *fiber.App) {
resp := fasthttp.AcquireResponse()
err = sc.DoWithRetry(httpc_aac, req, resp)
err = sc.DoWithRetry(misc.HlsAacClient, req, resp)
if err != nil {
return err
}
@@ -138,7 +117,7 @@ func Load(r *fiber.App) {
resp := fasthttp.AcquireResponse()
defer fasthttp.ReleaseResponse(resp)
err = sc.DoWithRetry(httpc, req, resp)
err = sc.DoWithRetry(misc.HlsClient, req, resp)
if err != nil {
return err
}
@@ -189,7 +168,7 @@ func Load(r *fiber.App) {
resp := fasthttp.AcquireResponse()
defer fasthttp.ReleaseResponse(resp)
err = sc.DoWithRetry(httpc_aac, req, resp)
err = sc.DoWithRetry(misc.HlsAacClient, req, resp)
if err != nil {
return err
}

45
lib/restream/init.go
View File

@@ -17,10 +17,6 @@ import (
"github.com/valyala/fasthttp"
)
var httpc *fasthttp.HostClient
var httpc_aac *fasthttp.HostClient
var httpc_image *fasthttp.HostClient
const defaultPartsCapacity = 24
type reader struct {
@@ -58,9 +54,9 @@ func (r *reader) Setup(url string, aac bool) error {
r.req.Header.Set("Accept-Encoding", "gzip, deflate, br, zstd")
if aac {
r.client = httpc_aac
r.client = misc.HlsAacClient
} else {
r.client = httpc
r.client = misc.HlsClient
}
err := sc.DoWithRetry(r.client, r.req, r.resp)
@@ -195,31 +191,6 @@ func (c *collector) Write(data []byte) (n int, err error) {
}
func Load(r *fiber.App) {
httpc = &fasthttp.HostClient{
Addr: cfg.HLSCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
}
httpc_aac = &fasthttp.HostClient{
Addr: cfg.HLSAACCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
}
httpc_image = &fasthttp.HostClient{
Addr: cfg.ImageCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
r.Get("/_/restream/:author/:track", func(c fiber.Ctx) error {
p, err := preferences.Get(c)
if err != nil {
@@ -303,7 +274,7 @@ func Load(r *fiber.App) {
resp := fasthttp.AcquireResponse()
defer fasthttp.ReleaseResponse(resp)
err = sc.DoWithRetry(httpc, req, resp)
err = sc.DoWithRetry(misc.HlsClient, req, resp)
if err != nil {
return err
}
@@ -327,7 +298,7 @@ func Load(r *fiber.App) {
for _, part := range parts {
req.SetRequestURIBytes(part)
err = sc.DoWithRetry(httpc, req, resp)
err = sc.DoWithRetry(misc.HlsClient, req, resp)
if err != nil {
return err
}
@@ -356,7 +327,7 @@ func Load(r *fiber.App) {
req.SetRequestURI(t.Artwork)
req.Header.Del("Accept-Encoding")
err := sc.DoWithRetry(httpc_image, req, resp)
err := sc.DoWithRetry(misc.ImageClient, req, resp)
if err != nil {
return err
}
@@ -382,7 +353,7 @@ func Load(r *fiber.App) {
resp := fasthttp.AcquireResponse()
defer fasthttp.ReleaseResponse(resp)
err = sc.DoWithRetry(httpc_aac, req, resp)
err = sc.DoWithRetry(misc.HlsAacClient, req, resp)
if err != nil {
return err
}
@@ -413,7 +384,7 @@ func Load(r *fiber.App) {
for _, part := range parts {
req.SetRequestURIBytes(part)
err = sc.DoWithRetry(httpc_aac, req, resp)
err = sc.DoWithRetry(misc.HlsAacClient, req, resp)
if err != nil {
return err
}
@@ -442,7 +413,7 @@ func Load(r *fiber.App) {
req.SetRequestURI(t.Artwork)
req.Header.Del("Accept-Encoding")
err := sc.DoWithRetry(httpc_image, req, resp)
err := sc.DoWithRetry(misc.ImageClient, req, resp)
if err != nil {
return err
}

9
lib/sc/init.go
View File

@@ -34,15 +34,6 @@ var httpc = &fasthttp.HostClient{
MaxIdleConnDuration: 1<<63 - 1,
}
var httpc_image = &fasthttp.HostClient{
Addr: cfg.ImageCDN + ":443",
IsTLS: true,
DialDualStack: true,
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
MaxIdleConnDuration: 1<<63 - 1,
StreamResponseBody: true,
}
var genericClient = &fasthttp.Client{
Dial: (&fasthttp.TCPDialer{DNSCacheDuration: cfg.DNSCacheTTL}).Dial,
}

2
lib/sc/track.go
View File

@@ -452,7 +452,7 @@ func (t Track) DownloadImage() ([]byte, string, error) {
resp := fasthttp.AcquireResponse()
defer fasthttp.ReleaseResponse(resp)
err := DoWithRetry(httpc_image, req, resp)
err := DoWithRetry(misc.ImageClient, req, resp)
if err != nil {
return nil, "", err
}

50
main.go
View File

@@ -1,6 +1,7 @@
package main
import (
"fmt"
"io/fs"
"log"
"math/rand"
@@ -26,6 +27,13 @@ import (
"git.maid.zone/stuff/soundcloak/templates"
)
func boolean(b bool) string {
if b {
return "Enabled"
}
return "Disabled"
}
type osfs struct{}
func (osfs) Open(name string) (fs.File, error) {
@@ -934,8 +942,48 @@ func main() {
return templates.Base(track.Title+" by "+track.Author.Username, templates.TrackInAlbums(track, p), templates.TrackHeader(prefs, track, false)).Render(c.RequestCtx(), c)
})
// cute
const art = `
⠀⠀⠀⠀⢀⡴⣆⠀⠀⠀⠀⠀⣠⡀⠀⠀⠀⠀⠀⠀⣼⣿⡗⠀⠀⠀⠀
⠀⠀⠀⣠⠟⠀⠘⠷⠶⠶⠶⠾⠉⢳⡄⠀⠀⠀⠀⠀⣧⣿⠀⠀⠀⠀⠀
⠀⠀⣰⠃⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⢻⣤⣤⣤⣤⣤⣿⢿⣄⠀⠀⠀⠀
___ ___ ⠀⠀⡇⠀⢀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⣧⠀⠀⠀⠀⠀⠀⠙⣷⡴⠶⣦
/ __|/ __| ⠀⠀⢱⡀⠀⠉⠉⠀⠀⠀⠀⠛⠃⠀⢠⡟⠂⠀⠀⢀⣀⣠⣤⠿⠞⠛⠋
\__ \ (__ ⣠⠾⠋⠙⣶⣤⣤⣤⣤⣤⣀⣠⣤⣾⣿⠴⠶⠚⠋⠉⠁⠀⠀⠀⠀⠀⠀
|___/\___| ⠛⠒⠛⠉⠉⠀⠀⠀⣴⠟⣣⡴⠛⠋⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀⠀
~~~~~~~~~~~~~~~~~~~~⠛⠛~~~~~~~~~~~~~~~~~~~~~~~~`
fmt.Println(art)
keys := [...]string{"Built from", "ProxyStreams", "ProxyImages", "Restream", "GetWebProfiles", "Listening on"}
// maps in go are unordered..
table := map[string]string{
"Built from": fmt.Sprintf("%s (%s)", cfg.Commit, cfg.CommitURL),
"ProxyStreams": boolean(cfg.ProxyStreams),
"ProxyImages": boolean(cfg.ProxyStreams),
"Restream": boolean(cfg.Restream),
"GetWebProfiles": boolean(cfg.GetWebProfiles),
"Listening on": cfg.Addr,
}
if cfg.Addr[0] == ':' {
table["Listening on"] = "127.0.0.1" + cfg.Addr
}
longest := ""
for key := range table {
if len(key) > len(longest) {
longest = key
}
}
longest += " :: "
for _, key := range keys {
fmt.Print(key)
fmt.Print(strings.Repeat(" ", len(longest)-len(key)-len(" :: ")) + " :: ")
fmt.Println(table[key])
}
fmt.Println("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
if cfg.CodegenConfig {
log.Println("Warning: you have CodegenConfig enabled, but the config was loaded dynamically.")
}
log.Fatal(app.Listen(cfg.Addr, fiber.ListenConfig{EnablePrefork: cfg.Prefork}))
log.Fatal(app.Listen(cfg.Addr, fiber.ListenConfig{EnablePrefork: cfg.Prefork, DisableStartupMessage: true}))
}