]> ruderich.org/simon Gitweb - safcm/safcm.git/blob - cmd/safcm-remote/main.go
remote: remove duplicate call to log.SetFlags()
[safcm/safcm.git] / cmd / safcm-remote / main.go
1 // Helper copied to the remote hosts to run commands and deploy configuration
2
3 // Copyright (C) 2021  Simon Ruderich
4 //
5 // This program is free software: you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation, either version 3 of the License, or
8 // (at your option) any later version.
9 //
10 // This program is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 // GNU General Public License for more details.
14 //
15 // You should have received a copy of the GNU General Public License
16 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
17
18 package main
19
20 import (
21         "fmt"
22         "log"
23         "os"
24
25         "ruderich.org/simon/safcm"
26         "ruderich.org/simon/safcm/cmd/safcm-remote/info"
27         "ruderich.org/simon/safcm/cmd/safcm-remote/run"
28         "ruderich.org/simon/safcm/cmd/safcm-remote/sync"
29 )
30
31 func main() {
32         // Timestamps are added by `safcm`
33         log.SetFlags(0)
34
35         if len(os.Args) != 1 {
36                 log.Fatalf("usage: %s", os.Args[0])
37         }
38
39         err := mainLoop()
40         if err != nil {
41                 log.Fatalf("%s: %v", os.Args[0], err)
42         }
43 }
44
45 func mainLoop() error {
46         conn := safcm.NewGobConn(os.Stdin, os.Stdout)
47
48         var logLevel safcm.LogLevel
49         logFunc := func(level safcm.LogLevel, format string, a ...interface{}) {
50                 if logLevel >= level {
51                         conn.Send(safcm.MsgLog{
52                                 Level: level,
53                                 Text:  fmt.Sprintf(format, a...),
54                         })
55                 }
56         }
57
58         var quitResp safcm.MsgQuitResp
59         for {
60                 x, err := conn.Recv()
61                 if err != nil {
62                         return err
63                 }
64
65                 var resp safcm.Msg
66                 switch x := x.(type) {
67                 case safcm.MsgInfoReq:
68                         logLevel = x.LogLevel // set log level globally
69                         resp = info.Handle(x, run.ExecRunner{}, logFunc)
70                 case safcm.MsgSyncReq:
71                         resp = sync.Handle(x, run.ExecRunner{}, logFunc)
72                 case safcm.MsgQuitReq:
73                         resp = quitResp
74                 default:
75                         return fmt.Errorf("unsupported message %#v", x)
76                 }
77
78                 err = conn.Send(resp)
79                 if err != nil {
80                         return err
81                 }
82                 if resp == quitResp {
83                         break
84                 }
85         }
86         return nil
87 }