nemon/main.go at main · SpaceDefer/nemon · 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
package main

import (
	"flag"
	"fmt"
	"log"
	"os"
	"runtime"
	"strings"

	"nemon/nemon"
)

var (
	key  = flag.String("key", "", "worker key, same as the coordinator") // key is the nemon product key
	mode = flag.String("mode", "worker", "coordinator or worker")        // mode is the mode nemon runs in
	dev  = flag.Bool("dev", false, "start dev")                          // dev decides whether to start dev environment or not
)

func Init() error {
	homeDir, err := os.UserHomeDir()
	if err != nil {
		return err
	}
	configDir := fmt.Sprintf(homeDir + "/.nemon_config/")
	fmt.Printf("config dir: %v\n", configDir)
	err = os.Mkdir(configDir, 0750)
	if err != nil && !os.IsExist(err) {
		return err
	}
	flag.Parse()
	hostname, err := os.Hostname()
	if err != nil {
		return err
	}
	username := os.Getenv("SUDO_USER")
	if err = os.Setenv("MODE", *mode); err != nil {
		return err
	}
	if err = os.Setenv("OS", runtime.GOOS); err != nil {
		return err
	}
	if err = os.Setenv("USERNAME", username); err != nil {
		return err
	}
	if err = os.Setenv("HOSTNAME", hostname); err != nil {
		return err
	}
	if err = os.Setenv("NEMONKEY", *key); err != nil {
		return err
	}
	if err = os.Setenv("CONFIG_DIR", configDir); err != nil {
		return err
	}
	if *dev {
		if err = os.Setenv("DEV", "dev"); err != nil {
			return err
		}
	}
	return nil
}

func main() {
	err := Init()
	if err != nil {
		log.Fatal(err)
	}
	if strings.ToLower(*mode) == "worker" {
		nemon.StartWorker()
	} else {
		nemon.StartCoordinator()
	}
}