7568 - mu - Soul of a tiny new machine. More thorough tests → More comprehensible and rewrite-friendly software → More resilient society.

about summary refs log tree commit diff stats

diff options

author	Kartik Agaram <vc@akkartik.com>	2021-01-27 20:24:41 -0800
committer	Kartik Agaram <vc@akkartik.com>	2021-01-27 20:24:41 -0800
commit	c98476aa0960d64ad4fc259945ad22bcd27cb5a7 (patch)
tree	4e3a03e081efa6ace8f2328597c9cb12eba28470 /116write-buffered.subx
parent	bcb656a7e8eafe5cf9cde4e176cda0c7978464a2 (diff)
download	mu-c98476aa0960d64ad4fc259945ad22bcd27cb5a7.tar.gz

7568

Diffstat (limited to '116write-buffered.subx')

0 files changed, 0 insertions, 0 deletions

etwtxt/commit/init.go?h=v0.1.1&id=06cffd89aeea4d34c987a1e7583af7525b8534e8'>06cffd8 ^

bd23ef0 ^

06cffd8 ^

bd23ef0 ^

06cffd8 ^

f2d85b6 ^

a0be15e ^

06cffd8 ^

cd635e6 ^

06cffd8 ^

d083ce6 ^

06cffd8 ^

bd23ef0 ^

920306c ^

efa99ed ^

df1d1ef ^

c896e6b ^

df1d1ef ^

2b0d4a5 ^

4f0847b ^

f2d85b6 ^

1f8b2ce ^

06cffd8 ^

06cffd8 ^

920306c ^

c896e6b ^

5310d08 ^

06cffd8 ^

f2d85b6 ^

1f8b2ce ^

f2d85b6 ^

bd23ef0 ^

f2d85b6 ^

67864b2 ^

f2d85b6 ^

06cffd8 ^

1f8b2ce ^

cd635e6 ^

06cffd8 ^

1f8b2ce ^

06cffd8 ^

cd635e6 ^

bded77f ^

cd635e6 ^

bded77f ^

cd635e6 ^

06cffd8 ^

1f8b2ce ^

cd635e6 ^

bd23ef0 ^

cd635e6 ^

313ab71 ^

a0be15e ^

cd635e6 ^

a0be15e ^

1f8b2ce ^

a0be15e ^

a0be15e ^

1f8b2ce ^

313ab71 ^

bd23ef0 ^

d083ce6 ^

1f8b2ce ^

bd23ef0 ^

1f8b2ce ^

313ab71 ^

1f8b2ce ^

1f8b2ce ^

cd635e6 ^

06cffd8 ^

1f8b2ce ^

cd635e6 ^

d083ce6 ^

1f8b2ce ^

cd635e6 ^

1f8b2ce ^

06cffd8 ^

cd635e6 ^

a0be15e ^

06cffd8 ^

a0be15e ^

1f8b2ce ^

920306c ^

b29e1c1 ^

bd23ef0 ^

cd635e6 ^

bd23ef0 ^

b29e1c1 ^

bd23ef0 ^

c896e6b ^

b29e1c1 ^

bd23ef0 ^

5310d08 ^

5310d08 ^

c896e6b ^

cd635e6 ^

c896e6b ^

cd635e6 ^

b29e1c1 ^

c896e6b ^

cd635e6 ^

5310d08 ^

d083ce6 ^

5310d08 ^

cd635e6 ^

b29e1c1 ^

d083ce6 ^

efa99ed ^

c896e6b ^

5310d08 ^

1f8b2ce ^

bd23ef0 ^

1f8b2ce ^

1
2
3

6
7

8
9

10
11

13
14

16
17

18
19

20
21

22
23

24
25

28
29

36
37
38

39
40
41

66
67

75
76

79
80
81

118
119

144
145
146

167
168

176
177

196
197

207
208
209

214
215

217
218

220
221

233
234

244
245

249
250

277
278

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)

// templates
var tmpls *template.Template

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

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

// static assets cache
var staticCache = &struct {
	index    []byte
	indexMod time.Time
	css      []byte
	cssMod   time.Time
}{
	index:    nil,
	indexMod: time.Time{},
	css:      nil,
	cssMod:   time.Time{},
}

func initGetwtxt() {
	checkFlags()
	titleScreen()
	initConfig()
	initLogging()
	tmpls = initTemplates()
	initDatabase()
	watchForInterrupt()
}

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

func initConfig() {

	viper.SetConfigName("getwtxt")
	viper.SetConfigType("json")
	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("ListenPort", 9001)
	viper.SetDefault("LogFile", "getwtxt.log")
	viper.SetDefault("DatabasePath", "getwtxt.db")
	viper.SetDefault("StdoutLogging", false)
	viper.SetDefault("ReCacheInterval", "1h")
	viper.SetDefault("DatabasePushInterval", "5m")

	confObj.Mu.Lock()
	confObj.Port = viper.GetInt("ListenPort")
	confObj.LogFile = viper.GetString("LogFile")
	confObj.DBPath = viper.GetString("DatabasePath")
	log.Printf("Using database: %v\n", confObj.DBPath)
	confObj.StdoutLogging = viper.GetBool("StdoutLogging")
	if confObj.StdoutLogging {
		log.Printf("Logging to stdout\n")
	} else {
		log.Printf("Logging to %v\n", confObj.LogFile)
	}
	confObj.CacheInterval = viper.GetDuration("StatusFetchInterval")
	log.Printf("User status fetch interval: %v\n", confObj.CacheInterval)
	confObj.DBInterval = viper.GetDuration("DatabasePushInterval")
	log.Printf("Database push interval: %v\n", confObj.DBInterval)
	confObj.LastCache = time.Now()
	confObj.LastPush = time.Now()
	confObj.Version = getwtxt
	confObj.Instance.Name = viper.GetString("Instance.SiteName")
	confObj.Instance.URL = viper.GetString("Instance.URL")
	confObj.Instance.Owner = viper.GetString("Instance.OwnerName")
	confObj.Instance.Mail = viper.GetString("Instance.Email")
	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()

	// reassign values to the config object
	confObj.Mu.Lock()
	confObj.LogFile = viper.GetString("LogFile")
	confObj.DBPath = viper.GetString("DatabasePath")
	confObj.StdoutLogging = viper.GetBool("StdoutLogging")
	confObj.CacheInterval = viper.GetDuration("StatusFetchInterval")
	confObj.DBInterval = viper.GetDuration("DatabasePushInterval")
	confObj.Instance.Name = viper.GetString("Instance.SiteName")
	confObj.Instance.URL = viper.GetString("Instance.URL")
	confObj.Instance.Owner = viper.GetString("Instance.OwnerName")
	confObj.Instance.Mail = viper.GetString("Instance.Email")
	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() {
	confObj.Mu.RLock()
	db, err := leveldb.OpenFile(confObj.DBPath, nil)
	confObj.Mu.RUnlock()
	if err != nil {
		log.Fatalf("%v\n", err)
	}

	// Send the database reference into
	// the aether.
	dbChan <- db

	pullDatabase()
	go cacheAndPush()
}

// 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 := <-dbChan
			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(dbChan)
			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:
`)
}


context:
space:
mode: