Merge pull request #905 from onflow/bastian/add-all-contracts-endpoint

Add endpoint to download ZIP file of all contracts

Author: turbolent

server/utils/emulator.go

@@ -19,13 +19,15 @@
 package utils
 
 import (
+	"archive/zip"
 	"encoding/json"
 	"net/http"
 	"slices"
 	"strconv"
 	"strings"
 
 	"github.com/gorilla/mux"
+	"github.com/onflow/cadence/common"
 	"github.com/onflow/cadence/runtime"
 	flowgo "github.com/onflow/flow-go/model/flow"
 
@@ -72,6 +74,8 @@ func NewEmulatorAPIServer(emulator emulator.Emulator, adapter *adapters.AccessAd
 
 	router.HandleFunc("/emulator/computationReport", r.ComputationReport).Methods("GET")
 
+	router.HandleFunc("/emulator/allContracts", r.AllContractsZip).Methods("GET")
+
 	return r
 }
 
@@ -327,3 +331,43 @@ func (m EmulatorAPIServer) Logs(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 }
+
+func (m EmulatorAPIServer) AllContractsZip(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Disposition", "attachment; filename=contracts.zip")
+	w.Header().Set("Content-Type", "application/zip")
+	w.WriteHeader(http.StatusOK)
+
+	zipW := zip.NewWriter(w)
+	defer func() {
+		err := zipW.Close()
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+		}
+	}()
+
+	for accountIndex := 1; ; accountIndex++ {
+		account, err := m.emulator.GetAccountByIndex(uint(accountIndex))
+		if err != nil {
+			break
+		}
+
+		for name, code := range account.Contracts {
+			location := common.AddressLocation{
+				Address: common.Address(account.Address),
+				Name:    name,
+			}
+
+			f, err := zipW.Create(location.ID() + cadenceFileSuffix)
+			if err != nil {
+				w.WriteHeader(http.StatusInternalServerError)
+				return
+			}
+
+			_, err = f.Write(code)
+			if err != nil {
+				w.WriteHeader(http.StatusInternalServerError)
+				return
+			}
+		}
+	}
+}