Add YTDLP_SITE_SECURE variable for use with https

This commit is contained in:
Carl Pearson
2024-09-16 05:18:19 -06:00
parent 7a6004bd32
commit 673c88cff4
2 changed files with 16 additions and 2 deletions

View File

@@ -3,8 +3,11 @@ package main
import (
"fmt"
"os"
"strings"
)
var GitSHA string
func getDataDir() string {
value, exists := os.LookupEnv("YTDLP_SITE_DATA_DIR")
if exists {
@@ -39,7 +42,16 @@ func getSessionAuthKey() ([]byte, error) {
return []byte{}, fmt.Errorf("please set %s", key)
}
var GitSHA string
func getSecure() bool {
key := "YTDLP_SITE_SECURE"
if value, exists := os.LookupEnv(key); exists {
lower := strings.ToLower(value)
if lower == "on" || lower == "1" || lower == "true" || lower == "yes" {
return true
}
}
return false
}
func getGitSHA() string {

View File

@@ -110,11 +110,13 @@ func main() {
staticGroup.Static("/", getDataDir())
e.GET("/temp/:token", tempHandler)
secure := getSecure()
store.Options = &sessions.Options{
Path: "/",
MaxAge: 30 * 24 * 60 * 60, // seconds
HttpOnly: true,
Secure: false, // needed for session to work over http
Secure: secure,
}
// start the transcode worker