probo/pkg/coredata/trust_center_document_access.go

// Copyright (c) 2025-2026 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

package coredata

import (
	"context"
	"errors"
	"fmt"
	"maps"
	"time"

	"github.com/jackc/pgx/v5"
	"github.com/jackc/pgx/v5/pgconn"
	"go.gearno.de/kit/pg"
	"go.probo.inc/probo/pkg/gid"
	"go.probo.inc/probo/pkg/page"
)

type (
	TrustCenterDocumentAccess struct {
		ID                  gid.GID                         `db:"id"`
		OrganizationID      gid.GID                         `db:"organization_id"`
		TrustCenterAccessID gid.GID                         `db:"trust_center_access_id"`
		DocumentID          *gid.GID                        `db:"document_id"`
		ReportID            *gid.GID                        `db:"report_id"`
		TrustCenterFileID   *gid.GID                        `db:"trust_center_file_id"`
		Status              TrustCenterDocumentAccessStatus `db:"status"`
		CreatedAt           time.Time                       `db:"created_at"`
		UpdatedAt           time.Time                       `db:"updated_at"`
	}

	TrustCenterDocumentAccesses []*TrustCenterDocumentAccess
)

func (tcda *TrustCenterDocumentAccess) CursorKey(orderBy TrustCenterDocumentAccessOrderField) page.CursorKey {
	switch orderBy {
	case TrustCenterDocumentAccessOrderFieldCreatedAt:
		return page.NewCursorKey(tcda.ID, tcda.CreatedAt)
	}

	panic(fmt.Sprintf("unsupported order by: %s", orderBy))
}

func (tcda *TrustCenterDocumentAccess) AuthorizationAttributes(ctx context.Context, conn pg.Querier) (map[string]string, error) {
	q := `SELECT organization_id FROM trust_center_document_accesses WHERE id = $1 LIMIT 1;`

	var organizationID gid.GID
	if err := conn.QueryRow(ctx, q, tcda.ID).Scan(&organizationID); err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return nil, ErrResourceNotFound
		}
		return nil, fmt.Errorf("cannot query trust center document access authorization attributes: %w", err)
	}

	return map[string]string{"organization_id": organizationID.String()}, nil
}

func (tcda *TrustCenterDocumentAccess) LoadByID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	accessID gid.GID,
) error {
	q := `
SELECT
    id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
FROM
    trust_center_document_accesses
WHERE
    %s
    AND id = @access_id
LIMIT 1;
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{"access_id": accessID}
	maps.Copy(args, scope.SQLArguments())

	rows, err := conn.Query(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot query trust center document access: %w", err)
	}

	access, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[TrustCenterDocumentAccess])
	if err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return ErrResourceNotFound
		}
		return fmt.Errorf("cannot collect trust center document access: %w", err)
	}

	*tcda = access

	return nil
}

func (tcda *TrustCenterDocumentAccess) LoadByTrustCenterAccessIDAndDocumentID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	documentID gid.GID,
) error {
	q := `
SELECT
    id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
FROM
    trust_center_document_accesses
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND document_id = @document_id
LIMIT 1;
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"document_id":            documentID,
	}
	maps.Copy(args, scope.SQLArguments())

	rows, err := conn.Query(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot query trust center document access: %w", err)
	}

	access, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[TrustCenterDocumentAccess])
	if err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return ErrResourceNotFound
		}

		return fmt.Errorf("cannot collect trust center document access: %w", err)
	}

	*tcda = access

	return nil
}

