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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
package main
import (
"bytes"
"fmt"
"golang.org/x/crypto/openpgp"
)
// error constants
const (
errNoFile = -(1 << iota)
errKeyring
errNoRepo
errNoCommit
errSignature
)
// spoolMsg type represents messages received from the spool and
// sent to workers
type spoolMsg struct {
ID string `yaml:"m_id"`
Repo string `yaml:"m_repo"`
Branch string `yaml:"m_branch"`
OldRev string `yaml:"m_oldrev"`
NewRev string `yaml:"m_newrev"`
Path string
}
// An action represents a script of a command configured on the server side
type action struct {
URL string `yaml:"a_url"`
Hash string `yaml:"a_hash"`
}
// commandCfg represents a command configured on the server side
type commandCfg struct {
Name string `yaml:"t_name"`
Keyrings []string `yaml:"t_keyrings"`
Actions []action `yaml:"t_actions"`
}
// workerCfg represents the static configuration of a worker
type workerCfg struct {
Name string `yaml:"w_name"`
Repos []string `yaml:"w_repos"`
Folder string `yaml:"w_folder"`
LogFile string `yaml:"w_logfile"`
CfgFile string `yaml:"w_cfgfile"`
// Keyrings []string `yaml:"w_keyrings"`
Commands []commandCfg `yaml:"w_commands"`
CommandKeys map[string]map[string]bool
}
// workerState represents the runtime state of a worker
type workerState struct {
Keys map[string]openpgp.KeyRing
MsgChan chan spoolMsg
StatusChan chan spoolMsg
}
// worker represents the configuration and state of a worker
type worker struct {
workerCfg `yaml:",inline"`
workerState
}
// masterCfg represents the static configuration of the master
type masterCfg struct {
Spooldir string `yaml:"s_spooldir"`
LogFile string `yaml:"s_logfile"`
LogPrefix string `yaml:"s_logprefix"`
Workers []worker `yaml:"s_workers"`
}
// masterState represents the runtime state of the master
type masterState struct {
Spooler chan spoolMsg
StatusChan chan spoolMsg
Repos map[string][]*worker
WorkingMsg map[string]int
}
// master represents the configuration and state of the master
type master struct {
masterCfg `yaml:",inline"`
masterState
}
// clientCmd is the type of commands sent by clients
type clientCmd struct {
Cmd string `yaml:"s_cmd"`
Args []string `yaml:"s_args"`
}
// clientMsg is the list of commands sent by a client
type clientMsg struct {
Commands []clientCmd `yaml:"scorsh"`
}
////////////////////////
func (cfg *master) String() string {
var buff bytes.Buffer
fmt.Fprintf(&buff, "spooldir: %s\n", cfg.Spooldir)
fmt.Fprintf(&buff, "logfile: %s\n", cfg.LogFile)
fmt.Fprintf(&buff, "logprefix: %s\n", cfg.LogPrefix)
fmt.Fprintf(&buff, "Workers: \n")
for _, w := range cfg.Workers {
fmt.Fprintf(&buff, "%s", &w)
}
return buff.String()
}
func (msg *spoolMsg) String() string {
var buff bytes.Buffer
fmt.Fprintf(&buff, "Id: %s\n", msg.ID)
fmt.Fprintf(&buff, "Repo: %s\n", msg.Repo)
fmt.Fprintf(&buff, "Branch: %s\n", msg.Branch)
fmt.Fprintf(&buff, "OldRev: %s\n", msg.OldRev)
fmt.Fprintf(&buff, "Newrev: %s\n", msg.NewRev)
fmt.Fprintf(&buff, "Path: %s\n", msg.Path)
return buff.String()
}
func (w *worker) String() string {
var buff bytes.Buffer
fmt.Fprintf(&buff, "Name: %s\n", w.Name)
fmt.Fprintf(&buff, "Repos: %s\n", w.Repos)
fmt.Fprintf(&buff, "Folder: %s\n", w.Folder)
fmt.Fprintf(&buff, "LogFile: %s\n", w.LogFile)
fmt.Fprintf(&buff, "CfgFile: %s\n", w.CfgFile)
// fmt.Fprintf(&buff, "Keyrings: %s\n", w.Keyrings)
return buff.String()
}
func (msg *clientMsg) String() string {
var buff bytes.Buffer
for _, c := range msg.Commands {
fmt.Fprintf(&buff, "s_cmd: %s\n", c.Cmd)
for _, a := range c.Args {
fmt.Fprintf(&buff, " s_args: %s\n", a)
}
}
return buff.String()
}
|
