ipng/s3-genindex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add logo feature

Browse Source
This commit is contained in:
Pim van Pelt
2025-12-03 12:32:36 +01:00
parent 60a149b669
commit 99ad6fbff1
3 changed files with 119 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/s3-genindex/main.go
View File

@@ -284,12 +284,14 @@ func generateS3HTML(entries []indexgen.FileEntry, opts *indexgen.Options, client
DirAppend bool
OutputFile string
IsRoot bool
WatermarkURL string
}{
DirName: opts.TopDir, // Use bucket name as directory name
Entries: entries,
DirAppend: opts.DirAppend,
OutputFile: opts.OutputFile,
IsRoot: isRoot,
WatermarkURL: opts.WatermarkURL,
}
// Generate HTML content in memory
@@ -345,6 +347,7 @@ func main() {
var s3URL string
var dryRun bool
var showIndexFiles bool
var watermarkURL string
// Set defaults
opts.DirAppend = true
@@ -359,6 +362,7 @@ func main() {
flag.StringVar(&excludeRegexStr, "x", "", "exclude files matching regular expression")
flag.BoolVar(&opts.Verbose, "v", false, "verbosely list every processed file")
flag.BoolVar(&showIndexFiles, "i", false, "show index.html files in directory listings")
flag.StringVar(&watermarkURL, "wm", "", "watermark logo URL to display in top left corner")
flag.Usage = func() {
fmt.Fprintf(os.Stderr, "Generate directory index files (recursive is ON, hidden files included by default).\n")
@@ -398,9 +402,10 @@ func main() {
}
}
// Set dry run and show index files flags
// Set dry run, show index files, and watermark URL
opts.DryRun = dryRun
opts.ShowIndexFiles = showIndexFiles
opts.WatermarkURL = watermarkURL
if s3URL != "" {
// Parse S3 URL

14
internal/indexgen/indexgen.go
View File

@@ -168,6 +168,7 @@ type Options struct {
Verbose bool
DryRun bool
ShowIndexFiles bool
WatermarkURL string
}
type FileEntry struct {
@@ -216,12 +217,14 @@ func ProcessDir(topDir string, opts *Options) error {
DirAppend bool
OutputFile string
IsRoot bool
WatermarkURL string
}{
DirName: dirName,
Entries: entries,
DirAppend: opts.DirAppend,
OutputFile: opts.OutputFile,
IsRoot: false, // Local filesystem always shows parent directory
WatermarkURL: opts.WatermarkURL,
}
if opts.DryRun {
@@ -467,6 +470,15 @@ const htmlTemplateString = `<!DOCTYPE html>
overflow-x: hidden;
text-overflow: ellipsis;
color: #999;
display: flex;
align-items: center;
}
.watermark {
height: 24px;
width: auto;
margin-right: 8px;
vertical-align: middle;
}
h1 a {
@@ -943,7 +955,7 @@ const htmlTemplateString = `<!DOCTYPE html>
</defs>
</svg>
<header>
<h1>{{.DirName}}</h1>
<h1>{{if .WatermarkURL}}<img src="{{.WatermarkURL}}" class="watermark" alt="Logo">{{end}}{{.DirName}}</h1>
</header>
<main>
<div class="listing">

60
internal/indexgen/indexgen_test.go
View File

@@ -102,12 +102,14 @@ func TestHTMLTemplate(t *testing.T) {
DirAppend bool
OutputFile string
IsRoot bool
WatermarkURL string
}{
DirName: "test-dir",
Entries: []FileEntry{},
DirAppend: false,
OutputFile: "index.html",
IsRoot: false,
WatermarkURL: "",
}
var buf bytes.Buffer
@@ -164,12 +166,14 @@ func TestHTMLTemplateWithEntries(t *testing.T) {
DirAppend bool
OutputFile string
IsRoot bool
WatermarkURL string
}{
DirName: "test-dir",
Entries: entries,
DirAppend: false,
OutputFile: "index.html",
IsRoot: false,
WatermarkURL: "",
}
var buf bytes.Buffer
@@ -196,6 +200,62 @@ func TestHTMLTemplateWithEntries(t *testing.T) {
}
}
func TestHTMLTemplateWithWatermark(t *testing.T) {
tmpl := GetHTMLTemplate()
if tmpl == nil {
t.Fatal("GetHTMLTemplate() returned nil")
}
// Test template execution with watermark
data := struct {
DirName string
Entries []FileEntry
DirAppend bool
OutputFile string
IsRoot bool
WatermarkURL string
}{
DirName: "test-dir",
Entries: []FileEntry{},
DirAppend: false,
OutputFile: "index.html",
IsRoot: false,
WatermarkURL: "https://example.com/logo.svg",
}
var buf bytes.Buffer
err := tmpl.Execute(&buf, data)
if err != nil {
t.Fatalf("Template execution with watermark failed: %v", err)
}
output := buf.String()
// Check that watermark image is included
if !bytes.Contains([]byte(output), []byte(`src="https://example.com/logo.svg"`)) {
t.Error("Template output should contain watermark image URL")
}
if !bytes.Contains([]byte(output), []byte(`class="watermark"`)) {
t.Error("Template output should contain watermark CSS class")
}
// Test without watermark
data.WatermarkURL = ""
buf.Reset()
err = tmpl.Execute(&buf, data)
if err != nil {
t.Fatalf("Template execution without watermark failed: %v", err)
}
outputNoWatermark := buf.String()
// Check that watermark image is NOT included when URL is empty
if bytes.Contains([]byte(outputNoWatermark), []byte(`class="watermark"`)) {
t.Error("Template output should not contain watermark when URL is empty")
}
}
func TestReadDirEntries(t *testing.T) {
// Create a temporary directory with test files
tempDir := t.TempDir()