func (tcda *TrustCenterDocumentAccess) LoadByTrustCenterAccessIDAndReportID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	reportID gid.GID,
) error {
	q := `
SELECT
    id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
FROM
    trust_center_document_accesses
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND report_id = @report_id
LIMIT 1;
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"report_id":              reportID,
	}
	maps.Copy(args, scope.SQLArguments())

	rows, err := conn.Query(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot query trust center document access: %w", err)
	}

	access, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[TrustCenterDocumentAccess])
	if err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return ErrResourceNotFound
		}

		return fmt.Errorf("cannot collect trust center document access: %w", err)
	}

	*tcda = access

	return nil
}

func (tcda *TrustCenterDocumentAccess) Insert(
	ctx context.Context,
	conn pg.Tx,
	scope Scoper,
) error {
	q := `
INSERT INTO trust_center_document_accesses (
    id,
    tenant_id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
) VALUES (
    @id,
    @tenant_id,
    @organization_id,
    @trust_center_access_id,
    @document_id,
    @report_id,
    @trust_center_file_id,
    @status::trust_center_document_access_status,
    @created_at,
    @updated_at
)
`

	args := pgx.StrictNamedArgs{
		"id":                     tcda.ID,
		"tenant_id":              scope.GetTenantID(),
		"organization_id":        tcda.OrganizationID,
		"trust_center_access_id": tcda.TrustCenterAccessID,
		"document_id":            tcda.DocumentID,
		"report_id":              tcda.ReportID,
		"trust_center_file_id":   tcda.TrustCenterFileID,
		"status":                 tcda.Status,
		"created_at":             tcda.CreatedAt,
		"updated_at":             tcda.UpdatedAt,
	}

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		var pgErr *pgconn.PgError
		if errors.As(err, &pgErr) {
			if pgErr.Code == "23505" {
				switch pgErr.ConstraintName {
				case "trust_center_document_accesse_trust_center_access_id_docume_key",
					"trust_center_document_accesse_trust_center_access_id_report_key",
					"trust_center_document_accesses_trust_center_file_id_key":
					return ErrResourceAlreadyExists
				}
			}
		}
		return fmt.Errorf("cannot insert trust center document access: %w", err)
	}

	return nil
}

func (tcda *TrustCenterDocumentAccess) Update(
	ctx context.Context,
	conn pg.Tx,
	scope Scoper,
) error {
	q := `
UPDATE trust_center_document_accesses SET
    status = @status::trust_center_document_access_status,
    updated_at = @updated_at
WHERE
    %s
    AND id = @id
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"id":         tcda.ID,
		"status":     tcda.Status,
		"updated_at": tcda.UpdatedAt,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot update trust center document access: %w", err)
	}

	return nil
}

func (tcda *TrustCenterDocumentAccess) Delete(
	ctx context.Context,
	conn pg.Tx,
	scope Scoper,
) error {
	q := `
DELETE FROM trust_center_document_accesses
WHERE
    %s
    AND id = @id
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"id": tcda.ID,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot delete trust center document access: %w", err)
	}

	return nil
}

func (tcdas *TrustCenterDocumentAccesses) CountByTrustCenterAccessID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
) (int, error) {
	q := `
SELECT
    COUNT(id)
FROM
    trust_center_document_accesses
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
	}
	maps.Copy(args, scope.SQLArguments())

	row := conn.QueryRow(ctx, q, args)

	var count int
	if err := row.Scan(&count); err != nil {
		return 0, fmt.Errorf("cannot scan count: %w", err)
	}

	return count, nil
}

func (tcdas *TrustCenterDocumentAccesses) CountPendingRequestByTrustCenterAccessID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
) (int, error) {
	q := `
SELECT
    COUNT(id)
FROM
    trust_center_document_accesses
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND status = 'REQUESTED'::trust_center_document_access_status
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
	}
	maps.Copy(args, scope.SQLArguments())

	row := conn.QueryRow(ctx, q, args)

	var count int
	if err := row.Scan(&count); err != nil {
		return 0, fmt.Errorf("cannot scan count: %w", err)
	}

	return count, nil
}

func (tcdas *TrustCenterDocumentAccesses) CountActiveByTrustCenterAccessID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
) (int, error) {
	q := `
SELECT
    COUNT(id)
FROM
    trust_center_document_accesses
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND status = 'GRANTED'::trust_center_document_access_status
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
	}
	maps.Copy(args, scope.SQLArguments())

	row := conn.QueryRow(ctx, q, args)

	var count int
	if err := row.Scan(&count); err != nil {
		return 0, fmt.Errorf("cannot scan count: %w", err)
	}

	return count, nil
}

