|
1 | 1 | package main |
2 | 2 |
|
3 | 3 | import ( |
4 | | - "log" |
5 | 4 | "os" |
6 | 5 | "runtime" |
7 | 6 |
|
| 7 | + "github.com/codegangsta/cli" |
8 | 8 | "github.com/deis/builder/fetcher" |
9 | 9 | "github.com/deis/builder/pkg" |
10 | | - "github.com/kelseyhightower/envconfig" |
| 10 | + "github.com/deis/builder/pkg/conf" |
| 11 | + "github.com/deis/builder/pkg/gitreceive" |
| 12 | + "github.com/deis/builder/pkg/log" |
| 13 | + "github.com/deis/builder/pkg/sshd" |
| 14 | +) |
| 15 | + |
| 16 | +const ( |
| 17 | + serverConfAppName = "deis-builder-server" |
| 18 | + gitReceiveConfAppName = "deis-builder-git-receive" |
11 | 19 | ) |
12 | 20 |
|
13 | 21 | func init() { |
14 | 22 | runtime.GOMAXPROCS(runtime.NumCPU()) |
15 | 23 | } |
16 | 24 |
|
17 | 25 | func main() { |
18 | | - var conf Config |
19 | | - if err := envconfig.Process("builder", &conf); err != nil { |
20 | | - log.Fatalf("error fetching config [%s]", err) |
21 | | - os.Exit(1) |
| 26 | + app := cli.NewApp() |
| 27 | + |
| 28 | + app.Commands = []cli.Command{ |
| 29 | + { |
| 30 | + Name: "server", |
| 31 | + Aliases: []string{"srv"}, |
| 32 | + Usage: "Run the git server", |
| 33 | + Action: func(c *cli.Context) { |
| 34 | + cnf := new(sshd.Config) |
| 35 | + if err := conf.EnvConfig(serverConfAppName, cnf); err != nil { |
| 36 | + log.Err("getting config for %s [%s]", serverConfAppName, err) |
| 37 | + os.Exit(1) |
| 38 | + } |
| 39 | + log.Info("starting fetcher on port %d", cnf.FetcherPort) |
| 40 | + go fetcher.Serve(cnf.FetcherPort) |
| 41 | + os.Exit(pkg.Run(cnf.SSHHostIP, cnf.SSHHostPort, "boot")) |
| 42 | + }, |
| 43 | + }, |
| 44 | + { |
| 45 | + Name: "git-receive", |
| 46 | + Aliases: []string{"gr"}, |
| 47 | + Usage: "Run the git-receive hook", |
| 48 | + Action: func(c *cli.Context) { |
| 49 | + cnf := new(gitreceive.Config) |
| 50 | + if err := conf.EnvConfig(gitReceiveConfAppName, cnf); err != nil { |
| 51 | + log.Err("Error getting config for %s [%s]", gitReceiveConfAppName, err) |
| 52 | + os.Exit(1) |
| 53 | + } |
| 54 | + if err := gitreceive.Run(cnf); err != nil { |
| 55 | + log.Err("running git receive hook [%s]", err) |
| 56 | + os.Exit(1) |
| 57 | + } |
| 58 | + }, |
| 59 | + }, |
22 | 60 | } |
23 | | - log.Printf("starting fetcher on port %d", conf.FetcherPort) |
24 | | - go fetcher.Serve(conf.FetcherPort) |
25 | | - os.Exit(pkg.Run(conf.SSHHostIP, conf.SSHHostPort, "boot")) |
| 61 | + |
| 62 | + app.Run(os.Args) |
26 | 63 | } |
0 commit comments