feat: envd-sshd can read public key path from environment variable (#954

) add Signed-off-by: Jinjing.Zhou <allenzhou@tensorchord.ai> Signed-off-by: Jinjing.Zhou <allenzhou@tensorchord.ai>
tensorchord · Sep 30, 2022 · 006f653 · 006f653
1 parent 7c41dde
commit 006f653
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd/envd-sshd/main.go b/cmd/envd-sshd/main.go
@@ -56,6 +56,7 @@ func main() {
 			Name:    flagAuthKey,
 			Usage:   "path to authorized keys file, defaults to " + config.ContainerAuthorizedKeysPath,
 			Value:   config.ContainerAuthorizedKeysPath,
+			EnvVars: []string{"ENVD_AUTHORIZED_KEYS_PATH"},
 			Aliases: []string{"a"},
 		},
 		&cli.BoolFlag{

diff --git a/pkg/lang/ir/compile.go b/pkg/lang/ir/compile.go
@@ -211,7 +211,7 @@ func (g Graph) GetEntrypoint(buildContextDir string) ([]string, error) {
 	}
 
 	template := `set -euo pipefail
-/var/envd/bin/envd-sshd --authorized-keys %s --port %d --shell %s &
+/var/envd/bin/envd-sshd --port %d --shell %s &
 %s
 wait -n`
 
@@ -235,7 +235,6 @@ wait -n`
 	}
 
 	cmd := fmt.Sprintf(template,
-		config.ContainerAuthorizedKeysPath,
 		config.SSHPortInContainer, g.Shell, customCmd.String())
 	ep = append(ep, cmd)