func (tcdas *TrustCenterDocumentAccesses) LoadAvailableByTrustCenterAccessID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	cursor *page.Cursor[TrustCenterDocumentAccessOrderField],
) error {
	q := `
WITH organization AS (
    SELECT tc.organization_id
    FROM trust_center_accesses tca
    INNER JOIN trust_centers tc ON tca.trust_center_id = tc.id
    WHERE tca.tenant_id = @tenant_id
        AND tca.id = @trust_center_access_id
),
tenant_organization AS (
    SELECT o.id AS organization_id
    FROM organizations o
    WHERE %s
),
all_items AS (
    SELECT
        d.id AS item_id,
        d.id AS document_id,
        NULL::text AS report_id,
        NULL::text AS trust_center_file_id,
        d.created_at AS item_created_at,
        d.updated_at AS item_updated_at
    FROM documents d, tenant_organization o
    WHERE d.organization_id = o.organization_id
        AND d.deleted_at IS NULL
        AND d.trust_center_visibility = 'PRIVATE'::trust_center_visibility

    UNION ALL

    SELECT
        r.report_id AS item_id,
        NULL::text AS document_id,
        r.report_id AS report_id,
        NULL::text AS trust_center_file_id,
        r.created_at AS item_created_at,
        r.updated_at AS item_updated_at
    FROM audits r, tenant_organization o
    WHERE r.organization_id = o.organization_id
        AND r.trust_center_visibility = 'PRIVATE'::trust_center_visibility
        AND r.report_id IS NOT NULL

    UNION ALL

    SELECT
        tcf.id AS item_id,
        NULL::text AS document_id,
        NULL::text AS report_id,
        tcf.id AS trust_center_file_id,
        tcf.created_at AS item_created_at,
        tcf.updated_at AS item_updated_at
    FROM trust_center_files tcf, tenant_organization o
    WHERE tcf.organization_id = o.organization_id
        AND (
            tcf.trust_center_visibility = 'PRIVATE'::trust_center_visibility
            OR tcf.trust_center_visibility = 'NONE'::trust_center_visibility
        )
),
final_items AS (
  SELECT
      COALESCE(tcda.id, ai.item_id) AS id,
      tcda.tenant_id,
      (SELECT organization_id FROM organization) AS organization_id,
      @trust_center_access_id AS trust_center_access_id,
      ai.document_id,
      ai.report_id,
      ai.trust_center_file_id,
      COALESCE(tcda.status, 'REQUESTED'::trust_center_document_access_status) AS status,
      COALESCE(tcda.created_at, ai.item_created_at) AS created_at,
      COALESCE(tcda.updated_at, ai.item_updated_at) AS updated_at
  FROM all_items ai
  LEFT JOIN trust_center_document_accesses tcda ON (
      tcda.trust_center_access_id = @trust_center_access_id
      AND (
          (tcda.document_id = ai.document_id AND ai.document_id IS NOT NULL)
          OR (tcda.report_id = ai.report_id AND ai.report_id IS NOT NULL)
          OR (tcda.trust_center_file_id = ai.trust_center_file_id AND ai.trust_center_file_id IS NOT NULL)
      )
  )
)
SELECT
    id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
FROM final_items
WHERE %s
`

	q = fmt.Sprintf(q, scope.SQLFragment(), cursor.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
	}
	maps.Copy(args, scope.SQLArguments())
	maps.Copy(args, cursor.SQLArguments())

	rows, err := conn.Query(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot query trust center document accesses: %w", err)
	}

	accesses, err := pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[TrustCenterDocumentAccess])
	if err != nil {
		return fmt.Errorf("cannot collect trust center document accesses: %w", err)
	}

	*tcdas = accesses

	return nil
}

