aides-repo-api/internal/services/taskservice/service.go

package taskservice

import (
	"os"
	"path"
	"strconv"

	"code.alt-gnome.ru/aides-infra/aides-repo-api/internal/app"
	"code.alt-gnome.ru/aides-infra/aides-repo-api/internal/models"
)

type Service struct {
	app *app.App
}

func New(app *app.App) *Service {
	return &Service{
		app: app,
	}
}

func (s *Service) onTaskComplete(task *models.Task) error {
	if err := s.app.Db.Save(&task).Error; err != nil {
		return err
	}

	grart := models.GitRepoAltRepoTask{
		ALTRepoID: 1,
		Repo:      task.Repo,
	}
	s.app.Db.Debug().Where(
		&grart,
	).FirstOrCreate(&grart)

	grart.LastTaskID = task.ID

	if err := s.app.Db.Save(&grart).Error; err != nil {
		return err
	}

	s.tasksCleanup(&grart, 0)

	return nil
}

func (s *Service) tasksCleanup(r *models.GitRepoAltRepoTask, N int) {
	excludedTaskIDs := []uint{}
	if r.CurrentTaskID != 0 {
		excludedTaskIDs = append(excludedTaskIDs, r.CurrentTaskID)
	}
	if r.LastTaskID != 0 {
		excludedTaskIDs = append(excludedTaskIDs, r.LastTaskID)
	}

	var lastNTaskIDs []uint
	s.app.Db.
		Debug().
		Table("tasks").
		Select("id").
		Where("repo_id = ?", r.ID).
		Where("status = ?", models.StatusCompleted).
		Where("id NOT IN ?", excludedTaskIDs).
		Order("created_at DESC").
		Limit(N).
		Pluck("id", &lastNTaskIDs)

	excludedTaskIDs = append(excludedTaskIDs, lastNTaskIDs...)

	var taskIDsToDelete []int
	s.app.Db.
		Debug().
		Model(&models.Task{}).
		Select("id").
		Where("repo_id = ?", r.RepoID).
		Where("files_removed = ?", false).
		Where("id NOT IN ?", excludedTaskIDs).
		Pluck("id", &taskIDsToDelete)

	if len(taskIDsToDelete) > 0 {
		s.app.Db.
			Debug().
			Model(&models.Task{}).
			Where("id IN ?", taskIDsToDelete).
			Update("files_removed", true)

		for _, id := range taskIDsToDelete {
			s.removeTaskFiles(id)
		}
	}
}

func (s *Service) removeTaskFiles(taskId int) {
	taskFolderPath := path.Join(
		s.app.Config.UploadDir,
		"tasks",
		strconv.Itoa(taskId),
	)
	os.RemoveAll(taskFolderPath)
}