summaryrefslogtreecommitdiff
path: root/main.go
blob: 734d257de8532f58b75262c20b5d704b42bbce94 (plain)
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package main

import (
	"embed"
	"fmt"
	"net/http"
	"os"
	"time"

	"go.wit.com/dev/alexflint/arg"
	"go.wit.com/gui"
	"go.wit.com/lib/protobuf/forgepb"
	"go.wit.com/log"
)

// are sent via -ldflags at buildtime
var VERSION string
var BUILDTIME string
var ARGNAME string = "forged"

//go:embed resources/*
var resources embed.FS

var HOSTNAME string = "forge.wit.com"
var LIBDIR string = "/var/lib/forged/"
var FORGEDIR string = "/home/forge"

func main() {
	me = new(mainType)
	gui.InitArg()
	me.pp = arg.MustParse(&argv)

	if argv.Bash {
		argv.doBash()
		os.Exit(0)
	}
	if len(argv.BashAuto) != 0 {
		argv.doBashAuto()
		os.Exit(0)
	}

	if argv.Hostname != "" {
		HOSTNAME = argv.Hostname
	}

	// the default forged dir is /home/forge
	if os.Getenv("FORGE_GOSRC") == "" {
		os.Setenv("FORGE_GOSRC", "/home/forge")
	}

	if os.Getenv("FORGE_PATCHDIR") == "" {
		os.Setenv("FORGE_PATCHDIR", "/var/lib/forged")
	}

	me.forge = forgepb.RawInitPB()

	if err := me.forge.LoadPatchsets(); err != nil {
		badExit(err)
	}

	if argv.List != nil {
		doList()
		okExit("")
	}

	if argv.Merge != nil {
		if err := doMerge(); err != nil {
			badExit(err)
		}
		okExit("")
	}

	if argv.Pull != nil {
		log.Info("pull here")
		okExit("")
	}

	if argv.Daemon == true {
		log.Info("Running in --daemon mode")
		http.HandleFunc("/", okHandler)
		// go https() // use caddy instead
		p := fmt.Sprintf(":%d", argv.Port)
		log.Println(argv.Version(), "HOSTNAME set to:", HOSTNAME)
		log.Println("Running on port", "http://localhost"+p)
		log.Println("Running on port", "http://localhost"+p+"/ipv6.png")
		err := http.ListenAndServe(p, nil)
		if err != nil {
			log.Println("Error starting server:", err)
		}
	} else {
		log.Info("--daemon was not set. Just list the patches.")
		doList()
	}
}

func formatDuration(d time.Duration) string {
	seconds := int(d.Seconds()) % 60
	minutes := int(d.Minutes()) % 60
	hours := int(d.Hours()) % 24
	days := int(d.Hours()) / 24

	result := ""
	if days > 0 {
		result += fmt.Sprintf("%dd ", days)
		return result
	}
	if hours > 0 {
		result += fmt.Sprintf("%dh ", hours)
		return result
	}
	if minutes > 0 {
		result += fmt.Sprintf("%dm ", minutes)
		return result
	}
	if seconds > 0 {
		result += fmt.Sprintf("%ds", seconds)
	}
	return result
}