writer.go

package dbexport

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"strings"
)

func replaceNewLine(content string) string {
	strings.ReplaceAll(content, "^M", "")
	lines := strings.Split(content, "\r\n")
	return strings.Join(lines, "\n")
}

func SaveDbObjects(dbObjects []DbObject) []string {
	var savedFiles []string

	for i := range dbObjects {
		dbObject := dbObjects[i]
		filePath := makeDbObjectPath(dbObject)
		fmt.Println("Salvando", dbObject.Type, ":", dbObject.Name, "->", i, "de", len(dbObjects))
		dbObject.Content = GetSqlForObject(dbObject.Type, dbObject.Name)
		if WriteSqlToFile(filePath, dbObject.Content) {
			savedFiles = append(savedFiles, filePath)
		}
	}

	return savedFiles
}

func makeDbObjectPath(dbObject DbObject) string {
	dir, err := os.Getwd()
	if err != nil {
		log.Fatal(err)
	}
	return dir + "/export/" + dbObject.Type + "/" + dbObject.Name + ".sql"
}

func WriteSqlToFile(filePath, sql string) bool {
	content := replaceNewLine(sql)
	byteContent := []byte(content)
	path := filepath.Dir(filePath)

	if _, err := os.Stat(path); os.IsNotExist(err) {
		os.MkdirAll(path, os.ModePerm)
	}

	err := ioutil.WriteFile(filePath, byteContent, os.ModePerm)

	if err != nil {
		log.Fatal(err)
		return false
	}

	return true
}