define resetHead function (#128)

cmd/reset.go

@@ -23,6 +23,40 @@ var (
 	resetRegexp = regexp.MustCompile(`HEAD@\{\d\}`)
 )
 
+func resetHead(arg, rootGoitPath string, head *store.Head, refs *store.Refs, conf *store.Config) error {
+	// get log record
+	reflog, err := store.NewReflog(rootGoitPath, head, refs)
+	if err != nil {
+		return fmt.Errorf("fail to initialize reflog: %w", err)
+	}
+	sp := strings.Split(arg, "HEAD@")[1]
+	headNum, err := strconv.Atoi(sp[1 : len(sp)-1])
+	if err != nil {
+		return fmt.Errorf("fail to convert number '%s': %w", arg, err)
+	}
+	logRecord, err := reflog.GetRecord(headNum)
+	if err != nil {
+		return fmt.Errorf("fail to get log record: %w", err)
+	}
+
+	// reset Head
+	prevHeadHash := head.Commit.Hash
+	if err := head.Reset(rootGoitPath, refs, logRecord.Hash); err != nil {
+		return fmt.Errorf("fail to reset HEAD: %w", err)
+	}
+
+	// log
+	newRecord := log.NewRecord(log.ResetRecord, prevHeadHash, logRecord.Hash, conf.GetUserName(), conf.GetEmail(), time.Now(), fmt.Sprintf("moving to %s", arg))
+	if err := gLogger.WriteHEAD(newRecord); err != nil {
+		return fmt.Errorf("log error: %w", err)
+	}
+	if err := gLogger.WriteBranch(newRecord, head.Reference); err != nil {
+		return fmt.Errorf("log error: %w", err)
+	}
+
+	return nil
+}
+
 // resetCmd represents the reset command
 var resetCmd = &cobra.Command{
 	Use:   "reset",
@@ -50,36 +84,10 @@ var resetCmd = &cobra.Command{
 			return errors.New("only one argument is acceptible. argument format is 'HEAD@{number}'")
 		}
 
-		if isSoft {
-			// get log record
-			reflog, err := store.NewReflog(client.RootGoitPath, client.Head, client.Refs)
-			if err != nil {
-				return fmt.Errorf("fail to initialize reflog: %w", err)
-			}
-			sp := strings.Split(args[0], "HEAD@")[1]
-			headNum, err := strconv.Atoi(sp[1 : len(sp)-1])
-			if err != nil {
-				return fmt.Errorf("fail to convert number '%s': %w", args[0], err)
-			}
-			logRecord, err := reflog.GetRecord(headNum)
-			if err != nil {
-				return fmt.Errorf("fail to get log record: %w", err)
-			}
-
-			// reset Head
-			prevHeadHash := client.Head.Commit.Hash
-			if err := client.Head.Reset(client.RootGoitPath, client.Refs, logRecord.Hash); err != nil {
+		if isSoft || isMixed || isHard {
+			if err := resetHead(args[0], client.RootGoitPath, client.Head, client.Refs, client.Conf); err != nil {
 				return fmt.Errorf("fail to reset HEAD: %w", err)
 			}
-
-			// log
-			newRecord := log.NewRecord(log.ResetRecord, prevHeadHash, logRecord.Hash, client.Conf.GetUserName(), client.Conf.GetEmail(), time.Now(), fmt.Sprintf("moving to %s", args[0]))
-			if err := gLogger.WriteHEAD(newRecord); err != nil {
-				return fmt.Errorf("log error: %w", err)
-			}
-			if err := gLogger.WriteBranch(newRecord, client.Head.Reference); err != nil {
-				return fmt.Errorf("log error: %w", err)
-			}
 		}
 
 		return nil