cmd/safcm-remote/sync/commands.go

   1 // MsgSyncReq: run commands on the remote host
   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 sync
  19
  20 import (
  21         "fmt"
  22         "os"
  23         "os/exec"
  24         "strings"
  25
  26         "ruderich.org/simon/safcm"
  27         "ruderich.org/simon/safcm/cmd/safcm-remote/run"
  28 )
  29
  30 func (s *Sync) syncCommands() error {
  31         // Run triggered commands first
  32         for _, path := range s.triggers {
  33                 for _, x := range s.req.Files[path].TriggerCommands {
  34                         err := s.syncCommand(x, path)
  35                         if err != nil {
  36                                 return err
  37                         }
  38                 }
  39         }
  40         // Regular commands afterwards so they can react on triggers if
  41         // necessary
  42         for _, x := range s.req.Commands {
  43                 err := s.syncCommand(x, "")
  44                 if err != nil {
  45                         return err
  46                 }
  47         }
  48         return nil
  49 }
  50
  51 func (s *Sync) syncCommand(command string, trigger string) error {
  52         s.resp.CommandChanges = append(s.resp.CommandChanges,
  53                 safcm.CommandChange{
  54                         Command: command,
  55                         Trigger: trigger,
  56                 })
  57         if s.req.DryRun {
  58                 return nil
  59         }
  60         change := &s.resp.CommandChanges[len(s.resp.CommandChanges)-1]
  61
  62         cmd := exec.Command("/bin/sh", "-c", command)
  63         cmd.Env = safcmEnviroment(s.req.Groups)
  64         // Cannot use cmd.CombinedOutputCmd() because we need another log
  65         // level (here the command is the actual change and not a side effect)
  66         // and different error handling.
  67         s.log.Verbosef("commands: running %s", run.QuoteForDebug(cmd))
  68         out, err := s.cmd.Runner.CombinedOutput(cmd)
  69         if len(out) > 0 {
  70                 s.log.Debug2f("commands: command output:\n%s", out)
  71         }
  72         change.Output = string(out)
  73         if err != nil {
  74                 change.Error = err.Error()
  75                 return fmt.Errorf("%q failed: %v", command, err)
  76         }
  77
  78         return nil
  79 }
  80
  81 func safcmEnviroment(groups []string) []string {
  82         env := os.Environ()
  83         // Provide additional environment variables so commands can check
  84         // group membership
  85         env = append(env,
  86                 fmt.Sprintf("SAFCM_GROUPS=%s", strings.Join(groups, " ")))
  87         for _, x := range groups {
  88                 env = append(env, fmt.Sprintf("SAFCM_GROUP_%s=%s", x, x))
  89         }
  90         return env
  91 }