fix: pullAll

Author: dujardin09

client/cmd/gnit/pull.go

@@ -55,29 +55,31 @@ func (p *Pull) ExecuteAll() error {
 		return err
 	}
 
-	fmt.Println("Fetching list of files...")
+	fmt.Println("Pulling all files from repository...")
 
-	query := fmt.Sprintf("%s.Repository.ListFiles()", p.config.RealmPath)
-	result, err := p.client.QueryRaw(query)
+	packageAlias := config.PackageAlias(p.config.RealmPath)
+	query := fmt.Sprintf("%s.Repository.SerializePullAll()", packageAlias)
+
+	serializedData, err := p.client.RunQuery(p.config.RealmPath, query)
 	if err != nil {
 		if p.sourceMode {
 			fmt.Println("Repository not found or empty, trying to pull realm source files...")
 			return p.pullRealmSource()
 		}
-		return fmt.Errorf("failed to list files: %w", err)
+		return fmt.Errorf("failed to pull files: %w", err)
 	}
 
-	if len(result) == 0 {
+	if len(serializedData) == 0 {
 		fmt.Println("No files found in repository")
 		if p.sourceMode {
 			return p.pullRealmSource()
 		}
 		return nil
 	}
 
-	files, err := parseFileList(result)
+	files, err := parseSerializedFiles(string(serializedData))
 	if err != nil {
-		return fmt.Errorf("failed to parse file list: %w", err)
+		return fmt.Errorf("failed to parse files: %w", err)
 	}
 
 	if len(files) == 0 {
@@ -88,12 +90,13 @@ func (p *Pull) ExecuteAll() error {
 		return nil
 	}
 
-	fmt.Printf("Found %d file(s), pulling all...\n", len(files))
+	fmt.Printf("Found %d file(s), writing to disk...\n", len(files))
 
-	for _, filename := range files {
-		if err := p.Execute(filename); err != nil {
-			return fmt.Errorf("failed to pull '%s': %w", filename, err)
+	for filename, content := range files {
+		if err := filesystem.WriteFile(filename, content); err != nil {
+			return fmt.Errorf("failed to write '%s': %w", filename, err)
 		}
+		fmt.Printf("  pulled: %s (%d bytes)\n", filename, len(content))
 	}
 
 	fmt.Printf("\nSuccessfully pulled %d file(s)\n", len(files))
@@ -169,6 +172,55 @@ func parseRealmFileList(output string) []string {
 	return files
 }
 
+func parseSerializedFiles(data string) (map[string][]byte, error) {
+	lines := strings.Split(data, "\n")
+	files := make(map[string][]byte)
+
+	for _, line := range lines {
+		if line == "" {
+			continue
+		}
+
+		parts := strings.SplitN(line, "|", 2)
+		if len(parts) != 2 {
+			continue
+		}
+
+		filename := unescapeString(parts[0])
+		content := unescapeString(parts[1])
+		files[filename] = []byte(content)
+	}
+
+	return files, nil
+}
+
+func unescapeString(s string) string {
+	result := ""
+	i := 0
+	for i < len(s) {
+		if s[i] == '\\' && i+1 < len(s) {
+			next := s[i+1]
+			if next == '\\' {
+				result += "\\"
+				i += 2
+			} else if next == '|' {
+				result += "|"
+				i += 2
+			} else if next == 'n' {
+				result += "\n"
+				i += 2
+			} else {
+				result += string(s[i])
+				i++
+			}
+		} else {
+			result += string(s[i])
+			i++
+		}
+	}
+	return result
+}
+
 func parseFileList(data string) ([]string, error) {
 	str := strings.TrimSpace(data)
 

realms/src/api.gno

@@ -154,3 +154,57 @@ func (r *Repository) ListFiles() []string {
 
 	return files
 }
+
+func (r *Repository) PullAll() map[string][]byte {
+	headCommit := r.GetHeadCommit()
+	if headCommit == nil {
+		return map[string][]byte{}
+	}
+
+	treeValue, exists := r.objects.Get(headCommit.Tree)
+	if !exists {
+		return map[string][]byte{}
+	}
+
+	tree := treeValue.(map[string]string)
+	result := make(map[string][]byte)
+
+	for path, objectHash := range tree {
+		fileValue, exists := r.objects.Get(objectHash)
+		if exists {
+			result[path] = fileValue.([]byte)
+		}
+	}
+
+	return result
+}
+
+func (r *Repository) SerializePullAll() []byte {
+	files := r.PullAll()
+
+	var result string
+	for path, content := range files {
+		escapedPath := escapeString(path)
+		escapedContent := escapeString(string(content))
+		result += escapedPath + "|" + escapedContent + "\n"
+	}
+
+	return []byte(result)
+}
+
+func escapeString(s string) string {
+	result := ""
+	for i := 0; i < len(s); i++ {
+		c := s[i]
+		if c == '\\' {
+			result += "\\\\"
+		} else if c == '|' {
+			result += "\\|"
+		} else if c == '\n' {
+			result += "\\n"
+		} else {
+			result += string(c)
+		}
+	}
+	return result
+}