UPDATE: Submission module

2026-05-04 09:55:17 +07:00
parent f3f2e09fd5
commit bcc2e192c1
48 changed files with 2918 additions and 359 deletions
--- a/internal/services/commitService.go
+++ b/internal/services/commitService.go
@@ -2,6 +2,7 @@ package services

 import (
 	"context"
+	"encoding/json"
 	"history-api/internal/dtos/request"
 	"history-api/internal/dtos/response"
 	"history-api/internal/gen/sqlc"
@@ -63,6 +64,12 @@ func (s *commitService) checkWritePermission(ctx context.Context, userID string,
 		return fiber.NewError(fiber.StatusForbidden, "You do not have permission to write to this project")
 	}

+	for _, s := range project.Submissions {
+		if s.Status == constants.StatusTypePending {
+			return fiber.NewError(fiber.StatusConflict, "Cannot create commit while there is a pending submission")
+		}
+	}
+
 	return nil
 }

@@ -90,9 +97,14 @@ func (s *commitService) CreateCommit(ctx context.Context, userID string, project
 		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid user ID")
 	}

+	snapshotJSON, err := json.Marshal(dto.SnapshotJson)
+	if err != nil {
+		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid snapshot JSON")
+	}
+
 	commit, err := cRepoTx.Create(ctx, sqlc.CreateCommitParams{
 		ProjectID:    projectUUID,
-		SnapshotJson: dto.SnapshotJson,
+		SnapshotJson: snapshotJSON,
 		UserID:       userUUID,
 		EditSummary:  convert.StringToText(dto.EditSummary),
 	})
--- a/internal/services/entityService.go
+++ b/internal/services/entityService.go
@@ -59,6 +59,13 @@ func (s *entityService) SearchEntities(ctx context.Context, req *request.SearchE
 		params.Name = req.Name
 	}