func (tcdas *TrustCenterDocumentAccesses) LoadAllByTrustCenterAccessID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
) error {
	q := `
SELECT
    id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
FROM
    trust_center_document_accesses
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
ORDER BY id ASC
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
	}
	maps.Copy(args, scope.SQLArguments())

	rows, err := conn.Query(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot query trust center document accesses: %w", err)
	}

	accesses, err := pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[TrustCenterDocumentAccess])
	if err != nil {
		return fmt.Errorf("cannot collect trust center document accesses: %w", err)
	}

	*tcdas = accesses

	return nil
}

func GrantByDocumentIDs(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	documentIDs []gid.GID,
	updatedAt time.Time,
) error {
	q := `
UPDATE trust_center_document_accesses
SET status = 'GRANTED'::trust_center_document_access_status, updated_at = @updated_at
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND document_id = ANY(@document_ids)
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"document_ids":           documentIDs,
		"updated_at":             updatedAt,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot grant trust center document accesses by document IDs: %w", err)
	}

	return nil
}

func RejectOrRevokeByDocumentIDs(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	documentIDs []gid.GID,
	updatedAt time.Time,
) error {
	q := `
UPDATE trust_center_document_accesses
SET
    status = CASE
        WHEN status = 'GRANTED'::trust_center_document_access_status THEN 'REVOKED'::trust_center_document_access_status
        ELSE 'REJECTED'::trust_center_document_access_status
    END,
    updated_at = @updated_at
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND document_id = ANY(@document_ids)
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"document_ids":           documentIDs,
		"updated_at":             updatedAt,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot reject trust center document accesses by document IDs: %w", err)
	}

	return nil
}

func GrantByReportIDs(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	reportIDs []gid.GID,
	updatedAt time.Time,
) error {
	q := `
UPDATE trust_center_document_accesses
SET status = 'GRANTED'::trust_center_document_access_status, updated_at = @updated_at
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND report_id = ANY(@report_ids)
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"report_ids":             reportIDs,
		"updated_at":             updatedAt,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot grant trust center document accesses by report IDs: %w", err)
	}

	return nil
}

