drone/internal/api/controller/space/export_progress.go

package space

import (
	"context"
	"fmt"

	apiauth "github.com/harness/gitness/internal/api/auth"
	"github.com/harness/gitness/internal/api/usererror"
	"github.com/harness/gitness/internal/auth"
	"github.com/harness/gitness/internal/services/exporter"
	"github.com/harness/gitness/types"
	"github.com/harness/gitness/types/enum"
	"github.com/pkg/errors"
)

type ExportProgressOutput struct {
	Repos []types.JobProgress `json:"repos"`
}

// ExportProgress returns progress of the export job.
func (c *Controller) ExportProgress(ctx context.Context,
	session *auth.Session,
	spaceRef string,
) (ExportProgressOutput, error) {
	space, err := c.spaceStore.FindByRef(ctx, spaceRef)
	if err != nil {
		return ExportProgressOutput{}, err
	}

	if err = apiauth.CheckSpace(ctx, c.authorizer, session, space, enum.PermissionSpaceView, false); err != nil {
		return ExportProgressOutput{}, err
	}

	progress, err := c.exporter.GetProgressForSpace(ctx, space.ID)
	if errors.Is(err, exporter.ErrNotFound) {
		return ExportProgressOutput{}, usererror.NotFound("No recent or ongoing export found for space.")
	}
	if err != nil {
		return ExportProgressOutput{}, fmt.Errorf("failed to retrieve export progress: %w", err)
	}

	return ExportProgressOutput{Repos: progress}, nil
}