Save restore_token per-device

Author: omarroth

cmd/doubletake/main.go

@@ -295,13 +295,21 @@ func main() {
 			log.Fatalf("test capture failed: %v", err)
 		}
 	} else {
+		restoreToken := ""
+		if creds := credStore.Lookup(info.DeviceID); creds != nil {
+			restoreToken = creds.RestoreToken
+		}
 		captureCfg := airplay.CaptureConfig{
 			FPS:           *fps,
 			Bitrate:       *bitrate,
 			HWAccel:       *hwaccel,
 			X11WindowID:   xid,
 			X11WindowName: *x11WindowName,
 			ShowCursor:    !*noCursor,
+			RestoreToken:  restoreToken,
+			SaveRestoreToken: func(token string) error {
+				return credStore.SaveRestoreToken(info.DeviceID, token)
+			},
 		}
 		var err error
 		capture, err = airplay.StartCapture(ctx, captureCfg)