func RejectOrRevokeByReportIDs(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	reportIDs []gid.GID,
	updatedAt time.Time,
) error {
	q := `
UPDATE trust_center_document_accesses
SET
    status = CASE
        WHEN status = 'GRANTED'::trust_center_document_access_status THEN 'REVOKED'::trust_center_document_access_status
        ELSE 'REJECTED'::trust_center_document_access_status
    END,
    updated_at = @updated_at
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND report_id = ANY(@report_ids)
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"report_ids":             reportIDs,
		"updated_at":             updatedAt,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot reject trust center document accesses by report IDs: %w", err)
	}

	return nil
}

type MergeTrustCenterDocumentAccessesData struct {
	ID     gid.GID                         `json:"id"`
	Status TrustCenterDocumentAccessStatus `json:"status"`
}

func (tcdas TrustCenterDocumentAccesses) MergeDocumentAccesses(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	organizationID gid.GID,
	trustCenterAccessID gid.GID,
	data []MergeTrustCenterDocumentAccessesData,
) error {
	q := `
WITH data AS (
    SELECT
        t.*
    FROM json_to_recordset(@data)
        AS t(
            id text,
            status trust_center_document_access_status
        )
)
MERGE INTO trust_center_document_accesses AS tcda
USING data
    ON data.id = tcda.document_id
    AND tcda.tenant_id = @tenant_id
    AND tcda.trust_center_access_id = @trust_center_access_id
WHEN MATCHED
    THEN UPDATE SET status = data.status, updated_at = @now::timestamptz
WHEN NOT MATCHED BY SOURCE
    AND tcda.tenant_id = @tenant_id
    AND tcda.trust_center_access_id = @trust_center_access_id
    AND tcda.document_id IS NOT NULL
    THEN DELETE
WHEN NOT MATCHED
    THEN INSERT (
        id,
        tenant_id,
        organization_id,
        trust_center_access_id,
        document_id,
        report_id,
        trust_center_file_id,
        status,
        created_at,
        updated_at
    )
    VALUES (
        generate_gid(decode_base64_unpadded(@tenant_id), @trust_center_document_access_entity_type),
        @tenant_id,
        @organization_id,
        @trust_center_access_id,
        data.id,
        NULL,
        NULL,
        data.status,
        @now::timestamptz,
        @now::timestamptz
    )
`

	args := pgx.StrictNamedArgs{
		"trust_center_document_access_entity_type": TrustCenterDocumentAccessEntityType,
		"tenant_id":              scope.GetTenantID(),
		"trust_center_access_id": trustCenterAccessID,
		"organization_id":        organizationID,
		"now":                    time.Now(),
		"data":                   data,
	}

	if _, err := conn.Exec(ctx, q, args); err != nil {
		return err
	}

	return nil
}

func (tcdas TrustCenterDocumentAccesses) BulkInsertDocumentAccesses(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	organizationID gid.GID,
	documentIDs []gid.GID,
	status TrustCenterDocumentAccessStatus,
	createdAt time.Time,
) error {
	if len(documentIDs) == 0 {
		return nil
	}

	q := `
WITH document_access_data AS (
    SELECT
        generate_gid(decode_base64_unpadded(@tenant_id), @trust_center_document_access_entity_type) AS id,
        @tenant_id AS tenant_id,
        @organization_id AS organization_id,
        @trust_center_access_id AS trust_center_access_id,
        unnest(@document_ids::text[]) AS document_id,
        null::text AS report_id,
        null::text AS trust_center_file_id,
        @status::trust_center_document_access_status AS status,
        @created_at::timestamptz AS created_at,
        @updated_at::timestamptz AS updated_at
)
INSERT INTO trust_center_document_accesses (
    id,
    tenant_id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
)
SELECT * FROM document_access_data
ON CONFLICT DO NOTHING
`

	args := pgx.StrictNamedArgs{
		"tenant_id":              scope.GetTenantID(),
		"organization_id":        organizationID,
		"trust_center_access_id": trustCenterAccessID,
		"document_ids":           documentIDs,
		"trust_center_document_access_entity_type": TrustCenterDocumentAccessEntityType,
		"status":     status,
		"created_at": createdAt,
		"updated_at": createdAt,
	}

	if _, err := conn.Exec(ctx, q, args); err != nil {
		return fmt.Errorf("cannot bulk insert trust center document accesses: %w", err)
	}

	return nil
}

func (tcdas TrustCenterDocumentAccesses) MergeReportAccesses(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	organizationID gid.GID,
	trustCenterAccessID gid.GID,
	data []MergeTrustCenterDocumentAccessesData,
) error {
	q := `
WITH data AS (
    SELECT
        t.*
    FROM json_to_recordset(@data)
        AS t(
            id text,
            status trust_center_document_access_status
        )
)
MERGE INTO trust_center_document_accesses AS tcda
USING data
    ON data.id = tcda.report_id
    AND tcda.tenant_id = @tenant_id
    AND tcda.trust_center_access_id = @trust_center_access_id
WHEN MATCHED
    THEN UPDATE SET status = data.status, updated_at = @now::timestamptz
WHEN NOT MATCHED BY SOURCE
    AND tcda.tenant_id = @tenant_id
    AND tcda.trust_center_access_id = @trust_center_access_id
    AND tcda.report_id IS NOT NULL
    THEN DELETE
WHEN NOT MATCHED
    THEN INSERT (
        id,
        tenant_id,
        organization_id,
        trust_center_access_id,
        document_id,
        report_id,
        trust_center_file_id,
        status,
        created_at,
        updated_at
    )
    VALUES (
        generate_gid(decode_base64_unpadded(@tenant_id), @trust_center_document_access_entity_type),
        @tenant_id,
        @organization_id,
        @trust_center_access_id,
        NULL,
        data.id,
        NULL,
        data.status,
        @now::timestamptz,
        @now::timestamptz
    )
`

	args := pgx.StrictNamedArgs{
		"trust_center_document_access_entity_type": TrustCenterDocumentAccessEntityType,
		"tenant_id":              scope.GetTenantID(),
		"trust_center_access_id": trustCenterAccessID,
		"organization_id":        organizationID,
		"now":                    time.Now(),
		"data":                   data,
	}

	if _, err := conn.Exec(ctx, q, args); err != nil {
		return err
	}

	return nil
}

func (tcdas TrustCenterDocumentAccesses) BulkInsertReportAccesses(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	organizationID gid.GID,
	reportIDs []gid.GID,
	status TrustCenterDocumentAccessStatus,
	createdAt time.Time,
) error {
	if len(reportIDs) == 0 {
		return nil
	}

	q := `
WITH report_access_data AS (
    SELECT
        generate_gid(decode_base64_unpadded(@tenant_id), @trust_center_document_access_entity_type) AS id,
        @tenant_id AS tenant_id,
        @organization_id AS organization_id,
        @trust_center_access_id AS trust_center_access_id,
        null::text AS document_id,
        unnest(@report_ids::text[]) AS report_id,
        null::text AS trust_center_file_id,
        @status::trust_center_document_access_status AS status,
        @created_at::timestamptz AS created_at,
        @updated_at::timestamptz AS updated_at
)
INSERT INTO trust_center_document_accesses (
    id,
    tenant_id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
)
SELECT * FROM report_access_data
ON CONFLICT DO NOTHING
`

	args := pgx.StrictNamedArgs{
		"tenant_id":       scope.GetTenantID(),
		"organization_id": organizationID,
		"trust_center_document_access_entity_type": TrustCenterDocumentAccessEntityType,
		"trust_center_access_id":                   trustCenterAccessID,
		"report_ids":                               reportIDs,
		"status":                                   status,
		"created_at":                               createdAt,
		"updated_at":                               createdAt,
	}

	if _, err := conn.Exec(ctx, q, args); err != nil {
		return fmt.Errorf("cannot bulk insert trust center report accesses: %w", err)
	}

	return nil
}

func (tcda *TrustCenterDocumentAccess) LoadByTrustCenterAccessIDAndTrustCenterFileID(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	trustCenterFileID gid.GID,
) error {
	q := `
SELECT
    id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
FROM
    trust_center_document_accesses
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND trust_center_file_id = @trust_center_file_id
LIMIT 1;
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"trust_center_file_id":   trustCenterFileID,
	}
	maps.Copy(args, scope.SQLArguments())

	rows, err := conn.Query(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot query trust center document access: %w", err)
	}

	access, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[TrustCenterDocumentAccess])
	if err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return ErrResourceNotFound
		}

		return fmt.Errorf("cannot collect trust center document access: %w", err)
	}

	*tcda = access

	return nil
}

