Ignore question if it is available only for premium plan

client.go

@@ -11,6 +11,8 @@ import (
 	"time"
 )
 
+var PaidOnlyError = fmt.Errorf("not support premium question for now")
+
 type LeetCode struct {
 	BaseURL        string
 	session, token string
@@ -71,6 +73,9 @@ func (lc *LeetCode) GetQuestionByFrontendID(ctx context.Context, id int) (*Quest
 	if pair == nil {
 		return nil, fmt.Errorf("cannot find problem")
 	}
+	if pair.PaidOnly {
+		return nil, PaidOnlyError
+	}
 
 	q, err := lc.GetQuestion(ctx, pair.Stat)
 	if err != nil {

core.go

@@ -33,8 +33,8 @@ var CmdMap = map[CmdName]Cmd{
 	HELP:     &HelpCmd{},
 }
 
-func buildPath(id, slug string) string {
+func buildPath(id int, slug string) string {
 	// TODO: changeable directory
-	format := "%s.%s.go"
+	format := "%d.%s.go"
 	return fmt.Sprintf(format, id, slug)
 }

exec.go

@@ -43,7 +43,7 @@ func (c *ExecCmd) Run(ctx context.Context, out io.Writer, args []string) error {
 	if err != nil {
 		return err
 	}
-	fp := buildPath(q.QuestionID, q.Slug)
+	fp := buildPath(q.FrontendQuestionID, q.Slug)
 	code, err := ioutil.ReadFile(fp)
 	if err != nil {
 		return err

generate.go

@@ -94,7 +94,7 @@ func (g *GenerateCmd) Run(ctx context.Context, out io.Writer, args []string) err
 	log.Printf("%s", buf.String())
 
 	// TODO: どうやってファイル保存とテストしやすさを分けようか？
-	path := buildPath(q.QuestionID, q.Slug)
+	path := buildPath(q.FrontendQuestionID, q.Slug)
 	fmt.Fprintf(out, "save at %q\n", path)
 	if err := ioutil.WriteFile(path, buf.Bytes(), 0644); err != nil {
 		return err

generate_test.go

@@ -12,7 +12,7 @@ func TestGenerateCmd(t *testing.T) {
 		name string
 		args []string
 	}{
-		{name: "SuccessGenerateFiles", args: []string{"1"}},
+		{name: "SuccessGenerateFiles", args: []string{"2"}},
 	}
 	for _, tt := range tests {
 		tt := tt
@@ -24,3 +24,22 @@ func TestGenerateCmd(t *testing.T) {
 		})
 	}
 }
+
+func TestGenerateCmd_Error(t *testing.T) {
+	tests := [...]struct {
+		name string
+		args []string
+		want error
+	}{
+		{name: "FailedPaidOnlyProblem", args: []string{"1602"}, want: PaidOnlyError},
+	}
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			cmd := &GenerateCmd{}
+			if err := cmd.Run(context.TODO(), ioutil.Discard, tt.args); err != tt.want {
+				t.Errorf("GenerateCmd() error = %v", err)
+			}
+		})
+	}
+}

test.go

@@ -42,7 +42,7 @@ func (c *TestCmd) Run(ctx context.Context, out io.Writer, args []string) error {
 	if err != nil {
 		return err
 	}
-	fp := buildPath(q.QuestionID, q.Slug)
+	fp := buildPath(q.FrontendQuestionID, q.Slug)
 	code, err := ioutil.ReadFile(fp)
 	if err != nil {
 		return err