Add evidence state history

Signed-off-by: gearnode <bryan@frimin.fr>
This commit is contained in:
gearnode
2025-01-30 21:37:48 -08:00
parent 9c0c5afa9c
commit 14380aa635
6 changed files with 1094 additions and 15 deletions

View File

@@ -249,6 +249,14 @@ type Evidence implements Node {
id: ID!
fileUrl: String!
state: EvidenceState!
stateTransisions(
first: Int
after: CursorKey
last: Int
before: CursorKey
): EvidenceStateTransitionConnection! @goField(forceResolver: true)
createdAt: Datetime!
updatedAt: Datetime!
}
@@ -256,3 +264,22 @@ type Evidence implements Node {
type Query {
node(id: ID!): Node!
}
type EvidenceStateTransitionConnection {
edges: [EvidenceStateTransitionEdge!]!
pageInfo: PageInfo!
}
type EvidenceStateTransitionEdge {
cursor: CursorKey!
node: EvidenceStateTransition!
}
type EvidenceStateTransition {
id: ID!
fromState: EvidenceState
toState: EvidenceState!
reason: String
createdAt: Datetime!
updatedAt: Datetime!
}

File diff suppressed because it is too large Load Diff

View File

@@ -0,0 +1,59 @@
// 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/probo/coredata"
"github.com/getprobo/probo/pkg/probo/coredata/page"
)
func NewEvidenceStateTransitionConnection(
p *page.Page[*coredata.EvidenceStateTransition],
) *EvidenceStateTransitionConnection {
var edges = make([]*EvidenceStateTransitionEdge, len(p.Data))
for i := range edges {
edges[i] = NewEvidenceStateTransitionEdge(p.Data[i])
}
return &EvidenceStateTransitionConnection{
Edges: edges,
PageInfo: NewPageInfo(p),
}
}
func NewEvidenceStateTransitionEdge(est *coredata.EvidenceStateTransition) *EvidenceStateTransitionEdge {
return &EvidenceStateTransitionEdge{
Cursor: est.CursorKey(),
Node: NewEvidenceStateTransition(est),
}
}
func NewEvidenceStateTransition(est *coredata.EvidenceStateTransition) *EvidenceStateTransition {
var fromState *EvidenceState
if est.FromState != nil {
val := EvidenceState((*est.FromState).String())
fromState = &val
}
return &EvidenceStateTransition{
ID: est.ID,
FromState: fromState,
ToState: EvidenceState(est.ToState.String()),
Reason: est.Reason,
CreatedAt: est.CreatedAt,
UpdatedAt: est.UpdatedAt,
}
}

View File

@@ -61,11 +61,12 @@ type ControlStateTransitionEdge struct {
}
type Evidence struct {
ID gid.GID `json:"id"`
FileURL string `json:"fileUrl"`
State EvidenceState `json:"state"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
ID gid.GID `json:"id"`
FileURL string `json:"fileUrl"`
State EvidenceState `json:"state"`
StateTransisions *EvidenceStateTransitionConnection `json:"stateTransisions"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
}
func (Evidence) IsNode() {}
@@ -81,6 +82,25 @@ type EvidenceEdge struct {
Node *Evidence `json:"node"`
}
type EvidenceStateTransition struct {
ID gid.GID `json:"id"`
FromState *EvidenceState `json:"fromState,omitempty"`
ToState EvidenceState `json:"toState"`
Reason *string `json:"reason,omitempty"`
CreatedAt time.Time `json:"createdAt"`
UpdatedAt time.Time `json:"updatedAt"`
}
type EvidenceStateTransitionConnection struct {
Edges []*EvidenceStateTransitionEdge `json:"edges"`
PageInfo *PageInfo `json:"pageInfo"`
}
type EvidenceStateTransitionEdge struct {
Cursor page.CursorKey `json:"cursor"`
Node *EvidenceStateTransition `json:"node"`
}
type Framework struct {
ID gid.GID `json:"id"`
Name string `json:"name"`

View File

@@ -40,6 +40,18 @@ func (r *controlResolver) Tasks(ctx context.Context, obj *types.Control, first *
return types.NewTaskConnection(page), nil
}
// StateTransisions is the resolver for the stateTransisions field.
func (r *evidenceResolver) StateTransisions(ctx context.Context, obj *types.Evidence, first *int, after *page.CursorKey, last *int, before *page.CursorKey) (*types.EvidenceStateTransitionConnection, error) {
cursor := types.NewCursor(first, after, last, before)
page, err := r.svc.ListEvidenceStateTransitions(ctx, obj.ID, cursor)
if err != nil {
return nil, fmt.Errorf("cannot list evidence state transitions: %w", err)
}
return types.NewEvidenceStateTransitionConnection(page), nil
}
// Controls is the resolver for the controls field.
func (r *frameworkResolver) Controls(ctx context.Context, obj *types.Framework, first *int, after *page.CursorKey, last *int, before *page.CursorKey) (*types.ControlConnection, error) {
cursor := types.NewCursor(first, after, last, before)
@@ -131,6 +143,9 @@ func (r *taskResolver) Evidences(ctx context.Context, obj *types.Task, first *in
// Control returns schema.ControlResolver implementation.
func (r *Resolver) Control() schema.ControlResolver { return &controlResolver{r} }
// Evidence returns schema.EvidenceResolver implementation.
func (r *Resolver) Evidence() schema.EvidenceResolver { return &evidenceResolver{r} }
// Framework returns schema.FrameworkResolver implementation.
func (r *Resolver) Framework() schema.FrameworkResolver { return &frameworkResolver{r} }
@@ -144,6 +159,7 @@ func (r *Resolver) Query() schema.QueryResolver { return &queryResolver{r} }
func (r *Resolver) Task() schema.TaskResolver { return &taskResolver{r} }
type controlResolver struct{ *Resolver }
type evidenceResolver struct{ *Resolver }
type frameworkResolver struct{ *Resolver }
type organizationResolver struct{ *Resolver }
type queryResolver struct{ *Resolver }

View File

@@ -284,3 +284,30 @@ func (s Service) ListTaskStateTransitions(
return page.NewPage(taskStateTransitions, cursor), nil
}
func (s Service) ListEvidenceStateTransitions(
ctx context.Context,
evidenceID gid.GID,
cursor *page.Cursor,
) (*page.Page[*coredata.EvidenceStateTransition], error) {
var evidenceStateTransitions coredata.EvidenceStateTransitions
err := s.pg.WithConn(
ctx,
func(conn pg.Conn) error {
return evidenceStateTransitions.LoadByEvidenceID(
ctx,
conn,
s.scope,
evidenceID,
cursor,
)
},
)
if err != nil {
return nil, err
}
return page.NewPage(evidenceStateTransitions, cursor), nil
}