func GrantByTrustCenterFileIDs(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	trustCenterFileIDs []gid.GID,
	updatedAt time.Time,
) error {
	q := `
UPDATE trust_center_document_accesses
SET status = 'GRANTED'::trust_center_document_access_status, updated_at = @updated_at
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND trust_center_file_id = ANY(@trust_center_file_ids)
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"trust_center_file_ids":  trustCenterFileIDs,
		"updated_at":             updatedAt,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot grant trust center document accesses by trust center file IDs: %w", err)
	}

	return nil
}

func RejectOrRevokeByTrustCenterFileIDs(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	trustCenterFileIDs []gid.GID,
	updatedAt time.Time,
) error {
	q := `
UPDATE trust_center_document_accesses
SET
    status = CASE
        WHEN status = 'GRANTED'::trust_center_document_access_status THEN 'REVOKED'::trust_center_document_access_status
        ELSE 'REJECTED'::trust_center_document_access_status
    END,
    updated_at = @updated_at
WHERE
    %s
    AND trust_center_access_id = @trust_center_access_id
    AND trust_center_file_id = ANY(@trust_center_file_ids)
`

	q = fmt.Sprintf(q, scope.SQLFragment())

	args := pgx.StrictNamedArgs{
		"trust_center_access_id": trustCenterAccessID,
		"trust_center_file_ids":  trustCenterFileIDs,
		"updated_at":             updatedAt,
	}
	maps.Copy(args, scope.SQLArguments())

	_, err := conn.Exec(ctx, q, args)
	if err != nil {
		return fmt.Errorf("cannot reject trust center document accesses by trust center file IDs: %w", err)
	}

	return nil
}

func (tcdas TrustCenterDocumentAccesses) MergeTrustCenterFileAccesses(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	organizationID gid.GID,
	trustCenterAccessID gid.GID,
	data []MergeTrustCenterDocumentAccessesData,
) error {
	q := `
WITH data AS (
    SELECT
        t.*
    FROM json_to_recordset(@data)
        AS t(
            id text,
            status trust_center_document_access_status
        )
)
MERGE INTO trust_center_document_accesses AS tcda
USING data
    ON data.id = tcda.trust_center_file_id
    AND tcda.tenant_id = @tenant_id
    AND tcda.trust_center_access_id = @trust_center_access_id
WHEN MATCHED
    THEN UPDATE SET status = data.status, updated_at = @now::timestamptz
WHEN NOT MATCHED BY SOURCE
    AND tcda.tenant_id = @tenant_id
    AND tcda.trust_center_access_id = @trust_center_access_id
    AND tcda.trust_center_file_id IS NOT NULL
    THEN DELETE
WHEN NOT MATCHED
    THEN INSERT (
        id,
        tenant_id,
        organization_id,
        trust_center_access_id,
        document_id,
        report_id,
        trust_center_file_id,
        status,
        created_at,
        updated_at
    )
    VALUES (
        generate_gid(decode_base64_unpadded(@tenant_id), @trust_center_document_access_entity_type),
        @tenant_id,
        @organization_id,
        @trust_center_access_id,
        NULL,
        NULL,
        data.id,
        data.status,
        @now::timestamptz,
        @now::timestamptz
    )
`

	args := pgx.StrictNamedArgs{
		"trust_center_document_access_entity_type": TrustCenterDocumentAccessEntityType,
		"tenant_id":              scope.GetTenantID(),
		"trust_center_access_id": trustCenterAccessID,
		"organization_id":        organizationID,
		"now":                    time.Now(),
		"data":                   data,
	}

	if _, err := conn.Exec(ctx, q, args); err != nil {
		return err
	}

	return nil
}

func (tcdas TrustCenterDocumentAccesses) BulkInsertTrustCenterFileAccesses(
	ctx context.Context,
	conn pg.Querier,
	scope Scoper,
	trustCenterAccessID gid.GID,
	organizationID gid.GID,
	trustCenterFileIDs []gid.GID,
	status TrustCenterDocumentAccessStatus,
	createdAt time.Time,
) error {
	q := `
WITH trust_center_file_access_data AS (
    SELECT
        generate_gid(decode_base64_unpadded(@tenant_id), @trust_center_document_access_entity_type) AS id,
        @tenant_id AS tenant_id,
        @organization_id AS organization_id,
        @trust_center_access_id AS trust_center_access_id,
        null::text AS document_id,
        null::text AS report_id,
        unnest(@trust_center_file_ids::text[]) AS trust_center_file_id,
        @status::trust_center_document_access_status AS status,
        @created_at::timestamptz AS created_at,
        @updated_at::timestamptz AS updated_at
)
INSERT INTO trust_center_document_accesses (
    id,
    tenant_id,
    organization_id,
    trust_center_access_id,
    document_id,
    report_id,
    trust_center_file_id,
    status,
    created_at,
    updated_at
)
SELECT * FROM trust_center_file_access_data
ON CONFLICT DO NOTHING
`

	args := pgx.StrictNamedArgs{
		"tenant_id":       scope.GetTenantID(),
		"organization_id": organizationID,
		"trust_center_document_access_entity_type": TrustCenterDocumentAccessEntityType,
		"trust_center_access_id":                   trustCenterAccessID,
		"trust_center_file_ids":                    trustCenterFileIDs,
		"status":                                   status,
		"created_at":                               createdAt,
		"updated_at":                               createdAt,
	}

	if _, err := conn.Exec(ctx, q, args); err != nil {
		return fmt.Errorf("cannot bulk insert trust center file accesses: %w", err)
	}

	return nil
}