Add browser

main.go

@@ -3,9 +3,13 @@ package main
 import (
 	"fmt"
 	"github.com/bigkevmcd/go-configparser"
+	"github.com/siteworxpro/img-proxy-url-generator/aws"
 	"github.com/siteworxpro/img-proxy-url-generator/generator"
 	"github.com/siteworxpro/img-proxy-url-generator/printer"
 	"github.com/urfave/cli/v2"
+	"html/template"
+	"log"
+	"net/http"
 	"os"
 )
 
@@ -15,23 +19,19 @@ var imgGenerator *generator.Generator
 
 var Version = "v0.0.0"
 
+var awsConfig aws.Config
+
 func main() {
 
 	pr := printer.NewPrinter()
 
-	app := &cli.App{
-		Name:           "img-proxy-url-generator",
-		Usage:          "URL Generator for the img proxy service",
-		DefaultCommand: "generate",
-		Version:        Version,
-		Commands: []*cli.Command{
-			{
-				Name:  "generate",
-				Usage: "Generate an image from a URL",
-				Action: func(c *cli.Context) error {
-					return run(c, pr)
-				},
-			},
+	var commands []*cli.Command
+
+	commands = append(commands, &cli.Command{
+		Name:  "generate",
+		Usage: "Generate an image from a URL",
+		Action: func(c *cli.Context) error {
+			return run(c, pr)
 		},
 		Flags: []cli.Flag{
 			&cli.StringFlag{
@@ -39,12 +39,6 @@ func main() {
 				Aliases:  []string{"i"},
 				Required: true,
 			},
-			&cli.StringFlag{
-				Name:        "config",
-				Aliases:     []string{"c"},
-				Usage:       "Config file to load from",
-				DefaultText: "imgproxy.cfg",
-			},
 			&cli.StringFlag{
 				Name:    "format",
 				Aliases: []string{"f"},
@@ -56,12 +50,39 @@ func main() {
 				Usage:   "Processing options to be passed to the generator ref: https://docs.imgproxy.net/usage/processing",
 			},
 		},
+	})
+
+	_, err := os.Stat("./templates")
+	if !os.IsNotExist(err) {
+		commands = append(commands, &cli.Command{
+			Name:  "server",
+			Usage: "Start a webserver for s3 file browsing",
+			Action: func(c *cli.Context) error {
+				return startServer(c, pr)
+			},
+		})
+	}
+
+	app := &cli.App{
+		Name:           "img-proxy-url-generator",
+		Usage:          "URL Generator for the img proxy service",
+		DefaultCommand: "generate",
+		Version:        Version,
+		Commands:       commands,
 		Action: func(c *cli.Context) error {
 			return run(c, pr)
 		},
+		Flags: []cli.Flag{
+			&cli.StringFlag{
+				Name:        "config",
+				Aliases:     []string{"c"},
+				Usage:       "Config file to load from",
+				DefaultText: "imgproxy.cfg",
+			},
+		},
 	}
 
-	err := app.Run(os.Args)
+	err = app.Run(os.Args)
 	if err != nil {
 		pr.LogError(err.Error())
 
@@ -69,6 +90,51 @@ func main() {
 	}
 }
 
+func startServer(c *cli.Context, p *printer.Printer) error {
+	err := initGenerator(c.String("config"))
+	if err != nil {
+		return err
+	}
+
+	awsClient := aws.NewClient(&awsConfig)
+
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		contToken := r.URL.Query().Get("next")
+
+		var next *string
+		if contToken == "" {
+			next = nil
+		} else {
+			next = &contToken
+		}
+
+		contents, err := awsClient.ListBucketContents(next)
+		if err != nil {
+			return
+		}
+
+		for i, content := range contents.Images {
+			contents.Images[i].Url, _ = signURL("s3://"+awsConfig.Bucket+"/"+content.Name, []string{"pr:sq"}, "")
+			contents.Images[i].Download, _ = signURL("s3://"+awsConfig.Bucket+"/"+content.Name, []string{""}, "")
+		}
+
+		file, _ := os.ReadFile("./templates/index.gohtml")
+
+		tmpl := template.Must(template.New("index").Parse(string(file)))
+
+		err = tmpl.Execute(w, contents)
+
+		if err != nil {
+			println(err.Error())
+		}
+	})
+
+	p.LogSuccess("Starting http server on port 8080. http://localhost:8080")
+	log.Fatal(http.ListenAndServe(":8080", nil))
+
+	return nil
+}
+
 func run(c *cli.Context, p *printer.Printer) error {
 	err := initGenerator(c.String("config"))
 
@@ -135,6 +201,13 @@ func initGenerator(config string) error {
 		return err
 	}
 
+	if p.HasSection("aws") {
+		awsConfig.AwsSecret, _ = p.Get("aws", "secret")
+		awsConfig.AwsKey, _ = p.Get("aws", "key")
+		awsConfig.AwsRole, _ = p.Get("aws", "role")
+		awsConfig.Bucket, _ = p.Get("aws", "bucket")
+	}
+
 	return nil
 }