19 "github.com/Sirupsen/logrus"
20 "github.com/iotbzh/xds-agent/lib/common"
21 "github.com/iotbzh/xds-agent/lib/xdsconfig"
22 "github.com/syncthing/syncthing/lib/config"
26 type SyncThing struct {
36 exitSTChan chan ExitChan
37 exitSTIChan chan ExitChan
38 client *common.HTTPClient
42 // ExitChan Channel used for process exit
43 type ExitChan struct {
48 // NewSyncThing creates a new instance of Syncthing
49 func NewSyncThing(conf *xdsconfig.Config, log *logrus.Logger) *SyncThing {
50 var url, apiKey, home, binDir string
53 stCfg := conf.FileConf.SThgConf
55 url = stCfg.GuiAddress
56 apiKey = stCfg.GuiAPIKey
62 url = "http://localhost:8384"
64 if url[0:7] != "http://" {
73 if binDir, err = filepath.Abs(filepath.Dir(os.Args[0])); err != nil {
74 binDir = "/usr/local/bin"
83 logsDir: conf.FileConf.LogsDir,
90 // Start Starts syncthing process
91 func (s *SyncThing) startProc(exeName string, args []string, env []string, eChan *chan ExitChan) (*exec.Cmd, error) {
93 // Kill existing process (useful for debug ;-) )
94 if os.Getenv("DEBUG_MODE") != "" {
95 exec.Command("bash", "-c", "pkill -9 "+exeName).Output()
98 path, err := exec.LookPath(path.Join(s.binDir, exeName))
100 return nil, fmt.Errorf("Cannot find %s executable in %s", exeName, s.binDir)
102 cmd := exec.Command(path, args...)
103 cmd.Env = os.Environ()
104 for _, ev := range env {
105 cmd.Env = append(cmd.Env, ev)
110 logFilename := filepath.Join(s.logsDir, exeName+".log")
112 outfile, err := os.Create(logFilename)
114 return nil, fmt.Errorf("Cannot create log file %s", logFilename)
117 cmdOut, err := cmd.StdoutPipe()
119 return nil, fmt.Errorf("Pipe stdout error for : %s", err)
122 go io.Copy(outfile, cmdOut)
130 *eChan = make(chan ExitChan, 1)
131 go func(c *exec.Cmd, oF *os.File) {
133 sts, err := c.Process.Wait()
135 s := sts.Sys().(syscall.WaitStatus)
136 status = s.ExitStatus()
141 s.log.Debugf("%s exited with status %d, err %v", exeName, status, err)
143 *eChan <- ExitChan{status, err}
149 // Start Starts syncthing process
150 func (s *SyncThing) Start() (*exec.Cmd, error) {
153 s.log.Infof(" ST home=%s", s.Home)
154 s.log.Infof(" ST url=%s", s.BaseURL)
159 "--gui-address=" + s.BaseURL,
163 args = append(args, "-gui-apikey=\""+s.APIKey+"\"")
164 s.log.Infof(" ST apikey=%s", s.APIKey)
166 if s.log.Level == logrus.DebugLevel {
167 args = append(args, "-verbose")
171 "STNODEFAULTFOLDER=1",
174 s.STCmd, err = s.startProc("syncthing", args, env, &s.exitSTChan)
176 // Use autogenerated apikey if not set by config.json
178 if fd, err := os.Open(filepath.Join(s.Home, "config.xml")); err == nil {
180 if b, err := ioutil.ReadAll(fd); err == nil {
181 re := regexp.MustCompile("<apikey>(.*)</apikey>")
182 key := re.FindStringSubmatch(string(b))
193 // StartInotify Starts syncthing-inotify process
194 func (s *SyncThing) StartInotify() (*exec.Cmd, error) {
196 exeName := "syncthing-inotify"
198 s.log.Infof(" STI url=%s", s.BaseURL)
201 "-target=" + s.BaseURL,
204 args = append(args, "-api="+s.APIKey)
205 s.log.Infof("%s uses apikey=%s", exeName, s.APIKey)
207 if s.log.Level == logrus.DebugLevel {
208 args = append(args, "-verbosity=4")
213 s.STICmd, err = s.startProc(exeName, args, env, &s.exitSTIChan)
218 func (s *SyncThing) stopProc(pname string, proc *os.Process, exit chan ExitChan) {
219 if err := proc.Signal(os.Interrupt); err != nil {
220 s.log.Infof("Proc interrupt %s error: %s", pname, err.Error())
224 case <-time.After(time.Second):
225 // A bigger bonk on the head.
226 if err := proc.Signal(os.Kill); err != nil {
227 s.log.Infof("Proc term %s error: %s", pname, err.Error())
232 s.log.Infof("%s stopped (PID %d)", pname, proc.Pid)
235 // Stop Stops syncthing process
236 func (s *SyncThing) Stop() {
240 s.stopProc("syncthing", s.STCmd.Process, s.exitSTChan)
244 // StopInotify Stops syncthing process
245 func (s *SyncThing) StopInotify() {
249 s.stopProc("syncthing-inotify", s.STICmd.Process, s.exitSTIChan)
253 // Connect Establish HTTP connection with Syncthing
254 func (s *SyncThing) Connect() error {
256 s.client, err = common.HTTPNewClient(s.BaseURL,
257 common.HTTPClientConfig{
259 HeaderClientKeyName: "X-Syncthing-ID",
262 msg := ": " + err.Error()
263 if strings.Contains(err.Error(), "connection refused") {
264 msg = fmt.Sprintf("(url: %s)", s.BaseURL)
266 return fmt.Errorf("ERROR: cannot connect to Syncthing %s", msg)
269 return fmt.Errorf("ERROR: cannot connect to Syncthing (null client)")
272 s.client.SetLogger(s.log)
277 // IDGet returns the Syncthing ID of Syncthing instance running locally
278 func (s *SyncThing) IDGet() (string, error) {
280 if err := s.client.HTTPGet("system/status", &data); err != nil {
283 status := make(map[string]interface{})
284 json.Unmarshal(data, &status)
285 return status["myID"].(string), nil
288 // ConfigGet returns the current Syncthing configuration
289 func (s *SyncThing) ConfigGet() (config.Configuration, error) {
291 config := config.Configuration{}
292 if err := s.client.HTTPGet("system/config", &data); err != nil {
295 err := json.Unmarshal(data, &config)
299 // ConfigSet set Syncthing configuration
300 func (s *SyncThing) ConfigSet(cfg config.Configuration) error {
301 body, err := json.Marshal(cfg)
305 return s.client.HTTPPost("system/config", string(body))