Add leetgo extract

README.md

@@ -93,6 +93,7 @@ Available Commands:
   test                    Run question test cases
   submit                  Submit solution
   edit                    Open solution in editor
+  extract                 Extract solution code from generated file
   contest                 Generate contest questions
   cache                   Manage local questions cache
   config                  Show configurations
@@ -160,10 +161,11 @@ code:
     # Replace some blocks of the generated code
     blocks:
       - name: beforeMarker
-        template: |
+        template: |+
           package main
 
           {{ if .NeedsDefinition -}} import . "github.com/j178/leetgo/testutils/go" {{- end }}
+
     # Functions that modify the generated code
     modifiers:
       - name: removeUselessComments

README_en.md

@@ -94,6 +94,7 @@ Available Commands:
   test                    Run question test cases
   submit                  Submit solution
   edit                    Open solution in editor
+  extract                 Extract solution code from generated file
   contest                 Generate contest questions
   cache                   Manage local questions cache
   config                  Show configurations
@@ -162,10 +163,11 @@ code:
     # Replace some blocks of the generated code
     blocks:
       - name: beforeMarker
-        template: |
+        template: |+
           package main
 
           {{ if .NeedsDefinition -}} import . "github.com/j178/leetgo/testutils/go" {{- end }}
+
     # Functions that modify the generated code
     modifiers:
       - name: removeUselessComments

cmd/edit.go

@@ -37,3 +37,25 @@ var editCmd = &cobra.Command{
 		return editor.Open(result.Files)
 	},
 }
+
+var extractCmd = &cobra.Command{
+	Use:   "extract qid",
+	Short: "Extract solution code from generated file",
+	Args:  cobra.ExactArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		c := leetcode.NewClient()
+		qs, err := leetcode.ParseQID(args[0], c)
+		if err != nil {
+			return err
+		}
+		if len(qs) > 1 {
+			return fmt.Errorf("multiple questions found")
+		}
+		code, err := lang.GetSolutionCode(qs[0])
+		if err != nil {
+			return err
+		}
+		cmd.Println(code)
+		return nil
+	},
+}

cmd/root.go

@@ -108,6 +108,7 @@ func initCommands() {
 		testCmd,
 		submitCmd,
 		editCmd,
+		extractCmd,
 		contestCmd,
 		cacheCmd,
 		configCmd,

config/config.go

@@ -8,10 +8,11 @@ import (
 	"strings"
 
 	"github.com/hashicorp/go-hclog"
-	"github.com/j178/leetgo/utils"
 	"github.com/mitchellh/go-homedir"
 	"github.com/spf13/viper"
 	"gopkg.in/yaml.v3"
+
+	"github.com/j178/leetgo/utils"
 )
 
 const (
@@ -191,11 +192,12 @@ func Default() *Config {
 					Blocks: []Block{
 						{
 							Name: "beforeMarker", Template: fmt.Sprintf(
-								`package main
+							`package main
 
 {{ if .NeedsDefinition -}} import . "%s" {{- end }}
+
 `, GoTestUtilsModPath,
-							),
+						),
 						},
 					},
 					Modifiers: []Modifier{