Add assets

Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
2025-05-30 15:07:57 -07:00
parent b315be96bc
commit fa1cd6fb51
38 changed files with 9121 additions and 7 deletions
--- a/pkg/coredata/asset.go
+++ b/pkg/coredata/asset.go
@@ -0,0 +1,427 @@
+// Copyright (c) 2025 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"
+	"fmt"
+	"maps"
+	"time"
+
+	"github.com/getprobo/probo/pkg/gid"
+	"github.com/getprobo/probo/pkg/page"
+	"github.com/jackc/pgx/v5"
+	"go.gearno.de/kit/pg"
+)
+
+type Asset struct {
+	ID              gid.GID        `db:"id"`
+	Name            string         `db:"name"`
+	Amount          int            `db:"amount"`
+	OwnerID         gid.GID        `db:"owner_id"`
+	OrganizationID  gid.GID        `db:"organization_id"`
+	Criticity       CriticityLevel `db:"criticity"`
+	AssetType       AssetType      `db:"asset_type"`
+	DataTypesStored string         `db:"data_types_stored"`
+	CreatedAt       time.Time      `db:"created_at"`
+	UpdatedAt       time.Time      `db:"updated_at"`
+}
+
+func (a *Asset) CursorKey(field AssetOrderField) page.CursorKey {
+	switch field {
+	case AssetOrderFieldCreatedAt:
+		return page.NewCursorKey(a.ID, a.CreatedAt)
+	case AssetOrderFieldAmount:
+		return page.NewCursorKey(a.ID, a.Amount)
+	case AssetOrderFieldCriticity:
+		return page.NewCursorKey(a.ID, a.Criticity)
+	}
+
+	panic(fmt.Sprintf("unsupported order by: %s", field))
+}
+
+type Assets []*Asset
+
+func (a *Asset) LoadByID(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+	assetID gid.GID,
+) error {
+	q := `
+SELECT
+	id,
+	name,
+	organization_id,
+	owner_id,
+	amount,
+	criticity,
+	asset_type,
+	data_types_stored,
+	created_at,
+	updated_at
+FROM
+	assets
+WHERE
+	%s
+	AND id = @asset_id
+LIMIT 1;
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"asset_id": assetID}
+	maps.Copy(args, scope.SQLArguments())
+
+	rows, err := conn.Query(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot query assets: %w", err)
+	}
+
+	asset, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[Asset])
+	if err != nil {
+		return fmt.Errorf("cannot collect asset: %w", err)
+	}
+
+	*a = asset
+
+	return nil
+}
+
+func (a *Asset) LoadByOwnerID(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+) error {
+	q := `
+SELECT
+	id,
+	name,
+	organization_id,
+	owner_id,
+	amount,
+	criticity,
+	asset_type,
+	data_types_stored,
+	created_at,
+	updated_at
+FROM
+	assets
+WHERE
+	%s
+	AND owner_id = @owner_id
+LIMIT 1;
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"owner_id": a.OwnerID}
+	maps.Copy(args, scope.SQLArguments())
+
+	rows, err := conn.Query(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot query assets: %w", err)
+	}
+
+	asset, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[Asset])
+	if err != nil {
+		return fmt.Errorf("cannot collect asset: %w", err)
+	}
+
+	*a = asset
+
+	return nil
+}
+
+func (a *Assets) LoadByOrganizationID(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+	organizationID gid.GID,
+	cursor *page.Cursor[AssetOrderField],
+) error {
+	q := `
+SELECT
+	id,
+	name,
+	organization_id,
+	owner_id,
+	amount,
+	criticity,
+	asset_type,
+	data_types_stored,
+	created_at,
+	updated_at
+FROM
+	assets
+WHERE
+	%s
+	AND organization_id = @organization_id
+	AND %s
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment(), cursor.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"organization_id": organizationID}
+	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 assets: %w", err)
+	}
+
+	assets, err := pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[Asset])
+	if err != nil {
+		return fmt.Errorf("cannot collect assets: %w", err)
+	}
+
+	*a = assets
+
+	return nil
+}
+
+func (a *Asset) Insert(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+) error {
+	q := `
+INSERT INTO assets (
+	id,
+	tenant_id,
+	name,
+	organization_id,
+	owner_id,
+	amount,
+	criticity,
+	asset_type,
+	data_types_stored,
+	created_at,
+	updated_at
+) VALUES (
+	@id,
+	@tenant_id,
+	@name,
+	@organization_id,
+	@owner_id,
+	@amount,
+	@criticity,
+	@asset_type,
+	@data_types_stored,
+	@created_at,
+	@updated_at
+)
+`
+
+	args := pgx.StrictNamedArgs{
+		"id":                a.ID,
+		"tenant_id":         scope.GetTenantID(),
+		"organization_id":   a.OrganizationID,
+		"name":              a.Name,
+		"owner_id":          a.OwnerID,
+		"amount":            a.Amount,
+		"criticity":         a.Criticity,
+		"asset_type":        a.AssetType,
+		"data_types_stored": a.DataTypesStored,
+		"created_at":        a.CreatedAt,
+		"updated_at":        a.UpdatedAt,
+	}
+
+	_, err := conn.Exec(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot insert asset: %w", err)
+	}
+
+	return nil
+}
+
+func (a *Asset) Update(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+) error {
+	q := `
+UPDATE assets
+SET
+	name = @name,
+	owner_id = @owner_id,
+	amount = @amount,
+	criticity = @criticity,
+	asset_type = @asset_type,
+	data_types_stored = @data_types_stored,
+	updated_at = @updated_at
+WHERE
+	%s
+	AND id = @id
+RETURNING
+	id,
+	name,
+	organization_id,
+	owner_id,
+	amount,
+	criticity,
+	asset_type,
+	data_types_stored,
+	created_at,
+	updated_at
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment())
+
+	args := pgx.StrictNamedArgs{
+		"id":                a.ID,
+		"name":              a.Name,
+		"owner_id":          a.OwnerID,
+		"amount":            a.Amount,
+		"criticity":         a.Criticity,
+		"asset_type":        a.AssetType,
+		"data_types_stored": a.DataTypesStored,
+		"updated_at":        time.Now(),
+	}
+	maps.Copy(args, scope.SQLArguments())
+
+	rows, err := conn.Query(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot update asset: %w", err)
+	}
+
+	asset, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[Asset])
+	if err != nil {
+		return fmt.Errorf("cannot collect updated asset: %w", err)
+	}
+
+	*a = asset
+
+	return nil
+}
+
+func (a *Asset) Delete(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+) error {
+	q := `
+DELETE FROM assets
+WHERE
+	%s
+	AND id = @id
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"id": a.ID}
+	maps.Copy(args, scope.SQLArguments())
+
+	_, err := conn.Exec(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot delete asset: %w", err)
+	}
+
+	return nil
+}
+
+type AssetVendor struct {
+	AssetID   int64     `db:"asset_id"`
+	VendorID  int64     `db:"vendor_id"`
+	CreatedAt time.Time `db:"created_at"`
+}
+
+func (a *Asset) CreateWithVendors(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+	organization *Organization,
+	vendorIDs []gid.GID,
+	now time.Time,
+) error {
+	if err := organization.LoadByID(ctx, conn, scope, a.OrganizationID); err != nil {
+		return fmt.Errorf("cannot load organization %q: %w", a.OrganizationID, err)
+	}
+
+	if err := a.Insert(ctx, conn, scope); err != nil {
+		return fmt.Errorf("cannot insert asset: %w", err)
+	}
+
+	if len(vendorIDs) > 0 {
+		for _, vendorID := range vendorIDs {
+			_, err := conn.Exec(ctx, `
+				INSERT INTO asset_vendors (tenant_id, asset_id, vendor_id, created_at)
+				VALUES ($1, $2, $3, $4)
+			`, scope.GetTenantID(), a.ID, vendorID, now)
+			if err != nil {
+				return fmt.Errorf("cannot insert asset vendor: %w", err)
+			}
+		}
+	}
+
+	return nil
+}
+
+func (a *Asset) UpdateWithVendors(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+	vendorIDs []gid.GID,
+	now time.Time,
+) error {
+	existing := &Asset{}
+	if err := existing.LoadByID(ctx, conn, scope, a.ID); err != nil {
+		return fmt.Errorf("cannot load asset: %w", err)
+	}
+
+	a.OrganizationID = existing.OrganizationID
+	a.CreatedAt = existing.CreatedAt
+	a.UpdatedAt = now
+
+	if err := a.Update(ctx, conn, scope); err != nil {
+		return fmt.Errorf("cannot update asset: %w", err)
+	}
+
+	_, err := conn.Exec(ctx, `
+		DELETE FROM asset_vendors
+		WHERE tenant_id = $1 AND asset_id = $2
+	`, scope.GetTenantID(), a.ID)
+	if err != nil {
+		return fmt.Errorf("cannot delete asset vendors: %w", err)
+	}
+
+	if len(vendorIDs) > 0 {
+		for _, vendorID := range vendorIDs {
+			_, err := conn.Exec(ctx, `
+				INSERT INTO asset_vendors (tenant_id, asset_id, vendor_id, created_at)
+				VALUES ($1, $2, $3, $4)
+			`, scope.GetTenantID(), a.ID, vendorID, now)
+			if err != nil {
+				return fmt.Errorf("cannot insert asset vendor: %w", err)
+			}
+		}
+	}
+
+	return nil
+}
+
+// UpdateWithVendorsTx updates an asset and its vendor relationships in a single transaction
+func (a *Asset) UpdateWithVendorsTx(
+	ctx context.Context,
+	db *pg.Client,
+	scope Scoper,
+	vendorIDs []gid.GID,
+	now time.Time,
+) error {
+	return db.WithTx(ctx, func(conn pg.Conn) error {
+		return a.UpdateWithVendors(ctx, conn, scope, vendorIDs, now)
+	})
+}