wip: bundle.NewBatchFromRawContent

Author: marcelosousa

bundle/bundle.go

@@ -18,6 +18,7 @@ package bundle
 
 import (
 	"context"
+	"fmt"
 
 	"github.com/rs/zerolog"
 
@@ -134,6 +135,25 @@ func NewBatch(documents map[string]deepcode.BundleFile) *Batch {
 	}
 }
 
+func NewBatchFromRawContent(documents map[string][]byte) *Batch {
+	bundleFiles := make(map[string]deepcode.BundleFile)
+
+	for key, rawData := range documents {
+		bundleFile, err := deepcode.BundleFileFrom(rawData)
+		if err != nil {
+			// TODO: logging the error
+			fmt.Printf("error creating a bundle from the file")
+			continue
+		}
+
+		bundleFiles[key] = bundleFile
+	}
+
+	return &Batch{
+		documents: bundleFiles,
+	}
+}
+
 // todo simplify the size computation
 // maybe consider an addFile / canFitFile interface with proper error handling
 func (b *Batch) canFitFile(uri string, content []byte) bool {