oneclickexport: add code host configs export functionality.

Author: sashaostrikov

cmd/frontend/oneclickexport/export.go

@@ -3,6 +3,7 @@ package oneclickexport
 import (
 	"archive/zip"
 	"bytes"
+	"context"
 	"io"
 	"io/ioutil"
 	"os"
@@ -14,7 +15,7 @@ import (
 type Exporter interface {
 	// Export function accepts an ExportRequest and returns bytes of a zip archive
 	// with requested data
-	Export(request *ExportRequest) ([]byte, error)
+	Export(ctx context.Context, request *ExportRequest) ([]byte, error)
 }
 
 var _ Exporter = &OneClickExporter{}
@@ -25,15 +26,16 @@ type OneClickExporter struct {
 }
 
 type ExportRequest struct {
-	IncludeSiteConfig bool `json:"includeSiteConfig"`
+	IncludeSiteConfig     bool `json:"includeSiteConfig"`
+	IncludeCodeHostConfig bool `json:"includeCodeHostConfig"`
 }
 
 // Export generates and returns a ZIP archive with the data, specified in request.
 // It works like this:
 // 1) tmp directory is created (exported files will end up in this directory and this directory is zipped in the end)
 // 2) ExportRequest is read and each corresponding processor is invoked
 // 3) Tmp directory is zipped after all the Processors finished their job
-func (e *OneClickExporter) Export(request *ExportRequest) ([]byte, error) {
+func (e *OneClickExporter) Export(ctx context.Context, request *ExportRequest) ([]byte, error) {
 	// 1) creating a tmp dir
 	dir, err := ioutil.TempDir(os.TempDir(), "export-*")
 	if err != nil {
@@ -43,7 +45,10 @@ func (e *OneClickExporter) Export(request *ExportRequest) ([]byte, error) {
 
 	// 2) tmp dir is passed to every processor
 	if request.IncludeSiteConfig {
-		e.cfgProcessors["siteConfig"].Process(&ConfigRequest{}, dir)
+		e.cfgProcessors["siteConfig"].Process(ctx, &ConfigRequest{}, dir)
+	}
+	if request.IncludeCodeHostConfig {
+		e.cfgProcessors["codeHostConfig"].Process(ctx, &ConfigRequest{}, dir)
 	}
 
 	// 3) after all request parts are processed, zip the tmp dir and return its bytes