Enable to set template file

Author: sachaos

cmd/prepare.go

@@ -19,6 +19,7 @@ var prepareCmd = &cobra.Command{
 		username := viper.GetString("username")
 		password := viper.GetString("password")
 		language := viper.GetString("language")
+		templatePath := viper.GetString("template")
 
 		contestId := args[0]
 
@@ -27,8 +28,6 @@ var prepareCmd = &cobra.Command{
 			return fmt.Errorf("invalid language")
 		}
 
-		fmt.Printf("Using template of %s\n", language)
-
 		ac, err := atcoder.NewAtCoder()
 		if err != nil {
 			return err
@@ -51,12 +50,15 @@ var prepareCmd = &cobra.Command{
 			selector = environment.DefaultEnvironmentSelector
 		}
 
-		return preparer.Prepare(contest, ".", selector.Select(language))
+		return preparer.Prepare(contest, ".", selector.Select(language), templatePath)
 	},
 }
 
 func init() {
 	rootCmd.AddCommand(prepareCmd)
 	prepareCmd.Flags().StringP("language", "l", "", "language")
 	viper.BindPFlag("language", prepareCmd.Flags().Lookup("language"))
+
+	prepareCmd.Flags().StringP("template", "t", "", "template file path")
+	viper.BindPFlag("template", prepareCmd.Flags().Lookup("template"))
 }

lib/preparer/prepare.go

@@ -65,8 +65,17 @@ func prepareTemplate(p string) (*template.Template, error) {
 	return template.New("src").Parse(string(all))
 }
 
-func Prepare(contest *atcoder.Contest, dir string, env *environment.Environment) error {
-	template, err := prepareTemplate(env.Template)
+func Prepare(contest *atcoder.Contest, dir string, env *environment.Environment, specifiedTemplatePath string) error {
+	var templatePath string
+	if specifiedTemplatePath == "" {
+		fmt.Printf("Using template of %s\n", env.Language)
+		templatePath = env.Template
+	} else {
+		fmt.Printf("Using local template: %s\n", specifiedTemplatePath)
+		templatePath = specifiedTemplatePath
+	}
+
+	template, err := prepareTemplate(templatePath)
 	if err != nil {
 		return err
 	}