History_Api/internal/services/projectService.go

package services

import (
	"context"
	"fmt"

	"github.com/gofiber/fiber/v3"
	"github.com/jackc/pgx/v5/pgtype"
	"golang.org/x/sync/errgroup"

	"history-api/internal/dtos/request"
	"history-api/internal/dtos/response"
	"history-api/internal/gen/sqlc"
	"history-api/internal/models"
	"history-api/internal/repositories"
	"history-api/pkg/constants"
	"history-api/pkg/convert"
)

type ProjectService interface {
	GetProjectByID(ctx context.Context, id string) (*response.ProjectResponse, error)
	GetProjectByUserID(ctx context.Context, userID string, dto *request.GetProjectsByUserDto) ([]*response.ProjectResponse, error)
	SearchProject(ctx context.Context, dto *request.SearchProjectDto) (*response.PaginatedResponse, error)
	DeleteProject(ctx context.Context, id string) error
	CreateProject(ctx context.Context, userID string, dto *request.CreateProjectDto) (*response.ProjectResponse, error)
	UpdateProject(ctx context.Context, id string, dto *request.UpdateProjectDto) (*response.ProjectResponse, error)
}

type projectService struct {
	projectRepo repositories.ProjectRepository
}

func NewProjectService(projectRepo repositories.ProjectRepository) ProjectService {
	return &projectService{
		projectRepo: projectRepo,
	}
}

func (s *projectService) GetProjectByID(ctx context.Context, id string) (*response.ProjectResponse, error) {
	projectUUID, err := convert.StringToUUID(id)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid project ID format")
	}

	project, err := s.projectRepo.GetByID(ctx, projectUUID)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusNotFound, "Project not found")
	}

	return project.ToResponse(), nil
}

func (s *projectService) GetProjectByUserID(ctx context.Context, userID string, dto *request.GetProjectsByUserDto) ([]*response.ProjectResponse, error) {
	userUUID, err := convert.StringToUUID(userID)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid user ID format")
	}

	limit := int32(20)
	if dto.Limit > 0 {
		limit = dto.Limit
	}

	arg := sqlc.GetProjectsByUserIdParams{
		UserID: userUUID,
		Limit:  limit,
	}

	if dto.CursorID != "" {
		if cursorID, err := convert.StringToUUID(dto.CursorID); err == nil {
			arg.CursorID = cursorID
		}
	}

	projects, err := s.projectRepo.GetByUserID(ctx, arg)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusInternalServerError, err.Error())
	}

	return models.ProjectsEntityToResponse(projects), nil
}

func (s *projectService) fillSearchArgs(arg *sqlc.SearchProjectsParams, dto *request.SearchProjectDto) {
	if dto.Sort != "" {
		arg.Sort = dto.Sort
	} else {
		arg.Sort = "updated_at"
	}

	arg.Order = "desc"
	if dto.Order == "asc" {
		arg.Order = "asc"
	}

	if len(dto.Statuses) > 0 {
		for _, statusStr := range dto.Statuses {
			statusType := constants.ParseProjectStatusTypeText(statusStr)
			if statusType != constants.ProjectStatusTypeUnknow {
				arg.Statuses = append(arg.Statuses, fmt.Sprintf("%d", statusType.Int16()))
			}
		}
	}

	if len(dto.UserIDs) > 0 {
		for _, id := range dto.UserIDs {
			if u, err := convert.StringToUUID(id); err == nil {
				arg.UserIds = append(arg.UserIds, u)
			}
		}
	}

	if dto.CreatedFrom != nil {
		arg.CreatedFrom = pgtype.Timestamptz{Time: *dto.CreatedFrom, Valid: true}
	}
	if dto.CreatedTo != nil {
		arg.CreatedTo = pgtype.Timestamptz{Time: *dto.CreatedTo, Valid: true}
	}

	if dto.Search != "" {
		arg.SearchText = pgtype.Text{String: dto.Search, Valid: true}
	}
}

