1 files changed, 53 insertions, 0 deletions
diff --git a/args/args.go b/args/args.go
new file mode 100644
index 0000000..9176d27
--- /dev/null
+++ b/args/args.go
@@ -0,0 +1,53 @@
+package args
+
+import (
+	"errors"
+	"flag"
+	"os"
+)
+
+type Arguments struct {
+	DatabasePath    string
+	TemplatePath    string
+	StaticPath      string
+	CloudflareToken string
+	CloudflareZone  string
+	Port            int
+	Server          bool
+	Migrate         bool
+}
+
+func GetArgs() (*Arguments, error) {
+	port := flag.Int("port", 8080, "Port to listen on")
+	databasePath := flag.String("database-path", "./hatecomputers.db", "Path to the SQLite database")
+	templatePath := flag.String("template-path", "./templates", "Path to the template directory")
+	staticPath := flag.String("static-path", "./static", "Path to the static directory")
+
+	server := flag.Bool("server", false, "Run the server")
+	migrate := flag.Bool("migrate", false, "Run the migrations")
+
+	flag.Parse()
+
+	cloudflareToken := os.Getenv("CLOUDFLARE_TOKEN")
+	cloudflareZone := os.Getenv("CLOUDFLARE_ZONE")
+
+	if cloudflareToken == "" {
+		return nil, errors.New("please set the CLOUDFLARE_TOKEN environment variable")
+	}
+	if cloudflareZone == "" {
+		return nil, errors.New("please set the CLOUDFLARE_ZONE environment variable")
+	}
+
+	arguments := &Arguments{
+		DatabasePath:    *databasePath,
+		TemplatePath:    *templatePath,
+		StaticPath:      *staticPath,
+		CloudflareToken: cloudflareToken,
+		CloudflareZone:  cloudflareZone,
+		Port:            *port,
+		Server:          *server,
+		Migrate:         *migrate,
+	}
+
+	return arguments, nil
+}