From ae12eb2a1551f28bed6139b79c6aaa08faf7b56d Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Wed, 2 Oct 2024 11:18:36 +0800
Subject: [PATCH] :sparkles: Process lifecycle management

---
 cli/bun.lockb                | Bin 11188 -> 13767 bytes
 cli/index.ts                 |   2 +
 cli/package.json             |   1 +
 cli/src/cmd/process-info.ts  | 146 +++++++++++++++++++++++++++++++++++
 pkg/navi/warden.go           |   5 +-
 pkg/sideload/applications.go |  51 +++++++++++-
 pkg/sideload/server.go       |   3 +
 pkg/warden/executor.go       |  21 +++--
 pkg/warden/manifest.go       |   5 ++
 9 files changed, 221 insertions(+), 13 deletions(-)
 create mode 100644 cli/src/cmd/process-info.ts

diff --git a/cli/bun.lockb b/cli/bun.lockb
index 58a91c9870d4017206d7a410021780462b8df85a..25871305ef1d43e2738dcb544926e65945a98319 100755
GIT binary patch
delta 3083
zcmc&$eNa<Z7QgQiNM5i~K1~84Aq0?OLPCV3eh@^&b+ug(g@RQF!-px+phUai4(Y0E
zE#n979os@t28H6%irUmEsHL{G6^qstTk(q>aKA*<P8oNr(*gIKm%LfK<L>N#Ju~;7
z`#bmCbIv{Yy?f7x1!pVj^@-=l*RjI`t_E3i??)@QzFrZj$vyJt;l5T`V}9b)Yd>r@
zrxKpDdkR$UFs&DO`$>en=3HOQg8}HJfR&7w_W{QFPQYQn4+0MaehTD4z>fh(ybn0e
z+g2AlDqt+&-7uaE@-4v8PBn0;8wTYf??g9nG~irY>awpXB^H0Lyv$ziD0LB%2@5!w
zcU4ywueCeZ>d*k-+5oR!pU<#FOvn_F3jzM;M%1A949rIWk46Jc60hMA;23}&ykLYw
z5?aLcSO!b>cvjVllIi79zn)JYA0BrwM=Ld1*WY&<KXlx;?Y~=6v~+OR=`5vYIQBHB
z=`LcLZe;(^TuEi@Ys^C0z*@Osa5W1oifIdL<`N)thyoBXHd^Rs<vs&h4#*f<>}O^|
z=runplS}0yEAtjDg#KIFAhL3CeuSiemOpI~nYjW$xQ6AewgbxIQL)I(jM4^wE0+W*
zMNj?|WZVWor~#`XiZg)H0f~6TLbB5UiBXfurq=?j%$rm$wlZJQLa~+Q0%)T+Jvb9`
zjRs*q1ct<)=CTR&Sur=sHv!w=g-JAr<AoXM0)!BxD2djy5>5r@lMhlqZ%}|R3<-}s
zfV`<04lw(Fhlmaa%w&@!^tw2mEt61%B%N)8g8}wHLK}h2;OKQ=n>ecA(%F+7E#hXl
z;V20<I7$)>4dV)dhJ+lFMZysPJ&m&<hmVM-|A(V}88lp{AcTenCc?K(NR&?&jQ|Ka
z`i?;W^dv_c8eaNSj`CP&XgdxX;&^CC$Pp(X@R4ut?w@P(d>9Hs6@i+cDF+il6POf2
zJQabBV)rBGqv{8jzSGaoN*^NAW}lPXF)lk}I6A-K#H5M2S6*?x{;>LkPRa1d8#k96
zIDVr{zNe<I`{w1Yz}CE=c}-`3i~(B=@!Ax=1vD-?@Vm{O`M(-?>6Gk^-|sYDjp|Mg
zGd&!0b^W-;(p2`3?6}8;8^v$er^(fuXOG|8QkegaBx~QYt~q4U+3dE22Y(I!nC=Ue
zxlg}eKTm2)SSV>fy!CMWaz*d?xC%vkV{qEk`P=SqySw+r$5(sBZ7DTl6UR60>gXOC
zv)(+@dgWZH;bZM+U3?YpRESQFVfjdOG4b8MkM<l*3R$KeKXhgP&gGv+Jj_fy_sfmb
zuCCszj|wtpb=|n{>Xa*9T`pO7q+gz%xa(5`=P10@m3w<e@QSxG=?h^pw{hy#3t#NC
zJQIDI8M!#u*!ElJXOlnJt$QKzH*0F@Jfpk9rMoYsbE5;_Z~N$W)1NeX7lK45U4y0-
zAr;)Wi;K73ej>U^AvEA3n>&1V%*~ED71BNBb8p;VP~CB2^ifMyU28cVX?gERjahy%
zV{iTmxpvAozZhzF25&1_`AL7B<Zj2s?kvyo`Ahyn$HQccI&S5xSkj-f>Yvlxi*svc
z6$Hv|eWlnhn|j-`_?@ps-yD58w9h<qQQ}5z)j3tw_?rt-@*+N|y<_>?cw2v|agH?n
ziRhH{QAhz3O>;vF7!~~k^lHk46)-Wh5PA(g27RoD3m;&ZIyxNDPAkKAc(zYz@%L1Q
z7ckP7wBFtQQA<tAboaz#FQE|sd2mGFHseXa_&?+QfTV(kI(Rq0<HVC1dX(Yy2`?-^
zj`4B0u6es?1Mdvf<1YtbJK_jEsGJjw{%{U`D4|8td&<ZcpM??LCKxYzRYSwCIush-
zlz7A7O@`kR++@6Gj2NFwyg4y~E>r?q5>F2(HNj{ORnQ4Ms3d}Jh_<DmQ9N=qg2pfm
zbci9Tp`jt<+kJOg?Q?W2Ivx{1r>N{Y;dvLHUlzB@QzrCS2pm`#(y@{@53!7yQq|Jb
zYLMf5PB!-(>djvNCCjWx)f@B%LbkvRzUZw>TRR&1ifk~GqJLfwPw8%|RLdiU8fD$P
zgJffZd^JDMuRKO`)pF((tpF{dzz_-m&@$^&(Eb2zR?Aa`N=YaHU_K4}8wn9g5usFo
z8KXgOf~B4G9@yvCQ+14j*+*x_$eD7wG{zJu6hu9X?Vqy~m+N8GbiEP%S?IwSllv+0
zO!vkklte<wB4UhsNCjkJz?UPTSoxnZTD;j43M!%Cfi)1iJ`e0YWy_dy*s@SmZMs{2
zVSC7^hJ_G}`jqDh8G`o-kMCgL+ttml^xIjc)&kKP4P-<@Cuw9+LjA=#XMK4mxA7z>
zr0P@B^wUuxLnD*+J_ox1k|fbpnhbhKV^AFN)ofcWsppwrPoA(Oc~z2A>6j)@YIE3i
zuHsdWQj=$1Y<xiSzj3ne37|9R(;yUAc}<4R>2R*EO7=3kP8+HGsR-BcDmp&3v?WeO
zcW9$jcmgK0V6W1ZZE!eVwU@Y7>uk=FQeCCfzLs7}h%!BW$^|C>%Z$!MzmJclGvgGD
Zg&#?^GJpQt;{E0T;?s`EQ<1oU{TEVF!-W6<

delta 1648
zcmcIkT}&KR6ux(0cbI{trPZ>-vXmm-Eg;j~MPQ|Lw;I#RkBMnDSQ4tFV%gHNh-h3N
z3Tf2PR+5tQ(k5+aZDRpRnsjaAUr9+5T*N-04~?R+zTtxpzDP|B^*b{=4{Bq4@g(Pd
z^PO|<x#!+H_k7X!QO9YsDqJ*pVb`z0<+rA%M?T9G{xP#JmAv!)&xdIC+v~0E^YfK8
z!V>>@$L8TGzfJ71K!}m#V7EnXW7NSnN*m|E*?$@QA>dhHKJXgs>ws5)%n8W;gT38H
z`ta@Wcko>ZdkSc+1Js9M=*qPnSAm=$+21piINBpjmu-I~ap*|TkPuN!(BL0R4RrS>
zlKti29`H?^2sq`o{l5V@&kogI=$EV=i&}9n9X=<Qn&YPmf{R|7k(X&y8IezrmJ`?R
zAfy>nRVvD9)cn{XGdDB2pB51DF>KGmCTR(_JZi~}%RO`g;}|VqTqMmE*L*obL=feo
zB3GmK3>eQ*tgPq3p0e0dZlnB(G<RIvg01sPE`m+#1LFwHMirle#lUhcb`PupOyx+~
zOf70$zDFl8-lPRJuH1K#o)=T1ZqoIb@|v4k0oUAg5+GIj60k=lJwK+5s?-X|sB|*F
zX&7fESaHB&Jrvs%MNk$-6wSA4$%9aVhcANVzaZx?hVnc^hz$Vi!&<n63GFP{KFsbi
zD2H!`vM@3)hdS~<mEUCLf6#vZR=*X8ef+h2|G$=>{@pX9lE>R7i)p{NgPz{xp&MWZ
zX+{U#13PN8NgsUzHd*XR`%4yO-UuEAzaHu?xsWdPUv%jM*%y4qPi2*{Flt!4ZwnXY
z%htp{Ka0aN*;&yt`mXXxjm>=Oc~raA*M#RRN}(LXo97>ZZ-{>kz74)OFO)a@FtnGA
zkQGiA@bFX);i)<;Y&}@Q={cB}VDx71@`VaI5D3s{pm(G5y5DcxF#RG;+*M>P`Ge2<
zk_hM5EsoDlW_AzGEAl|S88Smcq-ZQ?_?))#;i-%79`~%i41dJ-6U_w;d4-k{<us}O
zTb=EzQ-KtsLT1Fo8=a$d+YEV|#<v+hr`2`JbIZF|Ck7DJfLu7!v(yE@)B2vfI<fTV
z`nxOehwZg^ex1*0b+3N#gLoro^jiKpuC<GPgx~3KvwIV_l(lb8!~d9x3__SxTUDn_
zxoNG++u$^;ON(bx6NkS}DDqg;tT)3U;k35#?VV$R>pLHZCxZGSW^{*2?bY7Oe+E?J
z{8%O{Y+R|J(du1ji7b?<4h28Y_Q|)S^<T+13)j4l*v`@#8};Vbqp|WZ4FtT@SYxJ>
MH34_}SY3<qCq%I!+yDRo

diff --git a/cli/index.ts b/cli/index.ts
index 0d2b755..0ba722e 100644
--- a/cli/index.ts
+++ b/cli/index.ts
@@ -7,6 +7,7 @@ import { LogoutCommand } from "./src/cmd/logout.ts"
 import { ListServerCommand } from "./src/cmd/list.ts"
 import { StatusCommand } from "./src/cmd/status.ts"
 import { InfoCommand } from "./src/cmd/info.ts"
+import { ProcessCommand } from "./src/cmd/process-info.ts"
 
 const [node, app, ...args] = process.argv
 
@@ -29,4 +30,5 @@ cli.register(LogoutCommand)
 cli.register(ListServerCommand)
 cli.register(StatusCommand)
 cli.register(InfoCommand)
+cli.register(ProcessCommand)
 cli.runExit(args)
\ No newline at end of file
diff --git a/cli/package.json b/cli/package.json
index 4b95169..682ffcb 100644
--- a/cli/package.json
+++ b/cli/package.json
@@ -11,6 +11,7 @@
   },
   "dependencies": {
     "chalk": "^5.3.0",
+    "cli-table3": "^0.6.5",
     "clipanion": "^4.0.0-rc.4",
     "figlet": "^1.7.0",
     "ora": "^8.1.0"
diff --git a/cli/src/cmd/process-info.ts b/cli/src/cmd/process-info.ts
new file mode 100644
index 0000000..48cfa06
--- /dev/null
+++ b/cli/src/cmd/process-info.ts
@@ -0,0 +1,146 @@
+import { Command, Option, type Usage } from "clipanion"
+import { RsConfig } from "../utils/config.ts"
+import { createAuthHeader } from "../utils/auth.ts"
+import Table from "cli-table3"
+import chalk from "chalk"
+import ora from "ora"
+
+export class ProcessCommand extends Command {
+  static paths = [[`process`], [`ps`]]
+  static usage: Usage = {
+    category: `Networking`,
+    description: `Loading the application of RoadSign Server`,
+    details: `Fetching the configured things amount and other things of a connected server`,
+    examples: [
+      ["Fetch app directory from labeled server", `ps <label>`],
+      ["Fetch app logs from labeled server", `ps <label> <applicationId> logs`]
+    ]
+  }
+
+  label = Option.String({ required: true })
+  applicationId = Option.String({ required: false })
+  subcommand = Option.String({ required: false })
+  loop = Option.Boolean("--loop,--follow,-f", false, { description: "Keep updating the results" })
+
+  async execute() {
+    const config = await RsConfig.getInstance()
+
+    const server = config.config.servers.find(item => item.label === this.label)
+    if (server == null) {
+      this.context.stdout.write(chalk.red(`Server with label ${chalk.bold(this.label)} was not found.\n`))
+      return
+    }
+
+    const spinner = ora(`Fetching stats from server ${this.label}...`).start()
+    const prefStart = performance.now()
+
+
+    if (this.applicationId == null) {
+      try {
+        const res = await fetch(`${server.url}/cgi/applications`, {
+          headers: {
+            Authorization: createAuthHeader(server.credential)
+          }
+        })
+        if (res.status !== 200) {
+          throw new Error(await res.text())
+        }
+        const prefTook = performance.now() - prefStart
+        if (!this.loop) {
+          spinner.succeed(`Fetching completed in ${(prefTook / 1000).toFixed(2)}s 🎉`)
+        }
+
+        const table = new Table({
+          head: ["ID", "Status", "Command"],
+          colWidths: [20, 10, 48]
+        })
+
+        const statusMapping = ["Created", "Starting", "Started", "Exited", "Failed"]
+
+        const data = await res.json()
+        for (const app of data) {
+          table.push([app["id"], statusMapping[app["status"]], app["command"].join(" ")])
+        }
+
+        this.context.stdout.write(table.toString())
+      } catch (e) {
+        spinner.fail(`Server with label ${chalk.bold(this.label)} is not running! 😢`)
+        return
+      }
+    } else {
+      switch (this.subcommand) {
+        case "logs":
+          while (true) {
+            try {
+              const res = await fetch(`${server.url}/cgi/applications/${this.applicationId}/logs`, {
+                headers: {
+                  Authorization: createAuthHeader(server.credential)
+                }
+              })
+              if (res.status === 404) {
+                spinner.fail(`App with id ${chalk.bold(this.applicationId)} was not found! 😢`)
+                return
+              }
+              if (res.status !== 200) {
+                throw new Error(await res.text())
+              }
+              const prefTook = performance.now() - prefStart
+              if (!this.loop) {
+                spinner.succeed(`Fetching completed in ${(prefTook / 1000).toFixed(2)}s 🎉`)
+              }
+
+              this.context.stdout.write(await res.text())
+            } catch (e) {
+              spinner.fail(`Server with label ${chalk.bold(this.label)} is not running! 😢`)
+              return
+            }
+
+            if (!this.loop) {
+              break
+            } else {
+              spinner.text = "Updating..."
+              await new Promise(resolve => setTimeout(resolve, 3000))
+              this.context.stdout.write("\x1Bc")
+            }
+          }
+          break
+        case "start":
+        case "stop":
+        case "restart":
+          try {
+            const res = await fetch(`${server.url}/cgi/applications/${this.applicationId}/${this.subcommand}`, {
+              method: "POST",
+              headers: {
+                Authorization: createAuthHeader(server.credential)
+              }
+            })
+            if (res.status === 404) {
+              spinner.fail(`App with id ${chalk.bold(this.applicationId)} was not found! 😢`)
+              return
+            }
+            if (res.status === 500) {
+              this.context.stdout.write(chalk.red(`Server failed to perform action for application: ${await res.text()}\n`))
+              spinner.fail(`Failed to perform action ${chalk.bold(this.applicationId)}... 😢`)
+              return
+            }
+            if (res.status !== 200) {
+              throw new Error(await res.text())
+            }
+            const prefTook = performance.now() - prefStart
+            if (!this.loop) {
+              spinner.succeed(`Fetching completed in ${(prefTook / 1000).toFixed(2)}s 🎉`)
+            }
+          } catch (e) {
+            spinner.fail(`Server with label ${chalk.bold(this.label)} is not running! 😢`)
+            return
+          }
+          spinner.succeed(`Action for application ${chalk.bold(this.applicationId)} has been performed. 🎉`)
+          break
+        default:
+          this.context.stdout.write(chalk.red(`Subcommand ${chalk.bold(this.subcommand)} was not found.\n`))
+      }
+    }
+
+    process.exit(0)
+  }
+}
\ No newline at end of file
diff --git a/pkg/navi/warden.go b/pkg/navi/warden.go
index d3751f4..de7cc6c 100644
--- a/pkg/navi/warden.go
+++ b/pkg/navi/warden.go
@@ -15,8 +15,5 @@ func InitializeWarden(regions []*Region) {
 
 	// Hot swap
 	warden.InstancePool = pool
-
-	for _, instance := range warden.InstancePool {
-		instance.Wake()
-	}
+	warden.StartPool()
 }
diff --git a/pkg/sideload/applications.go b/pkg/sideload/applications.go
index 34335ce..306824a 100644
--- a/pkg/sideload/applications.go
+++ b/pkg/sideload/applications.go
@@ -8,8 +8,13 @@ import (
 )
 
 func getApplications(c *fiber.Ctx) error {
-	applications := lo.FlatMap(navi.R.Regions, func(item *navi.Region, idx int) []warden.Application {
-		return item.Applications
+	applications := lo.FlatMap(navi.R.Regions, func(item *navi.Region, idx int) []warden.ApplicationInfo {
+		return lo.Map(item.Applications, func(item warden.Application, index int) warden.ApplicationInfo {
+			return warden.ApplicationInfo{
+				Application: item,
+				Status:      warden.GetFromPool(item.ID).Status,
+			}
+		})
 	})
 
 	return c.JSON(applications)
@@ -24,3 +29,45 @@ func getApplicationLogs(c *fiber.Ctx) error {
 		return c.SendString(instance.Logs())
 	}
 }
+
+func letApplicationStart(c *fiber.Ctx) error {
+	if instance, ok := lo.Find(warden.InstancePool, func(item *warden.AppInstance) bool {
+		return item.Manifest.ID == c.Params("id")
+	}); !ok {
+		return fiber.NewError(fiber.StatusNotFound)
+	} else {
+		if err := instance.Wake(); err != nil {
+			return fiber.NewError(fiber.StatusInternalServerError, err.Error())
+		}
+		return c.SendStatus(fiber.StatusOK)
+	}
+}
+
+func letApplicationStop(c *fiber.Ctx) error {
+	if instance, ok := lo.Find(warden.InstancePool, func(item *warden.AppInstance) bool {
+		return item.Manifest.ID == c.Params("id")
+	}); !ok {
+		return fiber.NewError(fiber.StatusNotFound)
+	} else {
+		if err := instance.Stop(); err != nil {
+			return fiber.NewError(fiber.StatusInternalServerError, err.Error())
+		}
+		return c.SendStatus(fiber.StatusOK)
+	}
+}
+
+func letApplicationRestart(c *fiber.Ctx) error {
+	if instance, ok := lo.Find(warden.InstancePool, func(item *warden.AppInstance) bool {
+		return item.Manifest.ID == c.Params("id")
+	}); !ok {
+		return fiber.NewError(fiber.StatusNotFound)
+	} else {
+		if err := instance.Stop(); err != nil {
+			return fiber.NewError(fiber.StatusInternalServerError, err.Error())
+		}
+		if err := instance.Start(); err != nil {
+			return fiber.NewError(fiber.StatusInternalServerError, err.Error())
+		}
+		return c.SendStatus(fiber.StatusOK)
+	}
+}
diff --git a/pkg/sideload/server.go b/pkg/sideload/server.go
index ba2d840..8b5e31e 100644
--- a/pkg/sideload/server.go
+++ b/pkg/sideload/server.go
@@ -49,6 +49,9 @@ func InitSideload() *fiber.App {
 		cgi.Get("/regions/cfg/:id", getRegionConfig)
 		cgi.Get("/applications", getApplications)
 		cgi.Get("/applications/:id/logs", getApplicationLogs)
+		cgi.Post("/applications/:id/start", letApplicationStart)
+		cgi.Post("/applications/:id/stop", letApplicationStop)
+		cgi.Post("/applications/:id/restart", letApplicationRestart)
 
 		cgi.Post("/reload", doReload)
 	}
diff --git a/pkg/warden/executor.go b/pkg/warden/executor.go
index b49de91..ef5cf0f 100644
--- a/pkg/warden/executor.go
+++ b/pkg/warden/executor.go
@@ -2,10 +2,11 @@ package warden
 
 import (
 	"fmt"
-	"os"
+	"github.com/rs/zerolog/log"
 	"os/exec"
 	"path/filepath"
 	"strings"
+	"syscall"
 	"time"
 
 	"github.com/samber/lo"
@@ -87,15 +88,16 @@ func (v *AppInstance) Start() error {
 	// Monitor
 	go func() {
 		for {
-			if v.Cmd.Process == nil || v.Cmd.ProcessState == nil {
+			if v.Cmd != nil && v.Cmd.Process == nil {
 				v.Status = AppStarting
-			} else if !v.Cmd.ProcessState.Exited() {
+			} else if v.Cmd != nil && v.Cmd.ProcessState == nil {
 				v.Status = AppStarted
 			} else {
-				v.Status = lo.Ternary(v.Cmd.ProcessState.Success(), AppExited, AppFailure)
+				v.Status = lo.Ternary(v.Cmd == nil, AppExited, AppFailure)
+				v.Cmd = nil
 				return
 			}
-			time.Sleep(100 * time.Millisecond)
+			time.Sleep(1000 * time.Millisecond)
 		}
 	}()
 
@@ -104,8 +106,13 @@ func (v *AppInstance) Start() error {
 
 func (v *AppInstance) Stop() error {
 	if v.Cmd != nil && v.Cmd.Process != nil {
-		if err := v.Cmd.Process.Signal(os.Interrupt); err != nil {
-			v.Cmd.Process.Kill()
+		if err := v.Cmd.Process.Signal(syscall.SIGTERM); err != nil {
+			log.Warn().Int("pid", v.Cmd.Process.Pid).Err(err).Msgf("Failed to send SIGTERM to process...")
+			if err = v.Cmd.Process.Kill(); err != nil {
+				log.Error().Int("pid", v.Cmd.Process.Pid).Err(err).Msgf("Failed to kill process...")
+			} else {
+				v.Cmd = nil
+			}
 			return err
 		} else {
 			v.Cmd = nil
diff --git a/pkg/warden/manifest.go b/pkg/warden/manifest.go
index 2f3e1a7..58e794b 100644
--- a/pkg/warden/manifest.go
+++ b/pkg/warden/manifest.go
@@ -6,3 +6,8 @@ type Application struct {
 	Command     []string `json:"command" toml:"command"`
 	Environment []string `json:"environment" toml:"environment"`
 }
+
+type ApplicationInfo struct {
+	Application
+	Status AppStatus `json:"status"`
+}