main.go

// Copyright (C) 2020 David Vogel
//
// This file is part of Galago.
//
// Galago 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 2 of the License, or
// (at your option) any later version.
//
// Galago 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 Galago.  If not, see <http://www.gnu.org/licenses/>.

package main

import (
	"fmt"
	"net/http"
	"os"
	"path/filepath"
	"runtime"
	"time"

	_ "image/jpeg"
	_ "image/png"

	_ "golang.org/x/image/bmp"

	"github.com/Dadido3/configdb"
	"github.com/gorilla/mux"
	"github.com/mattn/go-colorable"
	"github.com/sirupsen/logrus"
	"github.com/snowzach/rotatefilehook"
)

var log = logrus.New()
var conf = configdb.NewOrPanic([]configdb.Storage{
	configdb.UseYAMLFile(filepath.Join(".", "config", "config.yaml")),
})
var router = mux.NewRouter()
var validExtensions = map[string]bool{".jpg": true, ".jpeg": true, ".png": true, ".bmp": true}

func main() {

	var verbosity logrus.Level
	if err := conf.Get(".Logging.Verbosity", &verbosity); err != nil {
		log.Fatalf("Can't load verbosity from config files: %v", err)
	}

	// Logging
	os.MkdirAll(filepath.Join(".", "log"), os.ModePerm)
	rotateFileHook, err := rotatefilehook.NewRotateFileHook(rotatefilehook.RotateFileConfig{
		Filename:   filepath.Join(".", "log", "newest.log"),
		MaxSize:    50, // megabytes
		MaxBackups: 3,
		MaxAge:     365, //days
		Level:      verbosity,
		Formatter: &logrus.TextFormatter{
			FullTimestamp:   true,
			TimestampFormat: time.RFC3339,
			CallerPrettyfier: func(f *runtime.Frame) (string, string) {
				//return fmt.Sprintf("%s()", f.Function), fmt.Sprintf("%s:%d", filepath.Base(f.File), f.Line)
				return fmt.Sprintf("%s():%d", f.Function, f.Line), ""
			},
		},
	})

	if err != nil {
		log.Fatalf("Failed to initialize file rotate hook: %v", err)
	}

	log.SetReportCaller(true)
	log.AddHook(rotateFileHook)
	log.SetLevel(verbosity)
	log.SetOutput(colorable.NewColorableStdout())
	log.SetFormatter(&logrus.TextFormatter{
		ForceColors:     true,
		FullTimestamp:   true,
		TimestampFormat: time.RFC3339,
		CallerPrettyfier: func(f *runtime.Frame) (string, string) {
			//return fmt.Sprintf("%s()", f.Function), fmt.Sprintf("%s:%d", filepath.Base(f.File), f.Line)
			return fmt.Sprintf("%s():%d", f.Function, f.Line), ""
		},
	})

	var cachePath string
	if err := conf.Get(".Cache.Path", &cachePath); err != nil {
		log.Fatalf("Can't load cache path from config files: %v", err)
	}
	os.MkdirAll(cachePath, os.ModePerm)
	cache = NewCache(cachePath)

	// Add routes to the webserver
	serverUIInit()

	loadSources()

	log.Infof("Galago %v started", version)

	var addr string
	if err := conf.Get(".Server.ListenAddress", &addr); err != nil {
		log.Fatalf("Can't load server listen address from config files: %v", err)
	}
	log.Infof("Listening at %v", addr)
	log.Fatalf("Webserver returned error: %v", http.ListenAndServe(addr, router))
}