Check if in a browser instead of if in OBS specifically to support other streaming software

Author: jaredcat

.gitignore

@@ -23,6 +23,7 @@ desktop.ini
 # Build and distribution files
 dist/
 build/
+test/
 
 # Logs
 *.log

internal/config/config.go

@@ -0,0 +1,64 @@
+// Package config provides configuration management for the OBS Random Videos application.
+// It handles parsing command-line flags and maintaining application settings.
+package config
+
+import (
+	"flag"
+	"os"
+)
+
+// Config holds the application configuration
+type Config struct {
+	// OutputFileName is the name of the generated HTML file
+	OutputFileName string
+	// Version is the application version
+	Version string
+	// MaxFileSize is the maximum file size to consider (in bytes, 0 = no limit)
+	MaxFileSize int64
+	// ConcurrentScans is the number of concurrent goroutines for directory scanning
+	ConcurrentScans int
+	// Directory is the directory to scan for media files
+	Directory string
+	// Recursive determines if subdirectories should be scanned
+	Recursive bool
+	// Verbose enables verbose output
+	Verbose bool
+	// AutoSanitize determines if file names should be automatically sanitized
+	AutoSanitize bool
+}
+
+// Default returns a Config with default values
+func Default(version string) *Config {
+	return &Config{
+		OutputFileName:  "obs-random-videos.html",
+		Version:         version,
+		MaxFileSize:     0, // No limit by default
+		ConcurrentScans: 4,
+		Directory:       ".",
+		Recursive:       true,
+		Verbose:         false,
+		AutoSanitize:    false,
+	}
+}
+
+// ParseFlags parses command line flags and returns a Config
+func ParseFlags(version string) *Config {
+	cfg := Default(version)
+
+	flag.StringVar(&cfg.Directory, "dir", cfg.Directory, "Directory to scan for media files")
+	flag.StringVar(&cfg.OutputFileName, "output", cfg.OutputFileName, "Output HTML filename")
+	flag.BoolVar(&cfg.Recursive, "recursive", cfg.Recursive, "Scan subdirectories")
+	flag.BoolVar(&cfg.Verbose, "verbose", cfg.Verbose, "Verbose output")
+	flag.BoolVar(&cfg.AutoSanitize, "sanitize", cfg.AutoSanitize, "Automatically sanitize problematic file names")
+	flag.Int64Var(&cfg.MaxFileSize, "max-size", cfg.MaxFileSize, "Maximum file size in bytes (0 = no limit)")
+	flag.IntVar(&cfg.ConcurrentScans, "concurrent", cfg.ConcurrentScans, "Number of concurrent file scanners")
+
+	// Check for DEBUG environment variable
+	if os.Getenv("DEBUG") != "" {
+		cfg.Verbose = true
+	}
+
+	flag.Parse()
+
+	return cfg
+}

internal/config/config_test.go

@@ -0,0 +1,59 @@
+package config
+
+import (
+	"testing"
+)
+
+func TestDefault(t *testing.T) {
+	version := "1.0.0"
+	cfg := Default(version)
+
+	if cfg.Version != version {
+		t.Errorf("Version = %v, want %v", cfg.Version, version)
+	}
+
+	if cfg.OutputFileName != "obs-random-videos.html" {
+		t.Errorf("OutputFileName = %v, want 'obs-random-videos.html'", cfg.OutputFileName)
+	}
+
+	if cfg.ConcurrentScans != 4 {
+		t.Errorf("ConcurrentScans = %v, want 4", cfg.ConcurrentScans)
+	}
+
+	if cfg.Recursive != true {
+		t.Errorf("Recursive = %v, want true", cfg.Recursive)
+	}
+
+	if cfg.Verbose != false {
+		t.Errorf("Verbose = %v, want false", cfg.Verbose)
+	}
+
+	if cfg.AutoSanitize != false {
+		t.Errorf("AutoSanitize = %v, want false", cfg.AutoSanitize)
+	}
+}
+
+func TestConfigFields(t *testing.T) {
+	cfg := &Config{
+		OutputFileName:  "test.html",
+		Version:         "2.0.0",
+		MaxFileSize:     1024,
+		ConcurrentScans: 8,
+		Directory:       "/test/dir",
+		Recursive:       false,
+		Verbose:         true,
+		AutoSanitize:    true,
+	}
+
+	if cfg.OutputFileName != "test.html" {
+		t.Errorf("OutputFileName = %v, want 'test.html'", cfg.OutputFileName)
+	}
+
+	if cfg.MaxFileSize != 1024 {
+		t.Errorf("MaxFileSize = %v, want 1024", cfg.MaxFileSize)
+	}
+
+	if cfg.ConcurrentScans != 8 {
+		t.Errorf("ConcurrentScans = %v, want 8", cfg.ConcurrentScans)
+	}
+}