+	if req.ProjectID != nil {
+		projectID, err := convert.StringToUUID(*req.ProjectID)
+		if err == nil {
+			params.ProjectID = projectID
+		}
+	}
+
 	entities, err := s.entityRepo.Search(ctx, params)
 	if err != nil {
 		return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to search entities")
--- a/internal/services/submissionService.go
+++ b/internal/services/submissionService.go
@@ -2,6 +2,7 @@ package services

 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"history-api/internal/dtos/request"
 	"history-api/internal/dtos/response"
@@ -32,6 +33,9 @@ type submissionService struct {
 	projectRepo    repositories.ProjectRepository
 	commitRepo     repositories.CommitRepository
 	userRepo       repositories.UserRepository
+	wikiRepo       repositories.WikiRepository
+	geometryRepo   repositories.GeometryRepository
+	entityRepo     repositories.EntityRepository
 	db             *pgxpool.Pool
 	c              cache.Cache
 }
@@ -41,6 +45,9 @@ func NewSubmissionService(
 	projectRepo repositories.ProjectRepository,
 	commitRepo repositories.CommitRepository,
 	userRepo repositories.UserRepository,
+	wikiRepo repositories.WikiRepository,
+	geometryRepo repositories.GeometryRepository,
+	entityRepo repositories.EntityRepository,
 	db *pgxpool.Pool,
 	c cache.Cache,
 ) SubmissionService {
@@ -49,6 +56,9 @@ func NewSubmissionService(
 		projectRepo:    projectRepo,
 		commitRepo:     commitRepo,
 		userRepo:       userRepo,
+		wikiRepo:       wikiRepo,
+		geometryRepo:   geometryRepo,
+		entityRepo:     entityRepo,
 		db:             db,
 		c:              c,
 	}
@@ -78,6 +88,17 @@ func (s *submissionService) CreateSubmission(ctx context.Context, userID string,
 		return nil, fiber.NewError(fiber.StatusBadRequest, "Commit does not belong to project")
 	}

+	project, err := s.projectRepo.GetByID(ctx, projectUUID)
+	if err != nil {
+		return nil, fiber.NewError(fiber.StatusNotFound, "Project not found")
+	}
+
+	for _, sub := range project.Submissions {
+		if sub.Status == constants.StatusTypePending {
+			return nil, fiber.NewError(fiber.StatusConflict, "There is already a pending submission for this project")
+		}
+	}
+
 	arg := sqlc.CreateSubmissionParams{
 		ProjectID: projectUUID,
 		CommitID:  commitUUID,
@@ -95,6 +116,18 @@ func (s *submissionService) CreateSubmission(ctx context.Context, userID string,
 }

 func (s *submissionService) UpdateSubmissionStatus(ctx context.Context, reviewerID string, submissionID string, dto *request.UpdateSubmissionStatusDto) (*response.SubmissionResponse, *fiber.Error) {
+	tx, err := s.db.Begin(ctx)
+	if err != nil {
+		return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to start transaction")
+	}
+	defer tx.Rollback(ctx)
+
+	submissionRepo := s.submissionRepo.WithTx(tx)
+	commitRepo := s.commitRepo.WithTx(tx)
+	entityRepo := s.entityRepo.WithTx(tx)
+	geometryRepo := s.geometryRepo.WithTx(tx)
+	wikiRepo := s.wikiRepo.WithTx(tx)
+
 	submissionUUID, err := convert.StringToUUID(submissionID)
 	if err != nil {
 		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid submission ID")
@@ -119,6 +152,404 @@ func (s *submissionService) UpdateSubmissionStatus(ctx context.Context, reviewer
 		return nil, fiber.NewError(fiber.StatusBadRequest, "Submission already processed")
 	}

+	commitUUID, err := convert.StringToUUID(submission.CommitID)
+	if err != nil {
+		return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid commit ID")
+	}
+
+	commit, err := s.commitRepo.GetByID(ctx, commitUUID)
+	if err != nil {
+		return nil, fiber.NewError(fiber.StatusNotFound, "Commit not found")
+	}
+
+	if commit.ProjectID != submission.ProjectID {
+		return nil, fiber.NewError(fiber.StatusBadRequest, "Commit does not belong to project")
+	}
+
+	if status == constants.StatusTypeApproved {
+		var snapshotData request.CommitSnapshot
+		err = json.Unmarshal(commit.SnapshotJson, &snapshotData)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to parse commit snapshot")
+		}
+
+		projectUUID, err := convert.StringToUUID(commit.ProjectID)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusBadRequest, "Invalid project ID")
+		}
+		currentEntity, err := s.entityRepo.GetByProjectID(ctx, projectUUID)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusNotFound, "Entity not found")
+		}
+
+		currentGeometry, err := s.geometryRepo.GetByProjectID(ctx, projectUUID)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusNotFound, "Geometry not found")
+		}
+
+		currentWiki, err := s.wikiRepo.GetByProjectID(ctx, projectUUID)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusNotFound, "Wiki not found")
+		}
+
+		persistItemIDs := make(map[string]struct{})
+		for _, item := range snapshotData.Entities {
+			persistItemIDs[item.ID] = struct{}{}
+		}
+		for _, item := range snapshotData.Geometries {
+			persistItemIDs[item.ID] = struct{}{}
+		}
+		for _, item := range snapshotData.Wikis {
+			persistItemIDs[item.ID] = struct{}{}
+		}
+
+		persistCurrentItemIDs := make(map[string]struct{})
+		for _, item := range currentEntity {
+			persistCurrentItemIDs[item.ID] = struct{}{}
+		}
+		for _, item := range currentGeometry {
+			persistCurrentItemIDs[item.ID] = struct{}{}
+		}
+		for _, item := range currentWiki {
+			persistCurrentItemIDs[item.ID] = struct{}{}
+		}
+
+		listDeleteEntities := make([]pgtype.UUID, 0)
+		for _, e := range currentEntity {
+			if _, ok := persistItemIDs[e.ID]; !ok {
+				itemUUID, err := convert.StringToUUID(e.ID)
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Invalid entity ID")
+				}
+				listDeleteEntities = append(listDeleteEntities, itemUUID)
+				delete(persistCurrentItemIDs, e.ID)
+			}
+		}
+
+		listDeleteGeometries := make([]pgtype.UUID, 0)
+		for _, g := range currentGeometry {
+			if _, ok := persistItemIDs[g.ID]; !ok {
+				itemUUID, err := convert.StringToUUID(g.ID)
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Invalid geometry ID")
+				}
+				listDeleteGeometries = append(listDeleteGeometries, itemUUID)
+				delete(persistCurrentItemIDs, g.ID)
+			}
+		}
+
+		listDeleteWikis := make([]pgtype.UUID, 0)
+		for _, w := range currentWiki {
+			if _, ok := persistItemIDs[w.ID]; !ok {
+				itemUUID, err := convert.StringToUUID(w.ID)
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Invalid wiki ID")
+				}
+				listDeleteWikis = append(listDeleteWikis, itemUUID)
+				delete(persistCurrentItemIDs, w.ID)
+			}
+		}
+
+		if len(listDeleteEntities) > 0 {
+			if err = entityRepo.DeleteByIDs(ctx, listDeleteEntities); err != nil {
+				return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to delete entities")
+			}
+		}
+
+		if len(listDeleteGeometries) > 0 {
+			if err = geometryRepo.DeleteByIDs(ctx, listDeleteGeometries); err != nil {
+				return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to delete geometries")
+			}
+		}
+
+		if len(listDeleteWikis) > 0 {
+			if err = wikiRepo.DeleteByIDs(ctx, listDeleteWikis); err != nil {
+				return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to delete wikis")
+			}
+		}
+
+		refEntityIDs := []string{}
+		for _, e := range snapshotData.Entities {
+			if e.Source == "ref" {
+				refEntityIDs = append(refEntityIDs, e.ID)
+			}
+		}
+		refEntities, _ := s.entityRepo.GetByIDs(ctx, refEntityIDs)
+		refEntityMap := make(map[string]bool)
+		for _, e := range refEntities {
+			refEntityMap[e.ID] = true
+		}
+
+		newEntities := make([]*request.EntitySnapshot, 0, len(snapshotData.Entities))
+		for i, entity := range snapshotData.Entities {
+			if entity.Operation == "delete" {
+				continue
+			}
+
+			entityUUID, err := convert.StringToUUID(entity.ID)
+			if err != nil {
+				return nil, fiber.NewError(fiber.StatusInternalServerError, "Invalid entity ID")
+			}
+
+			if _, ok := persistCurrentItemIDs[entity.ID]; ok {
+				_, err := entityRepo.Update(ctx, sqlc.UpdateEntityParams{
+					Name:        convert.StringToText(entity.Name),
+					Description: convert.StringToText(entity.Description),
+					Slug:        convert.PtrToText(entity.Slug),
+					Status:      convert.PtrToInt2(entity.Status),
+					ID:          entityUUID,
+				})
+
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to update entity: "+entity.ID)
+				}
+
+				newEntities = append(newEntities, snapshotData.Entities[i])
+
+			} else if entity.Source == "inline" {
+				_, err := entityRepo.Create(ctx, sqlc.CreateEntityParams{
+					ID:          entityUUID,
+					Name:        entity.Name,
+					Description: convert.StringToText(entity.Description),
+					ProjectID:   projectUUID,
+					Slug:        convert.PtrToText(entity.Slug),
+					Status:      convert.PtrToInt2(entity.Status),
+				})
+
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to create entity: "+entity.ID)
+				}
+
+				newEntities = append(newEntities, snapshotData.Entities[i])
+
+			} else if entity.Source == "ref" {
+				if !refEntityMap[entity.ID] {
+					continue
+				}
+				newEntities = append(newEntities, snapshotData.Entities[i])
+			}
+		}
+		snapshotData.Entities = newEntities
+
+		refGeometryIDs := []string{}
+		for _, g := range snapshotData.Geometries {
+			if g.Source == "ref" {
+				refGeometryIDs = append(refGeometryIDs, g.ID)
+			}
+		}
+		refGeometries, _ := s.geometryRepo.GetByIDs(ctx, refGeometryIDs)
+		refGeometryMap := make(map[string]bool)
+		for _, g := range refGeometries {
+			refGeometryMap[g.ID] = true
+		}
+
+		newGeometries := make([]*request.GeometrySnapshot, 0, len(snapshotData.Geometries))
+		for i, geo := range snapshotData.Geometries {
+			if geo.Operation == "delete" {
+				continue
+			}
+
+			geometryUUID, err := convert.StringToUUID(geo.ID)
+			if err != nil {
+				return nil, fiber.NewError(fiber.StatusInternalServerError, "Invalid geometry ID")
+			}
+
+			binding, _ := json.Marshal(geo.Binding)
+
+			if _, ok := persistCurrentItemIDs[geo.ID]; ok {
+				params := sqlc.UpdateGeometryParams{
+					ID:           geometryUUID,
+					GeoType:      pgtype.Int2{Int16: constants.ParseGeoTypeText(geo.Type).Int16(), Valid: true},
+					DrawGeometry: geo.DrawGeometry,
+					Binding:      binding,
+					TimeStart:    convert.PtrFloat64ToInt4(geo.TimeStart),
+					TimeEnd:      convert.PtrFloat64ToInt4(geo.TimeEnd),
+					ProjectID:    projectUUID,
+				}
+
+				if geo.BBox != nil {
+					params.UpdateBbox = pgtype.Bool{Bool: true, Valid: true}
+					params.MinLng = pgtype.Float8{Float64: geo.BBox.MinLng, Valid: true}
+					params.MinLat = pgtype.Float8{Float64: geo.BBox.MinLat, Valid: true}
+					params.MaxLng = pgtype.Float8{Float64: geo.BBox.MaxLng, Valid: true}
+					params.MaxLat = pgtype.Float8{Float64: geo.BBox.MaxLat, Valid: true}
+				}
+
+				_, err := geometryRepo.Update(ctx, params)
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to update geometry: "+geo.ID)
+				}
+				newGeometries = append(newGeometries, snapshotData.Geometries[i])
+
+			} else if geo.Source == "inline" {
+				params := sqlc.CreateGeometryParams{
+					ID:           geometryUUID,
+					GeoType:      constants.ParseGeoTypeText(geo.Type).Int16(),
+					DrawGeometry: geo.DrawGeometry,
+					Binding:      binding,
+					TimeStart:    convert.PtrFloat64ToInt4(geo.TimeStart),
+					TimeEnd:      convert.PtrFloat64ToInt4(geo.TimeEnd),
+					ProjectID:    projectUUID,
+				}
+				if geo.BBox != nil {
+					params.MinLng = geo.BBox.MinLng
+					params.MinLat = geo.BBox.MinLat
+					params.MaxLng = geo.BBox.MaxLng
+					params.MaxLat = geo.BBox.MaxLat
+				}
+
+				_, err := geometryRepo.Create(ctx, params)
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to create geometry: "+geo.ID)
+				}
+				newGeometries = append(newGeometries, snapshotData.Geometries[i])
+
+			} else if geo.Source == "ref" {
+				if !refGeometryMap[geo.ID] {
+					continue
+				}
+				newGeometries = append(newGeometries, snapshotData.Geometries[i])
+			}
+		}
+		snapshotData.Geometries = newGeometries
+
+		refWikiIDs := []string{}
+		for _, w := range snapshotData.Wikis {
+			if w.Source == "ref" {
+				refWikiIDs = append(refWikiIDs, w.ID)
+			}
+		}
+		refWikis, _ := s.wikiRepo.GetByIDs(ctx, refWikiIDs)
+		refWikiMap := make(map[string]bool)
+		for _, w := range refWikis {
+			refWikiMap[w.ID] = true
+		}
+
+		newWikis := make([]*request.WikiSnapshot, 0, len(snapshotData.Wikis))
+		for i, wiki := range snapshotData.Wikis {
+			if wiki.Operation == "delete" {
+				continue
+			}
+
+			wikiUUID, err := convert.StringToUUID(wiki.ID)
+			if err != nil {
+				return nil, fiber.NewError(fiber.StatusInternalServerError, "Invalid wiki ID")
+			}
+
+			if _, ok := persistCurrentItemIDs[wiki.ID]; ok {
+				_, err := wikiRepo.Update(ctx, sqlc.UpdateWikiParams{
+					ID:        wikiUUID,
+					Title:     convert.StringToText(wiki.Title),
+					Content:   wiki.Doc,
+					ProjectID: projectUUID,
+				})
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to update wiki: "+wiki.ID)
+				}
+				newWikis = append(newWikis, snapshotData.Wikis[i])
+
+			} else if wiki.Source == "inline" {
+				_, err := wikiRepo.Create(ctx, sqlc.CreateWikiParams{
+					ID:        wikiUUID,
+					Title:     convert.StringToText(wiki.Title),
+					Content:   wiki.Doc,
+					ProjectID: projectUUID,
+				})
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to create wiki: "+wiki.ID)
+				}
+				newWikis = append(newWikis, snapshotData.Wikis[i])
+
+			} else if wiki.Source == "ref" {
+				if !refWikiMap[wiki.ID] {
+					continue
+				}
+				newWikis = append(newWikis, snapshotData.Wikis[i])
+			}
+		}
+		snapshotData.Wikis = newWikis
+
+		err = geometryRepo.DeleteEntityGeometriesByProjectID(ctx, projectUUID)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to delete geometry entity: "+err.Error())
+		}
+		err = wikiRepo.DeleteEntityWikisByProjectID(ctx, projectUUID)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to delete wiki entity: "+err.Error())
+		}
+
+		validEntities := make(map[string]bool)
+		for _, e := range snapshotData.Entities {
+			validEntities[e.ID] = true
+		}
+		validGeometries := make(map[string]bool)
+		for _, g := range snapshotData.Geometries {
+			validGeometries[g.ID] = true
+		}
+		validWikis := make(map[string]bool)
+		for _, w := range snapshotData.Wikis {
+			validWikis[w.ID] = true
+		}
+
+		if len(snapshotData.GeometryEntity) > 0 {
+			geomLinks := make(map[string][]pgtype.UUID)
+			for _, link := range snapshotData.GeometryEntity {
+				if !validEntities[link.EntityID] || !validGeometries[link.GeometryID] {
+					continue
+				}
+				gID, _ := convert.StringToUUID(link.GeometryID)
+				geomLinks[link.EntityID] = append(geomLinks[link.EntityID], gID)
+			}
+
+			for eIDStr, gIDs := range geomLinks {
+				eID, _ := convert.StringToUUID(eIDStr)
+				err = geometryRepo.CreateEntityGeometries(ctx, sqlc.CreateEntityGeometriesParams{
+					EntityID:    eID,
+					GeometryIds: gIDs,
+					ProjectID:   projectUUID,
+				})
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to create geometry entity: "+err.Error())
+				}
+			}
+		}
+
+		if len(snapshotData.EntityWiki) > 0 {
+			wikiLinks := make(map[string][]pgtype.UUID)
+			for _, link := range snapshotData.EntityWiki {
+				if link.Operation == "delete" || (link.IsDeleted != nil && *link.IsDeleted == 1) {
+					continue
+				}
+				if !validEntities[link.EntityID] || !validWikis[link.WikiID] {
+					continue
+				}
+				wID, _ := convert.StringToUUID(link.WikiID)
+				wikiLinks[link.EntityID] = append(wikiLinks[link.EntityID], wID)
+			}
+
+			for eIDStr, wIDs := range wikiLinks {
+				eID, _ := convert.StringToUUID(eIDStr)
+				err = wikiRepo.CreateEntityWikis(ctx, sqlc.CreateEntityWikisParams{
+					EntityID:  eID,
+					WikiIds:   wIDs,
+					ProjectID: projectUUID,
+				})
+				if err != nil {
+					return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to create wiki entity: "+err.Error())
+				}
+			}
+		}
+
+		newSnapshot, err := json.Marshal(snapshotData)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to marshal snapshot")
+		}
+		_, err = commitRepo.UpdateSnapshot(ctx, commitUUID, newSnapshot)
+		if err != nil {
+			return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to update snapshot: "+err.Error())
+		}
+	}
+
 	arg := sqlc.UpdateSubmissionParams{
 		ID:         submissionUUID,
 		Status:     pgtype.Int2{Int16: status.Int16(), Valid: true},
@@ -126,11 +557,25 @@ func (s *submissionService) UpdateSubmissionStatus(ctx context.Context, reviewer
 		ReviewNote: convert.StringToText(dto.ReviewNote),
 	}

-	updatedSubmission, err := s.submissionRepo.Update(ctx, arg)
+	updatedSubmission, err := submissionRepo.Update(ctx, arg)
+
 	if err != nil {
 		return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to update submission status")
 	}

+	err = tx.Commit(ctx)
+	if err != nil {
+		return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to commit transaction")
+	}
+
+	if status == constants.StatusTypeApproved {
+		go func() {
+			bgCtx := context.Background()
+			_ = s.c.DelByPattern(bgCtx, "entity:search*")
+			_ = s.c.DelByPattern(bgCtx, "geometry:search*")
+			_ = s.c.DelByPattern(bgCtx, "wiki:search*")
+		}()
+	}
 	_ = s.c.Del(ctx, fmt.Sprintf("project:id:%s", submission.ProjectID))

 	return updatedSubmission.ToResponse(), nil
--- a/internal/services/wikiService.go
+++ b/internal/services/wikiService.go
@@ -58,6 +58,7 @@ func (s *wikiService) SearchWikis(ctx context.Context, req *request.SearchWikiDt
 	if req.Title != "" {
 		params.Title = req.Title
 	}
+
 	if req.EntityID != "" {
 		entityId, err := convert.StringToUUID(req.EntityID)
 		if err == nil {
@@ -65,6 +66,13 @@ func (s *wikiService) SearchWikis(ctx context.Context, req *request.SearchWikiDt
 		}
 	}

+	if req.ProjectID != nil {
+		projectID, err := convert.StringToUUID(*req.ProjectID)
+		if err == nil {
+			params.ProjectID = projectID
+		}
+	}
+
 	wikis, err := s.wikiRepo.Search(ctx, params)
 	if err != nil {
 		return nil, fiber.NewError(fiber.StatusInternalServerError, "Failed to search wikis")