kern/v4/d1/mock_env/mock_env.go

// package mock_env -- обеспечивает мок-окружение для тестов.
package mock_env

import (
	"os"
	"strings"

	"gitp78su.ipnodns.ru/svi/kern/v4/d0"
	"gitp78su.ipnodns.ru/svi/kern/v4/d0/helpers"
)

type IMockEnv interface {
	// Pwd -- возвращает текущий рабочий каталог процесса
	Pwd() *d0.LTxt
}

// MockEnv -- объект мок-окружения.
type MockEnv struct {
	owner    *d0.LOwner          // Владелец мок-окружения
	dictEnv  map[d0.Txt]*d0.LTxt // Словарь переменных окружения
	pwd      *d0.LTxt            // Текущий путь процесса
	fileName *d0.LTxt            // Полный путь к .env файлу
	strEnv   *d0.LTxt            // Фактическое содержимое .env файла
}

// MakeEnv -- возвращает новое мок-окружение.
func MakeEnv(owner *d0.LOwner) *MockEnv {
	sf := &MockEnv{
		owner:   owner,
		dictEnv: map[d0.Txt]*d0.LTxt{},
	}
	sf.setPath()
	sf.readEnv()
	sf.parseEnv()
	sf.SetAll()
	return sf
}

// Парсит полученный текст .env для установки переменных окружения.
func (sf *MockEnv) parseEnv() {
	lstStr := strings.SplitSeq(sf.strEnv.String(), "\n")
	for val := range lstStr {
		pair := strings.TrimSuffix(val, "\r")
		if pair == "" {
			continue
		}
		if strings.HasPrefix(val, "#") {
			continue
		}
		sf.parseEnvStr(d0.LetTxt(pair))
	}
}

// Парсит строку окружения.
func (sf *MockEnv) parseEnvStr(pair *d0.LTxt) {
	lstPair := strings.Split(pair.String(), "=")
	d0.If(len(lstPair) < 2).Hassert("MockEnv.parseEnvStr(): pair(%v) not have =", []any{pair})
	key := lstPair[0]
	d0.If(key == "").Hassert("MockEnv.parseEnvStr(): in pair(%v) key is empty", []any{pair})
	lstVal := lstPair[1:]

	val := ""
	for _, val_ := range lstVal {
		val += val_ + "="
	}
	val = strings.TrimSuffix(val, "=")

	d0.If(val == "").Hassert("MockEnv.parseEnvStr(): in pair(%v) val is empty", []any{pair})
	sf.dictEnv[d0.Txt(key)] = d0.LetTxt(val)
}

// Читает файл переменных окружения.
func (sf *MockEnv) readEnv() {
	binEnv, err := os.ReadFile(sf.fileName.String())
	d0.If(err != nil).
		Hassert("MockEnv.readEnv(): in read %v, err=\n\t%v", []any{sf.fileName,
			err})
	sf.strEnv = d0.LetTxt(d0.Txt(binEnv))
}

// Устанавливает текущий путь относительно проекта в dev.
func (sf *MockEnv) setPath() {
	dir, err := os.Getwd()
	helpers.If(err != nil).Hassert("MockEnv.setPath(): in get PWD, err=\n\t%v", []any{err})
	lstDir := strings.Split(dir, "/")
	var pwd strings.Builder
	fileName := ""
	isFind := d0.Bool(false)
	for _, point := range lstDir {
		_, _ = pwd.WriteString(point + "/")
		fileName = pwd.String() + ".env"
		_, err := os.ReadFile(fileName)
		if err == nil {
			isFind = true
			break
		}
	}
	helpers.If(!isFind).Hassert("MockEnv.setPath(): not find file .env", []any{})
	sf.pwd = d0.LetTxt(pwd.String() + "/bin_dev/" + string(sf.owner.Name()))
	err = os.MkdirAll(sf.pwd.String(), 0750)
	helpers.If(err != nil).Hassert("MockEnv.setPath(): in create PWD, err=\n\t%v", []any{err})
	err = os.Chdir(sf.pwd.String())
	helpers.If(err != nil).Hassert("MockEnv.setPath(): in set cwd(%v), err=\n\t%v", []any{sf.pwd, err})
	sf.fileName = d0.LetTxt(fileName)
}

// Pwd -- возвращает текущий рабочий каталог процесса.
func (sf *MockEnv) Pwd() *d0.LTxt {
	return sf.pwd
}

// SetAll -- поднимает все переменные окружения.
func (sf *MockEnv) SetAll() {
	sf.ResetAll()
	for key, val := range sf.dictEnv {
		sf.Reset(d0.LetTxt("key"))
		err := os.Setenv(string(key), val.String())
		d0.If(err != nil).
			Hassert("MockEnv.SetAll(): in set env %v on key %v, err=\n\t%v",
				[]any{key, val, err})
	}
}

// ResetAll -- сбрасывает все переменные окружения.
func (sf *MockEnv) ResetAll() {
	for key := range sf.dictEnv {
		sf.Reset(d0.LetTxt(key))
	}
}

// Reset -- сбрасывает конкретный ключ.
func (sf *MockEnv) Reset(key *d0.LTxt) {
	d0.If(key.Get() == "").
		Hassert("MockEnv.Reset(): key is empty",
			[]any{})
	err := os.Unsetenv(key.String())
	d0.If(err != nil).
		Hassert("MockEnv.Reset(): in reset env %v, err=\n\t%v",
			[]any{key, err})
}