Add nonconformity registries

Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
This commit is contained in:
Sacha Al Himdani
2025-08-14 13:37:38 +02:00
parent 210a05b54d
commit 00c6b9fde6
32 changed files with 8109 additions and 39 deletions

View File

@@ -151,6 +151,22 @@ enum AuditState
)
}
enum NonconformityRegistryStatus
@goModel(model: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryStatus") {
OPEN
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryStatusOpen"
)
IN_PROGRESS
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryStatusInProgress"
)
CLOSED
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryStatusClosed"
)
}
# Order Field Enums
enum UserOrderField
@goModel(model: "github.com/getprobo/probo/pkg/coredata.UserOrderField") {
@@ -584,6 +600,30 @@ enum AuditOrderField
)
}
enum NonconformityRegistryOrderField
@goModel(model: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryOrderField") {
CREATED_AT
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryOrderFieldCreatedAt"
)
REFERENCE_ID
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryOrderFieldReferenceId"
)
DATE_IDENTIFIED
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryOrderFieldDateIdentified"
)
DUE_DATE
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryOrderFieldDueDate"
)
STATUS
@goEnum(
value: "github.com/getprobo/probo/pkg/coredata.NonconformityRegistryOrderFieldStatus"
)
}
enum TrustCenterAccessOrderField
@goModel(model: "github.com/getprobo/probo/pkg/coredata.TrustCenterAccessOrderField") {
CREATED_AT
@@ -673,6 +713,14 @@ input AuditOrder
field: AuditOrderField!
}
input NonconformityRegistryOrder
@goModel(
model: "github.com/getprobo/probo/pkg/server/api/console/v1/types.NonconformityRegistryOrderBy"
) {
direction: OrderDirection!
field: NonconformityRegistryOrderField!
}
input TrustCenterAccessOrder
@goModel(
model: "github.com/getprobo/probo/pkg/server/api/console/v1/types.TrustCenterAccessOrderBy"
@@ -888,6 +936,14 @@ type Organization implements Node {
orderBy: AuditOrder
): AuditConnection! @goField(forceResolver: true)
nonconformityRegistries(
first: Int
after: CursorKey
last: Int
before: CursorKey
orderBy: NonconformityRegistryOrder
): NonconformityRegistryConnection! @goField(forceResolver: true)
trustCenter: TrustCenter @goField(forceResolver: true)
createdAt: Datetime!
@@ -1280,6 +1336,23 @@ type Audit implements Node {
updatedAt: Datetime!
}
type NonconformityRegistry implements Node {
id: ID!
organization: Organization! @goField(forceResolver: true)
referenceId: String!
description: String
audit: Audit! @goField(forceResolver: true)
dateIdentified: Datetime
rootCause: String!
correctiveAction: String
owner: People! @goField(forceResolver: true)
dueDate: Datetime
status: NonconformityRegistryStatus!
effectivenessCheck: String
createdAt: Datetime!
updatedAt: Datetime!
}
type Report implements Node {
id: ID!
objectKey: String!
@@ -1563,6 +1636,20 @@ type AuditEdge {
node: Audit!
}
type NonconformityRegistryConnection
@goModel(
model: "github.com/getprobo/probo/pkg/server/api/console/v1/types.NonconformityRegistryConnection"
) {
totalCount: Int! @goField(forceResolver: true)
edges: [NonconformityRegistryEdge!]!
pageInfo: PageInfo!
}
type NonconformityRegistryEdge {
cursor: CursorKey!
node: NonconformityRegistry!
}
# Root Types
type Query {
node(id: ID!): Node!
@@ -1782,6 +1869,17 @@ type Mutation {
deleteAudit(input: DeleteAuditInput!): DeleteAuditPayload!
uploadAuditReport(input: UploadAuditReportInput!): UploadAuditReportPayload!
deleteAuditReport(input: DeleteAuditReportInput!): DeleteAuditReportPayload!
# Nonconformity Registry mutations
createNonconformityRegistry(
input: CreateNonconformityRegistryInput!
): CreateNonconformityRegistryPayload!
updateNonconformityRegistry(
input: UpdateNonconformityRegistryInput!
): UpdateNonconformityRegistryPayload!
deleteNonconformityRegistry(
input: DeleteNonconformityRegistryInput!
): DeleteNonconformityRegistryPayload!
}
# Input Types
@@ -2243,6 +2341,39 @@ input DeleteAuditReportInput {
auditId: ID!
}
# Nonconformity Registry input types
input CreateNonconformityRegistryInput {
organizationId: ID!
referenceId: String!
description: String
auditId: ID!
dateIdentified: Datetime
rootCause: String!
correctiveAction: String
ownerId: ID!
dueDate: Datetime
status: NonconformityRegistryStatus!
effectivenessCheck: String
}
input UpdateNonconformityRegistryInput {
id: ID!
referenceId: String
description: String
dateIdentified: Datetime
rootCause: String
correctiveAction: String
ownerId: ID
auditId: ID
dueDate: Datetime
status: NonconformityRegistryStatus
effectivenessCheck: String
}
input DeleteNonconformityRegistryInput {
nonconformityRegistryId: ID!
}
# Payload Types
type CreateOrganizationPayload {
organizationEdge: OrganizationEdge!
@@ -2911,3 +3042,16 @@ type UploadAuditReportPayload {
type DeleteAuditReportPayload {
audit: Audit!
}
# Nonconformity Registry payload types
type CreateNonconformityRegistryPayload {
nonconformityRegistryEdge: NonconformityRegistryEdge!
}
type UpdateNonconformityRegistryPayload {
nonconformityRegistry: NonconformityRegistry!
}
type DeleteNonconformityRegistryPayload {
deletedNonconformityRegistryId: ID!
}

File diff suppressed because it is too large Load Diff

View File

@@ -0,0 +1,76 @@
// 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 types
import (
"github.com/getprobo/probo/pkg/coredata"
"github.com/getprobo/probo/pkg/gid"
"github.com/getprobo/probo/pkg/page"
)
type (
NonconformityRegistryOrderBy OrderBy[coredata.NonconformityRegistryOrderField]
NonconformityRegistryConnection struct {
TotalCount int
Edges []*NonconformityRegistryEdge
PageInfo PageInfo
Resolver any
ParentID gid.GID
}
)
func NewNonconformityRegistryConnection(
p *page.Page[*coredata.NonconformityRegistry, coredata.NonconformityRegistryOrderField],
parentType any,
parentID gid.GID,
) *NonconformityRegistryConnection {
edges := make([]*NonconformityRegistryEdge, len(p.Data))
for i, registry := range p.Data {
edges[i] = NewNonconformityRegistryEdge(registry, p.Cursor.OrderBy.Field)
}
return &NonconformityRegistryConnection{
Edges: edges,
PageInfo: *NewPageInfo(p),
Resolver: parentType,
ParentID: parentID,
}
}
func NewNonconformityRegistry(nr *coredata.NonconformityRegistry) *NonconformityRegistry {
return &NonconformityRegistry{
ID: nr.ID,
ReferenceID: nr.ReferenceID,
Description: nr.Description,
DateIdentified: nr.DateIdentified,
RootCause: nr.RootCause,
CorrectiveAction: nr.CorrectiveAction,
DueDate: nr.DueDate,
Status: nr.Status,
EffectivenessCheck: nr.EffectivenessCheck,
CreatedAt: nr.CreatedAt,
UpdatedAt: nr.UpdatedAt,
}
}
func NewNonconformityRegistryEdge(nr *coredata.NonconformityRegistry, orderField coredata.NonconformityRegistryOrderField) *NonconformityRegistryEdge {
return &NonconformityRegistryEdge{
Node: NewNonconformityRegistry(nr),
Cursor: nr.CursorKey(orderField),
}
}

View File

@@ -306,6 +306,24 @@ type CreateMeasurePayload struct {
MeasureEdge *MeasureEdge `json:"measureEdge"`
}
type CreateNonconformityRegistryInput struct {
OrganizationID gid.GID `json:"organizationId"`
ReferenceID string `json:"referenceId"`
Description *string `json:"description,omitempty"`
AuditID gid.GID `json:"auditId"`
DateIdentified *time.Time `json:"dateIdentified,omitempty"`
RootCause string `json:"rootCause"`
CorrectiveAction *string `json:"correctiveAction,omitempty"`
OwnerID gid.GID `json:"ownerId"`
DueDate *time.Time `json:"dueDate,omitempty"`
Status coredata.NonconformityRegistryStatus `json:"status"`
EffectivenessCheck *string `json:"effectivenessCheck,omitempty"`
}
type CreateNonconformityRegistryPayload struct {
NonconformityRegistryEdge *NonconformityRegistryEdge `json:"nonconformityRegistryEdge"`
}
type CreateOrganizationInput struct {
Name string `json:"name"`
}
@@ -571,6 +589,14 @@ type DeleteMeasurePayload struct {
DeletedMeasureID gid.GID `json:"deletedMeasureId"`
}
type DeleteNonconformityRegistryInput struct {
NonconformityRegistryID gid.GID `json:"nonconformityRegistryId"`
}
type DeleteNonconformityRegistryPayload struct {
DeletedNonconformityRegistryID gid.GID `json:"deletedNonconformityRegistryId"`
}
type DeleteOrganizationInput struct {
OrganizationID gid.GID `json:"organizationId"`
}
@@ -895,26 +921,52 @@ type MeasureFilter struct {
type Mutation struct {
}
type NonconformityRegistry struct {
ID gid.GID `json:"id"`
Organization *Organization `json:"organization"`
ReferenceID string `json:"referenceId"`
Description *string `json:"description,omitempty"`
Audit *Audit `json:"audit"`
DateIdentified *time.Time `json:"dateIdentified,omitempty"`
RootCause string `json:"rootCause"`
CorrectiveAction *string `json:"correctiveAction,omitempty"`
Owner *People `json:"owner"`
DueDate *time.Time `json:"dueDate,omitempty"`
Status coredata.NonconformityRegistryStatus `json:"status"`
EffectivenessCheck *string `json:"effectivenessCheck,omitempty"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
}
func (NonconformityRegistry) IsNode() {}
func (this NonconformityRegistry) GetID() gid.GID { return this.ID }
type NonconformityRegistryEdge struct {
Cursor page.CursorKey `json:"cursor"`
Node *NonconformityRegistry `json:"node"`
}
type Organization struct {
ID gid.GID `json:"id"`
Name string `json:"name"`
LogoURL *string `json:"logoUrl,omitempty"`
Users *UserConnection `json:"users"`
Connectors *ConnectorConnection `json:"connectors"`
Frameworks *FrameworkConnection `json:"frameworks"`
Controls *ControlConnection `json:"controls"`
Vendors *VendorConnection `json:"vendors"`
Peoples *PeopleConnection `json:"peoples"`
Documents *DocumentConnection `json:"documents"`
Measures *MeasureConnection `json:"measures"`
Risks *RiskConnection `json:"risks"`
Tasks *TaskConnection `json:"tasks"`
Assets *AssetConnection `json:"assets"`
Data *DatumConnection `json:"data"`
Audits *AuditConnection `json:"audits"`
TrustCenter *TrustCenter `json:"trustCenter,omitempty"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
ID gid.GID `json:"id"`
Name string `json:"name"`
LogoURL *string `json:"logoUrl,omitempty"`
Users *UserConnection `json:"users"`
Connectors *ConnectorConnection `json:"connectors"`
Frameworks *FrameworkConnection `json:"frameworks"`
Controls *ControlConnection `json:"controls"`
Vendors *VendorConnection `json:"vendors"`
Peoples *PeopleConnection `json:"peoples"`
Documents *DocumentConnection `json:"documents"`
Measures *MeasureConnection `json:"measures"`
Risks *RiskConnection `json:"risks"`
Tasks *TaskConnection `json:"tasks"`
Assets *AssetConnection `json:"assets"`
Data *DatumConnection `json:"data"`
Audits *AuditConnection `json:"audits"`
NonconformityRegistries *NonconformityRegistryConnection `json:"nonconformityRegistries"`
TrustCenter *TrustCenter `json:"trustCenter,omitempty"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
}
func (Organization) IsNode() {}
@@ -1251,6 +1303,24 @@ type UpdateMeasurePayload struct {
Measure *Measure `json:"measure"`
}
type UpdateNonconformityRegistryInput struct {
ID gid.GID `json:"id"`
ReferenceID *string `json:"referenceId,omitempty"`
Description *string `json:"description,omitempty"`
DateIdentified *time.Time `json:"dateIdentified,omitempty"`
RootCause *string `json:"rootCause,omitempty"`
CorrectiveAction *string `json:"correctiveAction,omitempty"`
OwnerID *gid.GID `json:"ownerId,omitempty"`
AuditID *gid.GID `json:"auditId,omitempty"`
DueDate *time.Time `json:"dueDate,omitempty"`
Status *coredata.NonconformityRegistryStatus `json:"status,omitempty"`
EffectivenessCheck *string `json:"effectivenessCheck,omitempty"`
}
type UpdateNonconformityRegistryPayload struct {
NonconformityRegistry *NonconformityRegistry `json:"nonconformityRegistry"`
}
type UpdateOrganizationInput struct {
OrganizationID gid.GID `json:"organizationId"`
Name *string `json:"name,omitempty"`

View File

@@ -2713,6 +2713,143 @@ func (r *mutationResolver) DeleteAuditReport(ctx context.Context, input types.De
}, nil
}
// CreateNonconformityRegistry is the resolver for the createNonconformityRegistry field.
func (r *mutationResolver) CreateNonconformityRegistry(ctx context.Context, input types.CreateNonconformityRegistryInput) (*types.CreateNonconformityRegistryPayload, error) {
prb := r.ProboService(ctx, input.OrganizationID.TenantID())
req := probo.CreateNonconformityRegistryRequest{
OrganizationID: input.OrganizationID,
ReferenceID: input.ReferenceID,
Description: input.Description,
AuditID: input.AuditID,
DateIdentified: input.DateIdentified,
RootCause: input.RootCause,
CorrectiveAction: input.CorrectiveAction,
OwnerID: input.OwnerID,
DueDate: input.DueDate,
Status: &input.Status,
EffectivenessCheck: input.EffectivenessCheck,
}
registry, err := prb.NonconformityRegistries.Create(ctx, &req)
if err != nil {
return nil, fmt.Errorf("cannot create nonconformity registry: %w", err)
}
return &types.CreateNonconformityRegistryPayload{
NonconformityRegistryEdge: types.NewNonconformityRegistryEdge(registry, coredata.NonconformityRegistryOrderFieldCreatedAt),
}, nil
}
// UpdateNonconformityRegistry is the resolver for the updateNonconformityRegistry field.
func (r *mutationResolver) UpdateNonconformityRegistry(ctx context.Context, input types.UpdateNonconformityRegistryInput) (*types.UpdateNonconformityRegistryPayload, error) {
prb := r.ProboService(ctx, input.ID.TenantID())
req := probo.UpdateNonconformityRegistryRequest{
ID: input.ID,
ReferenceID: input.ReferenceID,
Description: &input.Description,
DateIdentified: &input.DateIdentified,
RootCause: input.RootCause,
CorrectiveAction: &input.CorrectiveAction,
OwnerID: input.OwnerID,
AuditID: input.AuditID,
DueDate: &input.DueDate,
Status: input.Status,
EffectivenessCheck: &input.EffectivenessCheck,
}
registry, err := prb.NonconformityRegistries.Update(ctx, &req)
if err != nil {
return nil, fmt.Errorf("cannot update nonconformity registry: %w", err)
}
return &types.UpdateNonconformityRegistryPayload{
NonconformityRegistry: types.NewNonconformityRegistry(registry),
}, nil
}
// DeleteNonconformityRegistry is the resolver for the deleteNonconformityRegistry field.
func (r *mutationResolver) DeleteNonconformityRegistry(ctx context.Context, input types.DeleteNonconformityRegistryInput) (*types.DeleteNonconformityRegistryPayload, error) {
prb := r.ProboService(ctx, input.NonconformityRegistryID.TenantID())
err := prb.NonconformityRegistries.Delete(ctx, input.NonconformityRegistryID)
if err != nil {
return nil, fmt.Errorf("cannot delete nonconformity registry: %w", err)
}
return &types.DeleteNonconformityRegistryPayload{
DeletedNonconformityRegistryID: input.NonconformityRegistryID,
}, nil
}
// Organization is the resolver for the organization field.
func (r *nonconformityRegistryResolver) Organization(ctx context.Context, obj *types.NonconformityRegistry) (*types.Organization, error) {
prb := r.ProboService(ctx, obj.ID.TenantID())
registry, err := prb.NonconformityRegistries.Get(ctx, obj.ID)
if err != nil {
return nil, fmt.Errorf("cannot get nonconformity registry: %w", err)
}
organization, err := prb.Organizations.Get(ctx, registry.OrganizationID)
if err != nil {
return nil, fmt.Errorf("cannot get nonconformity registry organization: %w", err)
}
return types.NewOrganization(organization), nil
}
// Audit is the resolver for the audit field.
func (r *nonconformityRegistryResolver) Audit(ctx context.Context, obj *types.NonconformityRegistry) (*types.Audit, error) {
prb := r.ProboService(ctx, obj.ID.TenantID())
registry, err := prb.NonconformityRegistries.Get(ctx, obj.ID)
if err != nil {
return nil, fmt.Errorf("cannot get nonconformity registry: %w", err)
}
audit, err := prb.Audits.Get(ctx, registry.AuditID)
if err != nil {
return nil, fmt.Errorf("cannot get nonconformity registry audit: %w", err)
}
return types.NewAudit(audit), nil
}
// Owner is the resolver for the owner field.
func (r *nonconformityRegistryResolver) Owner(ctx context.Context, obj *types.NonconformityRegistry) (*types.People, error) {
prb := r.ProboService(ctx, obj.ID.TenantID())
registry, err := prb.NonconformityRegistries.Get(ctx, obj.ID)
if err != nil {
return nil, fmt.Errorf("cannot get nonconformity registry: %w", err)
}
people, err := prb.Peoples.Get(ctx, registry.OwnerID)
if err != nil {
return nil, fmt.Errorf("cannot get nonconformity registry owner: %w", err)
}
return types.NewPeople(people), nil
}
// TotalCount is the resolver for the totalCount field.
func (r *nonconformityRegistryConnectionResolver) TotalCount(ctx context.Context, obj *types.NonconformityRegistryConnection) (int, error) {
prb := r.ProboService(ctx, obj.ParentID.TenantID())
switch obj.Resolver.(type) {
case *organizationResolver:
count, err := prb.NonconformityRegistries.CountForOrganizationID(ctx, obj.ParentID)
if err != nil {
return 0, fmt.Errorf("cannot count nonconformity registries: %w", err)
}
return count, nil
}
return 0, fmt.Errorf("unsupported resolver: %T", obj.Resolver)
}
// LogoURL is the resolver for the logoUrl field.
func (r *organizationResolver) LogoURL(ctx context.Context, obj *types.Organization) (*string, error) {
prb := r.ProboService(ctx, obj.ID.TenantID())
@@ -3068,6 +3205,31 @@ func (r *organizationResolver) Audits(ctx context.Context, obj *types.Organizati
return types.NewAuditConnection(page, r, obj.ID), nil
}
// NonconformityRegistries is the resolver for the nonconformityRegistries field.
func (r *organizationResolver) NonconformityRegistries(ctx context.Context, obj *types.Organization, first *int, after *page.CursorKey, last *int, before *page.CursorKey, orderBy *types.NonconformityRegistryOrderBy) (*types.NonconformityRegistryConnection, error) {
prb := r.ProboService(ctx, obj.ID.TenantID())
pageOrderBy := page.OrderBy[coredata.NonconformityRegistryOrderField]{
Field: coredata.NonconformityRegistryOrderFieldCreatedAt,
Direction: page.OrderDirectionDesc,
}
if orderBy != nil {
pageOrderBy = page.OrderBy[coredata.NonconformityRegistryOrderField]{
Field: orderBy.Field,
Direction: orderBy.Direction,
}
}
cursor := types.NewCursor(first, after, last, before, pageOrderBy)
page, err := prb.NonconformityRegistries.ListForOrganizationID(ctx, obj.ID, cursor)
if err != nil {
return nil, fmt.Errorf("cannot list organization nonconformity registries: %w", err)
}
return types.NewNonconformityRegistryConnection(page, r, obj.ID), nil
}
// TrustCenter is the resolver for the trustCenter field.
func (r *organizationResolver) TrustCenter(ctx context.Context, obj *types.Organization) (*types.TrustCenter, error) {
prb := r.ProboService(ctx, obj.ID.TenantID())
@@ -3211,6 +3373,12 @@ func (r *queryResolver) Node(ctx context.Context, id gid.GID) (types.Node, error
panic(fmt.Errorf("cannot get audit: %w", err))
}
return types.NewAudit(audit), nil
case coredata.NonconformityRegistryEntityType:
nonconformityRegistry, err := prb.NonconformityRegistries.Get(ctx, id)
if err != nil {
panic(fmt.Errorf("cannot get nonconformity registry: %w", err))
}
return types.NewNonconformityRegistry(nonconformityRegistry), nil
case coredata.ReportEntityType:
report, err := prb.Reports.Get(ctx, id)
if err != nil {
@@ -3970,6 +4138,16 @@ func (r *Resolver) MeasureConnection() schema.MeasureConnectionResolver {
// Mutation returns schema.MutationResolver implementation.
func (r *Resolver) Mutation() schema.MutationResolver { return &mutationResolver{r} }
// NonconformityRegistry returns schema.NonconformityRegistryResolver implementation.
func (r *Resolver) NonconformityRegistry() schema.NonconformityRegistryResolver {
return &nonconformityRegistryResolver{r}
}
// NonconformityRegistryConnection returns schema.NonconformityRegistryConnectionResolver implementation.
func (r *Resolver) NonconformityRegistryConnection() schema.NonconformityRegistryConnectionResolver {
return &nonconformityRegistryConnectionResolver{r}
}
// Organization returns schema.OrganizationResolver implementation.
func (r *Resolver) Organization() schema.OrganizationResolver { return &organizationResolver{r} }
@@ -4055,6 +4233,8 @@ type frameworkConnectionResolver struct{ *Resolver }
type measureResolver struct{ *Resolver }
type measureConnectionResolver struct{ *Resolver }
type mutationResolver struct{ *Resolver }
type nonconformityRegistryResolver struct{ *Resolver }
type nonconformityRegistryConnectionResolver struct{ *Resolver }
type organizationResolver struct{ *Resolver }
type peopleConnectionResolver struct{ *Resolver }
type queryResolver struct{ *Resolver }