path: root/init.go

package main

import (
	"fmt"
	"html/template"
	"log"
	"os"
	"os/signal"
	"time"

	"github.com/fsnotify/fsnotify"
	"github.com/getwtxt/registry"
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
	"github.com/syndtr/goleveldb/leveldb"
)

const getwtxt = "0.1"

// command line flags
var (
	flagVersion *bool = pflag.BoolP("version", "v", false, "Display version information, then exit")
	flagHelp    *bool = pflag.BoolP("help", "h", false, "")
)

// config object
var confObj = &configuration{}

// signals to close the log file
var closelog = make(chan bool, 1)

// used to transmit database pointer after
// initialization
var dbChan = make(chan *leveldb.DB, 1)

// provides access to the database so it
// can be closed outside of the init function's
// scope.
var dbCloseChan = make(chan *leveldb.DB, 1)

// templates
var tmpls *template.Template

// registry index
var twtxtCache = registry.NewIndex()

// remote registry listing
var remoteRegistries = &RemoteRegistries{}

func init() {
	checkFlags()
	titleScreen()
	initConfig()
	initLogging()
	tmpls = initTemplates()
	initDatabase()
	go cacheAndPush()
	watchForInterrupt()
}

func checkFlags() {
	pflag.Parse()
	if *flagVersion {
		titleScreen()
		os.Exit(0)
	}
	if *flagHelp {
		titleScreen()
		helpScreen()
		os.Exit(0)
	}
}

func initConfig() {

	viper.SetConfigName("getwtxt")
	viper.AddConfigPath(".")
	viper.AddConfigPath("/usr/local/getwtxt")
	viper.AddConfigPath("/etc")
	viper.AddConfigPath("/usr/local/etc")

	log.Printf("Loading configuration ...\n")
	if err := viper.ReadInConfig(); err != nil {
		log.Printf("Error reading config file: %v\n", err)
		log.Printf("Using defaults ...\n")
	}

	// separate thread to watch for config file changes.
	// will log event then run rebindConfig()
	viper.WatchConfig()
	viper.OnConfigChange(func(e fsnotify.Event) {
		log.Printf("Config file change detected. Reloading...\n")
		rebindConfig()
	})

	viper.SetDefault("port", 9001)
	viper.SetDefault("logFile", "getwtxt.log")
	viper.SetDefault("databasePath", "getwtxt.db")
	viper.SetDefault("stdoutLogging", false)
	viper.SetDefault("reCacheInterval", "1h")
	viper.SetDefault("databasePushInterval", "5m")

	updateInterval := viper.GetString("reCacheInterval")
	dur, err := time.ParseDuration(updateInterval)
	if err != nil {
		log.Printf("Unable to parse registry cache update interval. Defaulting to hourly. Msg: %v\n", err)
		dur, _ = time.ParseDuration("1h")
	}

	dbPushInterval := viper.GetString("databasePushInterval")
	dbDur, err := time.ParseDuration(dbPushInterval)
	if err != nil {
		log.Printf("Unable to parse database push interval. Defaulting to every five minutes. Msg: %v\n", err)
		dbDur, _ = time.ParseDuration("5m")
	}

	thetime := time.Now()

	confObj.mu.Lock()
	confObj.port = viper.GetInt("port")
	confObj.logFile = viper.GetString("logFile")
	confObj.dbPath = viper.GetString("databasePath")
	confObj.stdoutLogging = viper.GetBool("stdoutLogging")
	confObj.cacheInterval = dur
	confObj.dbInterval = dbDur
	confObj.lastCache = thetime
	confObj.lastPush = thetime
	confObj.version = getwtxt
	confObj.Instance.Name = viper.GetString("instance.name")
	confObj.Instance.URL = viper.GetString("instance.url")
	confObj.Instance.Owner = viper.GetString("instance.owner")
	confObj.Instance.Mail = viper.GetString("instance.mail")
	confObj.Instance.Desc = viper.GetString("instance.description")
	confObj.mu.Unlock()
}

