aboutsummaryrefslogtreecommitdiff
path: root/main.go
blob: 5e836f1bea2c7674e4378a00f17c139bf2a149b4 (plain) (blame)
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
// SPDX-License-Identifier: AGPL-3.0-only
// SPDX-FileContributor: Runxi Yu <https://runxiyu.org>

package main

import (
	"errors"
	"flag"
	"net"
	"net/http"
	"syscall"

	"go.lindenii.runxiyu.org/lindenii-common/clog"
	_ "net/http/pprof"
)

func main() {
	config_path := flag.String(
		"config",
		"/etc/lindenii/forge.scfg",
		"path to configuration file",
	)
	flag.Parse()

	if err := load_config(*config_path); err != nil {
		clog.Fatal(1, "Loading configuration: "+err.Error())
	}
	if err := deploy_hooks_to_filesystem(); err != nil {
		clog.Fatal(1, "Deploying hooks to filesystem: "+err.Error())
	}
	if err := load_templates(); err != nil {
		clog.Fatal(1, "Loading templates: "+err.Error())
	}

	// UNIX socket listener for hooks
	var hooks_listener net.Listener
	var err error
	hooks_listener, err = net.Listen("unix", config.Hooks.Socket)
	if errors.Is(err, syscall.EADDRINUSE) {
		clog.Warn("Removing stale socket " + config.Hooks.Socket)
		if err = syscall.Unlink(config.Hooks.Socket); err != nil {
			clog.Fatal(1, "Removing stale socket: "+err.Error())
		}
		if hooks_listener, err = net.Listen("unix", config.Hooks.Socket); err != nil {
			clog.Fatal(1, "Listening hooks: "+err.Error())
		}
	} else if err != nil {
		clog.Fatal(1, "Listening hooks: "+err.Error())
	}
	clog.Info("Listening hooks on unix " + config.Hooks.Socket)
	go func() {
		if err = serve_git_hooks(hooks_listener); err != nil {
			clog.Fatal(1, "Serving hooks: "+err.Error())
		}
	}()

	// SSH listener
	ssh_listener, err := net.Listen(config.SSH.Net, config.SSH.Addr)
	if errors.Is(err, syscall.EADDRINUSE) && config.SSH.Net == "unix" {
		clog.Warn("Removing stale socket " + config.SSH.Addr)
		if err = syscall.Unlink(config.SSH.Addr); err != nil {
			clog.Fatal(1, "Removing stale socket: "+err.Error())
		}
		if ssh_listener, err = net.Listen(config.SSH.Net, config.SSH.Addr); err != nil {
			clog.Fatal(1, "Listening SSH: "+err.Error())
		}
	} else if err != nil {
		clog.Fatal(1, "Listening SSH: "+err.Error())
	}
	clog.Info("Listening SSH on " + config.SSH.Net + " " + config.SSH.Addr)
	go func() {
		if err = serve_ssh(ssh_listener); err != nil {
			clog.Fatal(1, "Serving SSH: "+err.Error())
		}
	}()

	// HTTP listener
	http_listener, err := net.Listen(config.HTTP.Net, config.HTTP.Addr)
	if errors.Is(err, syscall.EADDRINUSE) && config.HTTP.Net == "unix" {
		clog.Warn("Removing stale socket " + config.HTTP.Addr)
		if err = syscall.Unlink(config.HTTP.Addr); err != nil {
			clog.Fatal(1, "Removing stale socket: "+err.Error())
		}
		if http_listener, err = net.Listen(config.HTTP.Net, config.HTTP.Addr); err != nil {
			clog.Fatal(1, "Listening HTTP: "+err.Error())
		}
	} else if err != nil {
		clog.Fatal(1, "Listening HTTP: "+err.Error())
	}
	clog.Info("Listening HTTP on " + config.HTTP.Net + " " + config.HTTP.Addr)
	go func() {
		if err = http.Serve(http_listener, &http_router_t{}); err != nil {
			clog.Fatal(1, "Serving HTTP: "+err.Error())
		}
	}()

	// Pprof listener
	pprof_listener, err := net.Listen("tcp", "localhost:6060")
	if err != nil {
		clog.Fatal(1, "Listening pprof: "+err.Error())
	}
	clog.Info("Listening pprof on tcp localhost:6060")
	go func() {
		if err = http.Serve(pprof_listener, nil); err != nil {
			clog.Fatal(1, "Serving pprof: "+err.Error())
		}
	}()

	select {}
}