feat: add options to AddDir (#285)

Co-authored-by: Henrique Dias <[email protected]>

Author: acashmoney

add.go

@@ -97,7 +97,7 @@ func (s *Shell) AddLink(target string) (string, error) {
 }
 
 // AddDir adds a directory recursively with all of the files under it
-func (s *Shell) AddDir(dir string) (string, error) {
+func (s *Shell) AddDir(dir string, options ...AddOpts) (string, error) {
 	stat, err := os.Lstat(dir)
 	if err != nil {
 		return "", err
@@ -110,14 +110,18 @@ func (s *Shell) AddDir(dir string) (string, error) {
 	slf := files.NewSliceDirectory([]files.DirEntry{files.FileEntry(filepath.Base(dir), sf)})
 	reader := files.NewMultiFileReader(slf, true)
 
-	resp, err := s.Request("add").
-		Option("recursive", true).
-		Body(reader).
-		Send(context.Background())
+	rb := s.Request("add").Option("recursive", true)
+	for _, option := range options {
+		option(rb)
+	}
+
+	// Here we cannot use .Exec because "add" streams responses back for each file
+	// within the directory, and we only care about the last one, which is the directory
+	// itself.
+	resp, err := rb.Body(reader).Send(context.Background())
 	if err != nil {
 		return "", err
 	}
-
 	defer resp.Close()
 
 	if resp.Error != nil {

shell_test.go

@@ -125,6 +125,15 @@ func TestAddDir(t *testing.T) {
 	is.Equal(cid, "QmS4ustL54uo8FzR9455qaxZwuMiUhyvMcX9Ba8nUH4uVv")
 }
 
+func TestAddDirWithCidV1(t *testing.T) {
+	is := is.New(t)
+	s := NewShell(shellUrl)
+
+	cid, err := s.AddDir("./testdata", CidVersion(1))
+	is.Nil(err)
+	is.Equal(cid, "bafybeibgegl5yqme2jehvvneapbq7he5ahi3tmk4cpmlagrggeji6hzayq")
+}
+
 func TestAddDirOffline(t *testing.T) {
 	is := is.New(t)
 	s := NewShell("0.0.0.0:1234") // connect to an invalid address