server_sync_filesystem_stubs.go

package sync

import (
	"errors"
	"bufio"
	"strings"
	"runtime"
	"path"
	"path/filepath"
	"github.com/webdevops/go-shell"
	"github.com/webdevops/go-stubfilegenerator"
	"github.com/remeh/sizedwaitgroup"
)

// General sync
func (filesystem *Filesystem) SyncStubs() {
	switch filesystem.Connection.GetInstance().GetType() {
	case "local":
		fallthrough
	case "ssh":
		filesystem.generateStubs()
	case "docker":
		errors.New("Docker not supported")
	}
}

// Sync filesystem using rsync
func (filesystem *Filesystem) generateStubs() {
	connection := filesystem.Connection.GetInstance()

	cmd := shell.Cmd(connection.CommandBuilder("find", filesystem.Path, "-type", "f")...)
	output := cmd.Run().Stdout.String()

	removeBasePath := filesystem.Path
	localBasePath := filesystem.localPath()

	// build list and filter it by user filter list
	fileList := []string{}
	scanner := bufio.NewScanner(strings.NewReader(output))
	for scanner.Scan() {
		fileList = append(fileList, strings.TrimPrefix(scanner.Text(), removeBasePath))
	}
	fileList = filesystem.Filter.ApplyFilter(fileList)

	// generate stubs
	swg := sizedwaitgroup.New(runtime.GOMAXPROCS(0) * 10)
	stubGen := stubfilegenerator.NewStubGenerator()
	for _, filePath := range fileList {
		swg.Add()
		go func(filePath string, stubGen stubfilegenerator.StubGenerator) {
			defer swg.Done()
			localPath := path.Join(localBasePath, filePath)
			localAbsPath, _ := filepath.Abs(localPath)

			stubGen.TemplateVariables["PATH"] = localPath
			stubGen.Generate(localAbsPath)
		} (filePath, stubGen.Clone())
		swg.Wait()
	}
}