first commit

main.go

@@ -0,0 +1,150 @@
+/*
+ * LURE Updater - Automated updater bot for LURE packages
+ * Copyright (C) 2023 Elara Musayelyan
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package main
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/caarlos0/env/v8"
+	"github.com/go-git/go-git/v5"
+	"github.com/pelletier/go-toml/v2"
+	"github.com/spf13/pflag"
+	"go.elara.ws/logger"
+	"go.elara.ws/logger/log"
+	"lure.sh/lure-updater/internal/builtins"
+	"lure.sh/lure-updater/internal/config"
+	"go.etcd.io/bbolt"
+	"go.starlark.net/starlark"
+	"golang.org/x/crypto/bcrypt"
+	"golang.org/x/term"
+)
+
+func init() {
+	log.Logger = logger.NewPretty(os.Stderr)
+}
+
+func main() {
+	configPath := pflag.StringP("config", "c", "/etc/lure-updater/config.toml", "Path to config file")
+	dbPath := pflag.StringP("database", "d", "/etc/lure-updater/db", "Path to database file")
+	pluginDir := pflag.StringP("plugin-dir", "p", "/etc/lure-updater/plugins", "Path to plugin directory")
+	serverAddr := pflag.StringP("address", "a", ":8080", "Webhook server address")
+	genHash := pflag.BoolP("gen-hash", "g", false, "Generate a password hash for webhooks")
+	useEnv := pflag.BoolP("use-env", "E", false, "Use environment variables for configuration")
+	debug := pflag.BoolP("debug", "D", false, "Enable debug logging")
+	pflag.Parse()
+
+	if *debug {
+		log.Logger.SetLevel(logger.LogLevelDebug)
+	}
+
+	if *genHash {
+		fmt.Print("Password: ")
+		pwd, err := term.ReadPassword(int(os.Stdin.Fd()))
+		if err != nil {
+			log.Fatal("Error reading password").Err(err).Send()
+		}
+		hash, err := bcrypt.GenerateFromPassword(pwd, bcrypt.DefaultCost)
+		if err != nil {
+			log.Fatal("Error hashing password").Err(err).Send()
+		}
+		fmt.Printf("\n%s\n", hash)
+		return
+	}
+
+	db, err := bbolt.Open(*dbPath, 0o644, nil)
+	if err != nil {
+		log.Fatal("Error opening database").Err(err).Send()
+	}
+
+	cfg := &config.Config{}
+	if *useEnv {
+		err = env.Parse(cfg)
+		if err != nil {
+			log.Fatal("Error parsing environment variables").Err(err).Send()
+		}
+	} else {
+		fl, err := os.Open(*configPath)
+		if err != nil {
+			log.Fatal("Error opening config file").Err(err).Send()
+		}
+		err = toml.NewDecoder(fl).Decode(cfg)
+		if err != nil {
+			log.Fatal("Error decoding config file").Err(err).Send()
+		}
+		err = fl.Close()
+		if err != nil {
+			log.Fatal("Error closing config file").Err(err).Send()
+		}
+	}
+
+	if _, err := os.Stat(cfg.Git.RepoDir); os.IsNotExist(err) {
+		err = os.MkdirAll(cfg.Git.RepoDir, 0o755)
+		if err != nil {
+			log.Fatal("Error creating repository directory").Err(err).Send()
+		}
+
+		_, err := git.PlainClone(cfg.Git.RepoDir, false, &git.CloneOptions{
+			URL:      cfg.Git.RepoURL,
+			Progress: os.Stderr,
+		})
+		if err != nil {
+			log.Fatal("Error cloning repository").Err(err).Send()
+		}
+	} else if err != nil {
+		log.Fatal("Cannot stat configured repo directory").Err(err).Send()
+	}
+
+	starFiles, err := filepath.Glob(filepath.Join(*pluginDir, "*.star"))
+	if err != nil {
+		log.Fatal("Error finding plugin files").Err(err).Send()
+	}
+
+	if len(starFiles) == 0 {
+		log.Fatal("No plugins found. At least one plugin is required.").Send()
+	}
+
+	mux := http.NewServeMux()
+
+	for _, starFile := range starFiles {
+		pluginName := filepath.Base(strings.TrimSuffix(starFile, ".star"))
+		thread := &starlark.Thread{Name: pluginName}
+
+		predeclared := starlark.StringDict{}
+		builtins.Register(predeclared, &builtins.Options{
+			Name:   pluginName,
+			Config: cfg,
+			DB:     db,
+			Mux:    mux,
+		})
+
+		_, err = starlark.ExecFile(thread, starFile, nil, predeclared)
+		if err != nil {
+			log.Fatal("Error executing starlark file").Str("file", starFile).Err(err).Send()
+		}
+
+		log.Info("Initialized plugin").Str("name", pluginName).Send()
+	}
+
+	log.Info("Starting HTTP server").Str("addr", *serverAddr).Send()
+	http.ListenAndServe(*serverAddr, mux)
+}