func (s *projectService) SearchProject(ctx context.Context, dto *request.SearchProjectDto) (*response.PaginatedResponse, error) {
	if dto.Page < 1 {
		dto.Page = 1
	}
	if dto.Limit == 0 {
		dto.Limit = 20
	}
	offset := (dto.Page - 1) * dto.Limit

	arg := sqlc.SearchProjectsParams{
		Limit:  int32(dto.Limit),
		Offset: int32(offset),
	}

	s.fillSearchArgs(&arg, dto)

	var rows []*models.ProjectEntity
	var totalRecords int64

	g, gCtx := errgroup.WithContext(ctx)

	g.Go(func() error {
		var err error
		rows, err = s.projectRepo.Search(gCtx, arg)
		return err
	})

	g.Go(func() error {
		countArg := sqlc.CountProjectsParams{
			Statuses:    arg.Statuses,
			UserIds:     arg.UserIds,
			SearchText:  arg.SearchText,
			CreatedFrom: arg.CreatedFrom,
			CreatedTo:   arg.CreatedTo,
		}
		var err error
		totalRecords, err = s.projectRepo.Count(gCtx, countArg)
		return err
	})

	if err := g.Wait(); err != nil {
		return nil, fiber.NewError(fiber.StatusInternalServerError, err.Error())
	}

	projects := models.ProjectsEntityToResponse(rows)

	return response.BuildPaginatedResponse(projects, totalRecords, dto.Page, dto.Limit), nil
}

func (s *projectService) CreateProject(ctx context.Context, userID string, dto *request.CreateProjectDto) (*response.ProjectResponse, error) {
	userUUID, err := convert.StringToUUID(userID)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid user ID format")
	}

	arg := sqlc.CreateProjectParams{
		Title:         dto.Title,
		Description:   convert.PtrToText(dto.Description),
		ProjectStatus: constants.ProjectStatusTypePrivate.Int16(),
		UserID:        userUUID,
	}

	if dto.Status != nil {
		statusType := constants.ParseProjectStatusTypeText(*dto.Status)
		if statusType == constants.ProjectStatusTypeUnknow {
			return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid status type")
		}
		arg.ProjectStatus = statusType.Int16()
	}

	project, err := s.projectRepo.Create(ctx, arg)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to create project")
	}

	return project.ToResponse(), nil
}

func (s *projectService) UpdateProject(ctx context.Context, id string, dto *request.UpdateProjectDto) (*response.ProjectResponse, error) {
	projectUUID, err := convert.StringToUUID(id)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid project ID format")
	}

	_, err = s.projectRepo.GetByID(ctx, projectUUID)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusNotFound, "Project not found")
	}

	arg := sqlc.UpdateProjectParams{
		ID: projectUUID,
	}

	if dto.Title != nil {
		arg.Title = convert.PtrToText(dto.Title)
	}
	if dto.Description != nil {
		arg.Description = convert.PtrToText(dto.Description)
	}
	if dto.Status != nil {
		statusType := constants.ParseProjectStatusTypeText(*dto.Status)
		if statusType == constants.ProjectStatusTypeUnknow {
			return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid status type")
		}
		arg.Status = pgtype.Int2{Int16: statusType.Int16(), Valid: true}
	}

	project, err := s.projectRepo.Update(ctx, arg)
	if err != nil {
		return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to update project")
	}

	return project.ToResponse(), nil
}

func (s *projectService) DeleteProject(ctx context.Context, id string) error {
	projectUUID, err := convert.StringToUUID(id)
	if err != nil {
		return fiber.NewError(fiber.StatusBadRequest, "Invalid project ID format")
	}

	_, err = s.projectRepo.GetByID(ctx, projectUUID)
	if err != nil {
		return fiber.NewError(fiber.StatusNotFound, "Project not found")
	}

	err = s.projectRepo.Delete(ctx, projectUUID)
	if err != nil {
		return fiber.NewError(fiber.StatusInternalServerError, "Failed to delete project")
	}

	return nil
}