add reset hard (#128)

cmd/reset.go

@@ -12,6 +12,7 @@ import (
 	"time"
 
 	"github.com/JunNishimura/Goit/internal/log"
+	"github.com/JunNishimura/Goit/internal/object"
 	"github.com/JunNishimura/Goit/internal/store"
 	"github.com/spf13/cobra"
 )
@@ -51,6 +52,20 @@ func resetIndex(rootGoitPath string, logRecord *store.LogRecord, index *store.In
 	return nil
 }
 
+func resetWorkingTree(rootGoitPath string, index *store.Index) error {
+	for _, entry := range index.Entries {
+		obj, err := object.GetObject(rootGoitPath, entry.Hash)
+		if err != nil {
+			fmt.Errorf("fail to get object: %w", err)
+		}
+		if err := obj.ReflectToWorkingTree(rootGoitPath, string(entry.Path)); err != nil {
+			fmt.Errorf("fail to reflect %s to working directory: %w", string(entry.Path), err)
+		}
+	}
+
+	return nil
+}
+
 // resetCmd represents the reset command
 var resetCmd = &cobra.Command{
 	Use:   "reset",
@@ -107,6 +122,13 @@ var resetCmd = &cobra.Command{
 			}
 		}
 
+		// reset working tree
+		if isHard {
+			if err := resetWorkingTree(client.RootGoitPath, client.Idx); err != nil {
+				return fmt.Errorf("fail to reset working tree: %w", err)
+			}
+		}
+
 		return nil
 	},
 }