services/export/ProjectUser.go - develop - wax_boy/semaphore_custom

services/export/ProjectUser.go
84 строки2 KB
Denis Gukov
sec/golangci (#3817)
02 май 2026, 16:52
02 май 2026, 16:5271e2baf
package export

import (
	"strconv"

	"github.com/semaphoreui/semaphore/db"
)

type ProjectUserExporter struct {
	ValueMap[db.ProjectUser]
}

func (e *ProjectUserExporter) load(store db.Store, exporter DataExporter, progress Progress) error {

	projs, err := exporter.getLoadedKeysInt(Project, GlobalScope)
	if err != nil {
		return err
	}

	for _, projId := range projs {
		users, err := store.GetProjectUsers(projId, db.RetrieveQueryParams{})
		if err != nil {
			return err
		}

		err = e.appendValues(getUsers(users, projId), strconv.Itoa(projId))
		if err != nil {
			return err
		}
	}

	return nil
}

func getUsers(vals []db.UserWithProjectRole, projId int) []db.ProjectUser {
	values := make([]db.ProjectUser, 0)

	for _, val := range vals {
		values = append(values, db.ProjectUser{
			UserID:    val.ID,
			Role:      val.Role,
			ProjectID: projId,
		})
	}

	return values
}

func (e *ProjectUserExporter) restore(store db.Store, exporter DataExporter, progress Progress) (err error) {
	return e.restoreValues(store, exporter, progress, e)
}

func (e *ProjectUserExporter) restoreValue(val EntityObject[db.ProjectUser], store db.Store, exporter DataExporter) (err error) {
	old := val.value

	old.ProjectID, err = exporter.getNewKeyInt(Project, GlobalScope, old.ProjectID)
	if err != nil {
		return err
	}

	old.UserID, err = exporter.getNewKeyInt(User, GlobalScope, old.UserID)
	if err != nil {
		return err
	}

	newObj, err := store.CreateProjectUser(old)
	if err != nil {
		return err
	}

	return exporter.mapKeys(e.getName(), val.scope, old.GetDbKey(), newObj.GetDbKey())
}

func (e *ProjectUserExporter) exportDependsOn() []string {
	return []string{User, Project}
}

func (e *ProjectUserExporter) importDependsOn() []string {
	return []string{User, Project}
}

func (e *ProjectUserExporter) getName() string {
	return ProjectUser
}