X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=cmd%2Fsafcm%2Fsync.go;h=2f297c8dc69288aa4017f57d4e741562740784d9;hb=09dd8f99fd8598b809f4d58d4a325310b555981f;hp=10edc68755e98c81e2c890745d3e3f706d3a3691;hpb=b0f49e5d47786984e24731b200d3d3d7d6add263;p=safcm%2Fsafcm.git

diff --git a/cmd/safcm/sync.go b/cmd/safcm/sync.go
index 10edc68..2f297c8 100644
--- a/cmd/safcm/sync.go
+++ b/cmd/safcm/sync.go
@@ -44,6 +44,7 @@ type Sync struct {
 
 	isTTY bool
 
+	loop    *frontend.Loop
 	logFunc func(level safcm.LogLevel, escaped bool, msg string)
 }
 
@@ -119,10 +120,11 @@ func MainSync(args []string) error {
 			allHosts:  allHosts,
 			allGroups: allGroups,
 			isTTY:     isTTY,
+			loop:      loop,
 		}
 		s.logFunc = func(level safcm.LogLevel, escaped bool,
 			msg string) {
-			loop.Log(s, level, escaped, msg)
+			s.loop.Log(s, level, escaped, msg)
 		}
 		hosts = append(hosts, s)
 	}
@@ -214,11 +216,11 @@ are only available after the hosts were contacted.
 func logEvent(x frontend.Event, level safcm.LogLevel, isTTY bool, failed *bool) {
 	// We have multiple event sources so this is somewhat ugly.
 	var prefix, data string
-	var color Color
+	var color frontend.Color
 	if x.Error != nil {
 		prefix = "[error]"
 		data = x.Error.Error()
-		color = ColorRed
+		color = frontend.ColorRed
 		// We logged an error, tell the caller
 		*failed = true
 	} else if x.Log.Level != 0 {
@@ -237,7 +239,7 @@ func logEvent(x frontend.Event, level safcm.LogLevel, isTTY bool, failed *bool)
 			prefix = "[debug2]"
 		default:
 			prefix = fmt.Sprintf("[INVALID=%d]", x.Log.Level)
-			color = ColorRed
+			color = frontend.ColorRed
 		}
 		data = x.Log.Text
 	} else {
@@ -254,19 +256,19 @@ func logEvent(x frontend.Event, level safcm.LogLevel, isTTY bool, failed *bool)
 			x.ConnEvent.Data = "remote helper upload in progress"
 		default:
 			prefix = fmt.Sprintf("[INVALID=%d]", x.ConnEvent.Type)
-			color = ColorRed
+			color = frontend.ColorRed
 		}
 		data = x.ConnEvent.Data
 	}
 
 	host := x.Host.Name()
 	if color != 0 {
-		host = ColorString(isTTY, color, host)
+		host = frontend.ColorString(isTTY, color, host)
 	}
 	// Make sure to escape control characters to prevent terminal
 	// injection attacks
 	if !x.Escaped {
-		data = EscapeControlCharacters(isTTY, data)
+		data = frontend.EscapeControlCharacters(isTTY, data)
 	}
 	log.Printf("%-9s [%s] %s", prefix, host, data)
 }
@@ -319,25 +321,3 @@ func (s *Sync) logDebugf(format string, a ...interface{}) {
 func (s *Sync) logVerbosef(format string, a ...interface{}) {
 	s.log(safcm.LogVerbose, false, fmt.Sprintf(format, a...))
 }
-
-// sendRecv sends a message over conn and waits for the response. Any MsgLog
-// messages received before the final (non MsgLog) response are passed to
-// s.log.
-func (s *Sync) sendRecv(conn *rpc.Conn, msg safcm.Msg) (safcm.Msg, error) {
-	err := conn.Send(msg)
-	if err != nil {
-		return nil, err
-	}
-	for {
-		x, err := conn.Recv()
-		if err != nil {
-			return nil, err
-		}
-		log, ok := x.(safcm.MsgLog)
-		if ok {
-			s.log(log.Level, false, log.Text)
-			continue
-		}
-		return x, nil
-	}
-}