main.go

package main

import (
	"net/url"
	"path/filepath"
	"runtime"
	"strings"

	"github.com/samicrusader/andesite-the-eye.eu/pkg/config"
	"github.com/samicrusader/andesite-the-eye.eu/pkg/db"
	"github.com/samicrusader/andesite-the-eye.eu/pkg/fsdb"
	"github.com/samicrusader/andesite-the-eye.eu/pkg/handler"
	"github.com/samicrusader/andesite-the-eye.eu/pkg/idata"

	"github.com/aymerick/raymond"
	"github.com/nektro/go-util/arrays/stringsu"
	"github.com/nektro/go-util/util"
	"github.com/nektro/go-util/vflag"
	etc "github.com/nektro/go.etc"
	"github.com/nektro/go.etc/htp"

	. "github.com/nektro/go-util/alias"
	_ "github.com/samicrusader/andesite-the-eye.eu/statik"
)

// version var overwritten by build_all.sh
var (
	Version = "the-eye.eu-vMASTER"
)

func main() {
	etc.AppID = "andesite"
	etc.Version = Version
	etc.FixBareVersion()
	util.Log("Initializing Andesite " + etc.Version + "...")

	vflag.IntVar(&idata.Config.Version, "version", idata.RequiredConfigVersion, "Config version to use.")
	vflag.StringVar(&idata.Config.Root, "root", "", "Path of root directory for files")
	vflag.StringVar(&idata.Config.Public, "public", "", "Public root of files to serve")
	vflag.StringArrayVar(&idata.Config.SearchOn, "enable-search", []string{}, "Set to a root ID to enable file search for that directory.")
	vflag.StringArrayVar(&idata.Config.SearchOff, "disable-search", []string{}, "Set to a root ID to disable file search for that directory.")
	vflag.BoolVar(&idata.Config.NoAutoScan, "disable-file-scanner", false, "Disables file scanning but keeps search and fsdb enabled, useful if you're using a separate hashing proggy.")
	vflag.IntVar(&idata.Config.SearchResultAmt, "search-result-limit", 25, "limit of results to return in /api/search")
	flagDGS := vflag.String("discord-guild-id", "", "")
	flagDBT := vflag.String("discord-bot-token", "", "")
	vflag.BoolVar(&idata.Config.Verbose, "verbose", false, "")
	vflag.BoolVar(&idata.Config.VerboseFS, "fsdb-verbose", false, "")
	vflag.StringArrayVar(&idata.Config.OffHashes, "disable-hash", []string{}, "disable hashing for dirs (one per dir)")
	vflag.IntVar(&idata.Config.ScanSimul, "scan-concurrency", runtime.NumCPU(), "number of threads to use for fsdb scan queueing")
	vflag.StringArrayVar(&idata.Config.HashExcl, "hashing-exclude", []string{}, "number of threads to use for fsdb queuing")
	vflag.StringArrayVar(&idata.Config.CRootsPub, "custom-root-public", []string{}, "")
	vflag.StringArrayVar(&idata.Config.CRootsPrv, "custom-root-private", []string{}, "")
	vflag.BoolVar(&config.GlobalSearchOff, "disable-global-search", false, "")
	etc.PreInit()

	etc.Init(&idata.Config, "./files/", db.SaveOAuth2InfoCb)

	//

	for i, item := range idata.Config.Clients {
		if item.For == "discord" {
			if len(*flagDGS) > 0 {
				idata.Config.Clients[i].Extra1 = *flagDGS
			}
			if len(*flagDBT) > 0 {
				idata.Config.Clients[i].Extra2 = *flagDBT
			}
		}
	}

	if idata.Config.Version == 0 {
		idata.Config.Version = 1
	}
	if idata.Config.Version != idata.RequiredConfigVersion {
		util.DieOnError(
			E(F("Current idata.Config.json version '%d' does not match required version '%d'.", idata.Config.Version, idata.RequiredConfigVersion)),
			F("Visit https://github.com/nektro/andesite/blob/master/docs/config/v%d.md for more info.", idata.RequiredConfigVersion),
		)
	}

	idata.Config.SearchOn = stringsu.Depupe(idata.Config.SearchOn)
	idata.Config.SearchOff = stringsu.Depupe(idata.Config.SearchOff)

	for _, item := range idata.Config.OffHashes {
		idata.DisableHash(item)
	}

	for _, item := range idata.Config.CRootsPub {
		idata.Config.RootsPub = append(idata.Config.RootsPub, strings.SplitN(item, "=", 2))
	}
	for _, item := range idata.Config.CRootsPrv {
		idata.Config.RootsPrv = append(idata.Config.RootsPrv, strings.SplitN(item, "=", 2))
	}

	//
	// database initialization

	db.Init()

	db.Upgrade()

	//
	// graceful stop

	util.RunOnClose(func() {
		util.Log("Gracefully shutting down...")

		util.Log("Saving database to disk")
		db.Close()

		util.Log("Done!")
	})

	//
	// handlebars helpers

	raymond.RegisterHelper("url_name", func(x string) string {
		return strings.Replace(url.PathEscape(x), "%2F", "/", -1)
	})
	raymond.RegisterHelper("add_i", func(a, b int) int {
		return a + b
	})
	raymond.RegisterHelper("user_name", func(x int64) string {
		u, ok := db.User{}.ByID(x)
		if !ok {
			return ""
		}
		return u.FullName()
	})
	raymond.RegisterHelper("dl_url", func(h, b, p string) string {
		b = strings.TrimSuffix(b, "/")
		return h + b + p
	})
	raymond.RegisterHelper("grab_url", func(h, b, p, k string) string {
		b = strings.TrimSuffix(b, "/")
		if len(k) > 0 {
			return strings.Replace(h, "://", "://"+k+"@", 1) + b + p
		}
		return h + p
	})

	//
	// init roots

	if len(idata.Config.Root) > 0 {
		idata.Config.Root, _ = filepath.Abs(filepath.Clean(strings.ReplaceAll(idata.Config.Root, "~", idata.HomedirPath)))
		util.DieOnError(util.Assert(util.DoesDirectoryExist(idata.Config.Root), "Please pass a valid directory as a root parameter!"))
		idata.DataPathsPrv["files"] = idata.Config.Root
	}
	for _, item := range idata.Config.RootsPrv {
		ab, err := filepath.Abs(item[1])
		util.DieOnError(err)
		idata.DataPathsPrv[item[0]] = ab
	}

	if len(idata.Config.Public) > 0 {
		idata.Config.Public, _ = filepath.Abs(filepath.Clean(strings.ReplaceAll(idata.Config.Public, "~", idata.HomedirPath)))
		util.DieOnError(util.Assert(util.DoesDirectoryExist(idata.Config.Public), "Public root directory does not exist. Aborting!"))
		idata.DataPathsPub["public"] = idata.Config.Public
	}
	for _, item := range idata.Config.RootsPub {
		ab, err := filepath.Abs(item[1])
		util.DieOnError(err)
		idata.DataPathsPub[item[0]] = ab
	}

	//
	// initialize file database in background

	if len(idata.Config.SearchOn) > 0 {
		htp.Register("/search", "GET", handler.HandleSearch)
		htp.Register("/api/search", "GET", handler.HandleSearchAPI)

		for _, item := range idata.Config.SearchOn {
			go fsdb.Init(idata.DataPathsPub, item)
			go fsdb.Init(idata.DataPathsPrv, item)
		}
	}
	if len(idata.Config.SearchOff) > 0 {
		for _, item := range idata.Config.SearchOff {
			fsdb.DeInit(idata.DataPathsPub, item)
			fsdb.DeInit(idata.DataPathsPrv, item)
		}
	}

	//
	// http server setup

	handler.Init()
	idata.Config.HTTPBase = htp.Base()

	if len(idata.DataPathsPrv) > 0 {
		for k, v := range idata.DataPathsPrv {
			htp.Register("/"+k+"/*", "GET", handler.HandleDirectoryListing(handler.HandleFileListing))
			util.Log("Sharing private files as", k, "from", v)
		}

		htp.Register("/regen_passkey", "GET", handler.HandleRegenPasskey)
		htp.Register("/logout", "GET", handler.HandleLogout)
		htp.Register("/open/*", "GET", handler.HandleDirectoryListing(handler.HandleShareListing))

		htp.Register("/admin", "GET", handler.HandleAdmin)
		htp.Register("/admin/users", "GET", handler.HandleAdminUsers)
		htp.Register("/admin/roots", "GET", handler.HandleAdminRoots)

		htp.Register("/api/access/create", "POST", handler.HandleAccessCreate)
		htp.Register("/api/access/update", "POST", handler.HandleAccessUpdate)
		htp.Register("/api/access/delete", "POST", handler.HandleAccessDelete)

		htp.Register("/api/share/create", "POST", handler.HandleShareCreate)
		htp.Register("/api/share/update", "POST", handler.HandleShareUpdate)
		htp.Register("/api/share/delete", "POST", handler.HandleShareDelete)

		htp.Register("/api/access_discord_role/create", "POST", handler.HandleDiscordRoleAccessCreate)
		htp.Register("/api/access_discord_role/update", "POST", handler.HandleDiscordRoleAccessUpdate)
		htp.Register("/api/access_discord_role/delete", "POST", handler.HandleDiscordRoleAccessDelete)
	}

	if len(idata.DataPathsPub) > 0 {
		for k, v := range idata.DataPathsPub {
			htp.Register("/"+k+"/*", "GET", handler.HandleDirectoryListing(handler.HandlePublicListing))
			util.Log("Sharing public files as", k, "from", v)
		}
	}

	//
	// start http server

	etc.StartServer()
}