app-server/app-server.go at master · postgres-ci/app-server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package main

import (
	log "github.com/Sirupsen/logrus"
	"github.com/postgres-ci/app-server/src/app"
	"github.com/postgres-ci/app-server/src/common"
	"github.com/postgres-ci/app-server/src/tools/render"

	"flag"
	"fmt"
	"os"
)

var (
	debug        bool
	pathToConfig string
)

const usage = `
Postgres-CI app-server

Usage:
    -c /path/to/config.yaml (if not set, then app-server will use environment variables)
    -debug (enable debug mode)

Environment variables:

    ADDRESS   - bind address ip:port
    TEMPLATES - path to templates dir
    LOG_LEVEL - one of: info/warning/error

    == PostgreSQL server credentials

    DB_HOST
    DB_PORT
    DB_USERNAME
    DB_PASSWORD
    DB_DATABASE
`

func main() {

	flag.BoolVar(&debug, "debug", false, "")
	flag.StringVar(&pathToConfig, "c", "", "")

	flag.Usage = func() {

		fmt.Println(usage)

		os.Exit(0)
	}

	flag.Parse()

	if log.IsTerminal() {

		log.SetFormatter(&log.TextFormatter{
			FullTimestamp:   true,
			TimestampFormat: "2006-01-02 15:04:05 MST",
		})

	} else {

		log.SetFormatter(&log.JSONFormatter{
			TimestampFormat: "2006-01-02 15:04:05 MST",
		})
	}

	config, err := common.ReadConfig(pathToConfig)

	if err != nil {

		log.Fatalf("Error reading configuration file: %v", err)
	}

	if debug {
		log.SetLevel(log.DebugLevel)
	} else {
		log.SetLevel(config.LogLevel())
	}

	if err := render.Init(config.Templates); err != nil {

		log.Fatalf("Error when loading templates: %v", err)
	}

	app := app.New(config)

	app.Run()
}