diff --git a/controller/copy_build.go b/controller/copy_build.go index 399dd68..20b090e 100644 --- a/controller/copy_build.go +++ b/controller/copy_build.go @@ -13,7 +13,7 @@ import ( socketio "github.com/googollee/go-socket.io" ) -func copyBuild(server *socketio.Server, plant string) { +func copyBuild(server *socketio.Server, plant string, drive string) { // Load from environment in real-life code! user := os.Getenv("ADM_USER") pass := os.Getenv("ADM_PASS") @@ -21,7 +21,7 @@ func copyBuild(server *socketio.Server, plant string) { // latest build latestbuild := lastestBuild() src := latestbuild - plantServer := fmt.Sprintf("\\\\%v\\e$\\lst", plant) + plantServer := fmt.Sprintf("\\\\%v\\%v$\\lst", plant, drive) // Build PowerShell psScript := fmt.Sprintf(` diff --git a/controller/update_channel.go b/controller/update_channel.go index d4a055e..42df928 100644 --- a/controller/update_channel.go +++ b/controller/update_channel.go @@ -17,6 +17,7 @@ import ( type UpdatePayload struct { Action string `json:"action"` Target string `json:"target"` + Drive string `json:"drive"` } func registerUpdateChannel(server *socketio.Server) { @@ -31,7 +32,7 @@ func registerUpdateChannel(server *socketio.Server) { server.OnEvent("/", "update", func(s socketio.Conn, payload UpdatePayload) { switch strings.ToLower(payload.Action) { case "copy": - copyLatestBuild(server, payload.Target) + copyLatestBuild(server, payload.Target, payload.Drive) case "update": updateServer(server, payload.Target) @@ -80,10 +81,10 @@ func updateServer(server *socketio.Server, target string) { } } -func copyLatestBuild(server *socketio.Server, target string) { +func copyLatestBuild(server *socketio.Server, target string, drive string) { server.BroadcastToRoom("/", "update", "updateLogs", fmt.Sprintf("🚀 Copying latest build to %v", target)) - copyBuild(server, target) + copyBuild(server, target, drive) } func triggerRemoteUpdate(server *socketio.Server, remoteURL string, payload UpdatePayload) {