44 "context"
55 "fmt"
66 "os"
7+ "os/exec"
78 "os/signal"
9+ "runtime"
810 "syscall"
911 "time"
1012
@@ -16,31 +18,23 @@ import (
1618)
1719
1820var gatewayCmd = & cobra.Command {
19- Example : `infisical gateway` ,
20- Short : "Used to infisical gateway" ,
2121 Use : "gateway" ,
22+ Short : "Run the Infisical gateway or manage its systemd service" ,
23+ Long : "Run the Infisical gateway in the foreground or manage its systemd service installation. Use 'gateway install' to set up the systemd service." ,
24+ Example : `infisical gateway --token=<token>
25+ sudo infisical gateway install --token=<token> --domain=<domain>` ,
2226 DisableFlagsInUseLine : true ,
2327 Args : cobra .NoArgs ,
2428 Run : func (cmd * cobra.Command , args []string ) {
2529 token , err := util .GetInfisicalToken (cmd )
2630 if err != nil {
27- util .HandleError (err , "Unable to parse flag" )
31+ util .HandleError (err , "Unable to parse token flag" )
2832 }
2933
3034 if token == nil {
3135 util .HandleError (fmt .Errorf ("Token not found" ))
3236 }
3337
34- domain , err := cmd .Flags ().GetString ("domain" )
35- if err != nil {
36- util .HandleError (err , "Unable to parse domain flag" )
37- }
38-
39- // Try to install systemd service if possible
40- if err := gateway .InstallGatewaySystemdService (token .Token , domain ); err != nil {
41- log .Warn ().Msgf ("Failed to install systemd service: %v" , err )
42- }
43-
4438 Telemetry .CaptureEvent ("cli-command:gateway" , posthog .NewProperties ().Set ("version" , util .CLI_VERSION ))
4539
4640 sigCh := make (chan os.Signal , 1 )
@@ -110,6 +104,50 @@ var gatewayCmd = &cobra.Command{
110104 },
111105}
112106
107+ var gatewayInstallCmd = & cobra.Command {
108+ Use : "install" ,
109+ Short : "Install and enable systemd service for the gateway (requires sudo)" ,
110+ Long : "Install and enable systemd service for the gateway. Must be run with sudo on Linux." ,
111+ Example : "sudo infisical gateway install --token=<token> --domain=<domain>" ,
112+ DisableFlagsInUseLine : true ,
113+ Args : cobra .NoArgs ,
114+ Run : func (cmd * cobra.Command , args []string ) {
115+ if runtime .GOOS != "linux" {
116+ util .HandleError (fmt .Errorf ("systemd service installation is only supported on Linux" ))
117+ }
118+
119+ if os .Geteuid () != 0 {
120+ util .HandleError (fmt .Errorf ("systemd service installation requires root/sudo privileges" ))
121+ }
122+
123+ token , err := util .GetInfisicalToken (cmd )
124+ if err != nil {
125+ util .HandleError (err , "Unable to parse flag" )
126+ }
127+
128+ if token == nil {
129+ util .HandleError (fmt .Errorf ("Token not found" ))
130+ }
131+
132+ domain , err := cmd .Flags ().GetString ("domain" )
133+ if err != nil {
134+ util .HandleError (err , "Unable to parse domain flag" )
135+ }
136+
137+ if err := gateway .InstallGatewaySystemdService (token .Token , domain ); err != nil {
138+ util .HandleError (err , "Failed to install systemd service" )
139+ }
140+
141+ enableCmd := exec .Command ("systemctl" , "enable" , "infisical-gateway" )
142+ if err := enableCmd .Run (); err != nil {
143+ util .HandleError (err , "Failed to enable systemd service" )
144+ }
145+
146+ log .Info ().Msg ("Successfully installed and enabled infisical-gateway service" )
147+ log .Info ().Msg ("To start the service, run: sudo systemctl start infisical-gateway" )
148+ },
149+ }
150+
113151var gatewayRelayCmd = & cobra.Command {
114152 Example : `infisical gateway relay` ,
115153 Short : "Used to run infisical gateway relay" ,
@@ -139,9 +177,12 @@ var gatewayRelayCmd = &cobra.Command{
139177
140178func init () {
141179 gatewayCmd .Flags ().String ("token" , "" , "Connect with Infisical using machine identity access token" )
180+ gatewayInstallCmd .Flags ().String ("token" , "" , "Connect with Infisical using machine identity access token" )
181+ gatewayInstallCmd .Flags ().String ("domain" , "" , "Domain of your self-hosted Infisical instance" )
142182
143183 gatewayRelayCmd .Flags ().String ("config" , "" , "Relay config yaml file path" )
144184
185+ gatewayCmd .AddCommand (gatewayInstallCmd )
145186 gatewayCmd .AddCommand (gatewayRelayCmd )
146187 rootCmd .AddCommand (gatewayCmd )
147188}
0 commit comments