func initLogging() {

	// only open a log file if it's necessary
	confObj.mu.RLock()

	if confObj.stdoutLogging {
		log.SetOutput(os.Stdout)

	} else {

		logfile, err := os.OpenFile(confObj.logFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600)
		if err != nil {
			log.Printf("Could not open log file: %v\n", err)
		}

		// Listen for the signal to close the log file
		// in a separate thread. Passing it as an argument
		// to prevent race conditions when the config is
		// reloaded.
		go func(logfile *os.File) {

			<-closelog
			log.Printf("Closing log file ...\n")

			err = logfile.Close()
			if err != nil {
				log.Printf("Couldn't close log file: %v\n", err)
			}
		}(logfile)

		log.SetOutput(logfile)
	}
	confObj.mu.RUnlock()
}

func rebindConfig() {

	// signal to close the log file then wait
	confObj.mu.RLock()
	if !confObj.stdoutLogging {
		closelog <- true
	}
	confObj.mu.RUnlock()

	// re-parse update interval
	nter := viper.GetString("reCacheInterval")
	dur, err := time.ParseDuration(nter)
	if err != nil {
		log.Printf("Unable to parse update interval. Defaulting to once daily. Msg: %v\n", err)
		dur, _ = time.ParseDuration("1h")
	}

	// re-parse database backup interval
	dbPushInterval := viper.GetString("databasePushInterval")
	dbDur, err := time.ParseDuration(dbPushInterval)
	if err != nil {
		log.Printf("Unable to parse database push interval. Defaulting to every five minutes. Msg: %v\n", err)
		dbDur, _ = time.ParseDuration("5m")
	}

	// reassign values to the config object
	confObj.mu.Lock()
	confObj.port = viper.GetInt("port")
	confObj.logFile = viper.GetString("logFile")
	confObj.stdoutLogging = viper.GetBool("stdoutLogging")
	confObj.dbPath = viper.GetString("databasePath")
	confObj.cacheInterval = dur
	confObj.dbInterval = dbDur
	confObj.Instance.Name = viper.GetString("instance.name")
	confObj.Instance.URL = viper.GetString("instance.url")
	confObj.Instance.Owner = viper.GetString("instance.owner")
	confObj.Instance.Mail = viper.GetString("instance.mail")
	confObj.Instance.Desc = viper.GetString("instance.description")
	confObj.mu.Unlock()

	// reinitialize logging
	initLogging()
}

// Parse the HTML templates
func initTemplates() *template.Template {
	return template.Must(template.ParseFiles("assets/tmpl/index.html"))
}

// Pull DB data into cache, if available
func initDatabase() {
	db, err := leveldb.OpenFile(confObj.dbPath, nil)
	if err != nil {
		log.Fatalf("%v\n", err)
	}

	dbChan <- db
	dbCloseChan <- db

	pullDatabase()
}

// Watch for SIGINT aka ^C
// Close the log file then exit
func watchForInterrupt() {
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)

	go func() {
		for sigint := range c {

			log.Printf("\n\nCaught %v. Cleaning up ...\n", sigint)
			confObj.mu.RLock()

			// Close the database cleanly
			log.Printf("Closing database connection to %v...\n", confObj.dbPath)
			db := <-dbCloseChan
			if err := db.Close(); err != nil {
				log.Printf("%v\n", err)
			}

			if !confObj.stdoutLogging {
				// signal to close the log file
				closelog <- true
			}

			confObj.mu.RUnlock()
			close(dbCloseChan)
			close(closelog)

			// Let everything catch up
			time.Sleep(100 * time.Millisecond)
			os.Exit(0)
		}
	}()
}

func titleScreen() {
	fmt.Printf(`
	
            _            _        _
  __ _  ___| |___      _| |___  _| |_
 / _  |/ _ \ __\ \ /\ / / __\ \/ / __|
| (_| |  __/ |_ \ V  V /| |_ >  <| |_
 \__, |\___|\__| \_/\_/  \__/_/\_\\__|
 |___/
             version ` + getwtxt + `
      github.com/getwtxt/getwtxt
               GPL  v3	
`)
}

func helpScreen() {
	fmt.Printf(`
              Help File

  Sections:
`)
}