@@ -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
59
pkg/api/console/v1/types/evidence_state_transition.go
Normal file
59
pkg/api/console/v1/types/evidence_state_transition.go
Normal 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,
|
||||
}
|
||||
}
|
||||
@@ -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"`
|
||||
|
||||
@@ -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 }
|
||||
|
||||
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user