set custom dockerfile path

Author: yoonsio

cmd/lk/agent.go

@@ -82,6 +82,13 @@ var (
 		Hidden:   true,
 	}
 
+	dockerFileFlag = &cli.StringFlag{
+		Name:     "dockerfile",
+		Usage:    "Path to the Dockerfile to use for the agent. If unset, will use the Dockerfile in the working directory.",
+		Required: false,
+		Aliases:  []string{"f"},
+	}
+
 	AgentCommands = []*cli.Command{
 		{
 			Name:    "agent",
@@ -98,6 +105,7 @@ var (
 						secretsFileFlag,
 						silentFlag,
 						regionFlag,
+						dockerFileFlag,
 					},
 					ArgsUsage: "[working-dir]",
 				},
@@ -119,6 +127,7 @@ var (
 					Flags: []cli.Flag{
 						secretsFlag,
 						secretsFileFlag,
+						dockerFileFlag,
 					},
 					ArgsUsage: "[working-dir]",
 				},
@@ -357,8 +366,11 @@ func createAgent(ctx context.Context, cmd *cli.Command) error {
 		return err
 	}
 
-	if err := requireDockerfile(ctx, cmd, workingDir, settingsMap); err != nil {
-		return err
+	dockerfile := cmd.String("dockerfile")
+	if dockerfile == "" {
+		if err := requireDockerfile(ctx, cmd, workingDir, settingsMap); err != nil {
+			return err
+		}
 	}
 
 	if err := agentfs.CheckSDKVersion(workingDir, settingsMap); err != nil {
@@ -381,6 +393,7 @@ func createAgent(ctx context.Context, cmd *cli.Command) error {
 	}
 
 	lkConfig.Agent.ID = resp.AgentId
+	lkConfig.Agent.Dockerfile = dockerfile
 	if err := lkConfig.SaveTOMLFile(workingDir, tomlFilename); err != nil {
 		return err
 	}
@@ -391,7 +404,7 @@ func createAgent(ctx context.Context, cmd *cli.Command) error {
 	}
 
 	fmt.Printf("Created agent with ID [%s]\n", util.Accented(resp.AgentId))
-	err = agentfs.Build(ctx, resp.AgentId, project)
+	err = agentfs.Build(ctx, resp.AgentId, project, dockerfile)
 	if err != nil {
 		return err
 	}
@@ -499,6 +512,8 @@ func deployAgent(ctx context.Context, cmd *cli.Command) error {
 		return err
 	}
 
+	dockerfile := cmd.String("dockerfile")
+
 	req = &lkproto.DeployAgentRequest{
 		AgentId: agentId,
 	}
@@ -532,7 +547,7 @@ func deployAgent(ctx context.Context, cmd *cli.Command) error {
 	}
 
 	fmt.Printf("Updated agent [%s]\n", util.Accented(resp.AgentId))
-	err = agentfs.Build(ctx, resp.AgentId, project)
+	err = agentfs.Build(ctx, resp.AgentId, project, dockerfile)
 	if err != nil {
 		return err
 	}

pkg/agentfs/build.go

@@ -35,7 +35,7 @@ import (
 	"github.com/livekit/protocol/logger"
 )
 
-func Build(ctx context.Context, id string, projectConfig *config.ProjectConfig) error {
+func Build(ctx context.Context, id string, projectConfig *config.ProjectConfig, dockerfile string) error {
 	baseUrl := projectConfig.URL
 	if strings.HasPrefix(projectConfig.URL, "ws") {
 		baseUrl = strings.Replace(projectConfig.URL, "ws", "http", 1)
@@ -58,6 +58,10 @@ func Build(ctx context.Context, id string, projectConfig *config.ProjectConfig)
 	params.Add("agent_id", id)
 	fullUrl := fmt.Sprintf("%s/build?%s", agentsUrl, params.Encode())
 
+	if dockerfile != "" {
+		params.Add("dockerfile", dockerfile)
+	}
+
 	at := auth.NewAccessToken(projectConfig.APIKey, projectConfig.APISecret)
 	at.SetAgentGrant(&auth.AgentGrant{
 		Admin: true,

pkg/config/livekit.go

@@ -52,8 +52,9 @@ type LiveKitTOMLProjectConfig struct {
 }
 
 type LiveKitTOMLAgentConfig struct {
-	ID      string   `toml:"id"`
-	Regions []string `toml:"regions"`
+	ID         string   `toml:"id"`
+	Regions    []string `toml:"regions"`
+	Dockerfile string   `toml:"dockerfile"`
 }
 
 func NewLiveKitTOML(forSubdomain string) *LiveKitTOML {