kandev/apps/backend/internal/gitlab/service_search.go

package gitlab

import (
	"context"

	"go.uber.org/zap"
)

// SearchUserMRs returns MRs matching a filter for the configured user.
func (s *Service) SearchUserMRs(ctx context.Context, filter, customQuery string) ([]*MR, error) {
	client := s.Client()
	if client == nil {
		return nil, ErrNoClient
	}
	return client.SearchMRs(ctx, filter, customQuery)
}

// SearchUserMRsPaged returns paginated MRs.
func (s *Service) SearchUserMRsPaged(ctx context.Context, filter, customQuery string, page, perPage int) (*MRSearchPage, error) {
	client := s.Client()
	if client == nil {
		return nil, ErrNoClient
	}
	return client.SearchMRsPaged(ctx, filter, customQuery, page, perPage)
}

// SearchUserIssues returns issues matching a filter for the configured user.
func (s *Service) SearchUserIssues(ctx context.Context, filter, customQuery, milestone string) ([]*Issue, error) {
	client := s.Client()
	if client == nil {
		return nil, ErrNoClient
	}
	return client.ListIssues(ctx, filter, customQuery, milestone)
}

// SearchUserIssuesPaged returns paginated issues.
func (s *Service) SearchUserIssuesPaged(ctx context.Context, filter, customQuery, milestone string, page, perPage int) (*IssueSearchPage, error) {
	client := s.Client()
	if client == nil {
		return nil, ErrNoClient
	}
	return client.ListIssuesPaged(ctx, filter, customQuery, milestone, page, perPage)
}

// GetStats aggregates open MR / awaiting-review / open-issue counts.
// Per-count failures are logged and degraded to zero so the dashboard
// renders partial data instead of failing the entire request — but the
// log gives ops something to triage when counts look wrong.
func (s *Service) GetStats(ctx context.Context) (*Stats, error) {
	client := s.Client()
	return s.getStatsWithClient(ctx, client)
}

func (s *Service) GetStatsForWorkspace(ctx context.Context, workspaceID string) (*Stats, error) {
	client, err := s.ClientForWorkspace(ctx, workspaceID)
	if err != nil {
		return nil, err
	}
	return s.getStatsWithClient(ctx, client)
}

func (s *Service) getStatsWithClient(ctx context.Context, client Client) (*Stats, error) {
	if client == nil {
		return &Stats{}, nil
	}
	username, err := client.GetAuthenticatedUser(ctx)
	if err != nil {
		s.logger.Warn("gitlab stats: resolve authenticated user", zap.Error(err))
		return &Stats{}, nil
	}
	if username == "" {
		return &Stats{}, nil
	}
	open, err := client.SearchMRsPaged(ctx, "scope=created_by_me&state=opened", "", 1, 1)
	openMRs := 0
	if err != nil {
		s.logger.Warn("gitlab stats: open MRs", zap.Error(err))
	} else if open != nil {
		openMRs = open.TotalCount
	}
	awaiting, err := client.SearchMRsPaged(ctx, "reviewer_username="+username+"&state=opened", "", 1, 1)
	awaitingMRs := 0
	if err != nil {
		s.logger.Warn("gitlab stats: MRs awaiting review", zap.Error(err))
	} else if awaiting != nil {
		awaitingMRs = awaiting.TotalCount
	}
	issues, err := client.ListIssuesPaged(ctx, "scope=assigned_to_me&state=opened", "", "", 1, 1)
	openIssues := 0
	if err != nil {
		s.logger.Warn("gitlab stats: open issues", zap.Error(err))
	} else if issues != nil {
		openIssues = issues.TotalCount
	}
	return &Stats{
		OpenMRs:              openMRs,
		MRsAwaitingMyReview:  awaitingMRs,
		OpenIssuesAssignedMe: openIssues,
	}, nil
}