@@ -249,6 +249,14 @@ type Evidence implements Node {
|
|||||||
id: ID!
|
id: ID!
|
||||||
fileUrl: String!
|
fileUrl: String!
|
||||||
state: EvidenceState!
|
state: EvidenceState!
|
||||||
|
|
||||||
|
stateTransisions(
|
||||||
|
first: Int
|
||||||
|
after: CursorKey
|
||||||
|
last: Int
|
||||||
|
before: CursorKey
|
||||||
|
): EvidenceStateTransitionConnection! @goField(forceResolver: true)
|
||||||
|
|
||||||
createdAt: Datetime!
|
createdAt: Datetime!
|
||||||
updatedAt: Datetime!
|
updatedAt: Datetime!
|
||||||
}
|
}
|
||||||
@@ -256,3 +264,22 @@ type Evidence implements Node {
|
|||||||
type Query {
|
type Query {
|
||||||
node(id: ID!): Node!
|
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 {
|
type Evidence struct {
|
||||||
ID gid.GID `json:"id"`
|
ID gid.GID `json:"id"`
|
||||||
FileURL string `json:"fileUrl"`
|
FileURL string `json:"fileUrl"`
|
||||||
State EvidenceState `json:"state"`
|
State EvidenceState `json:"state"`
|
||||||
CreatedAt time.Time `json:"createdAt"`
|
StateTransisions *EvidenceStateTransitionConnection `json:"stateTransisions"`
|
||||||
UpdatedAt time.Time `json:"updatedAt"`
|
CreatedAt time.Time `json:"createdAt"`
|
||||||
|
UpdatedAt time.Time `json:"updatedAt"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Evidence) IsNode() {}
|
func (Evidence) IsNode() {}
|
||||||
@@ -81,6 +82,25 @@ type EvidenceEdge struct {
|
|||||||
Node *Evidence `json:"node"`
|
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 {
|
type Framework struct {
|
||||||
ID gid.GID `json:"id"`
|
ID gid.GID `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
|
|||||||
@@ -40,6 +40,18 @@ func (r *controlResolver) Tasks(ctx context.Context, obj *types.Control, first *
|
|||||||
return types.NewTaskConnection(page), nil
|
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.
|
// 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) {
|
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)
|
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.
|
// Control returns schema.ControlResolver implementation.
|
||||||
func (r *Resolver) Control() schema.ControlResolver { return &controlResolver{r} }
|
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.
|
// Framework returns schema.FrameworkResolver implementation.
|
||||||
func (r *Resolver) Framework() schema.FrameworkResolver { return &frameworkResolver{r} }
|
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} }
|
func (r *Resolver) Task() schema.TaskResolver { return &taskResolver{r} }
|
||||||
|
|
||||||
type controlResolver struct{ *Resolver }
|
type controlResolver struct{ *Resolver }
|
||||||
|
type evidenceResolver struct{ *Resolver }
|
||||||
type frameworkResolver struct{ *Resolver }
|
type frameworkResolver struct{ *Resolver }
|
||||||
type organizationResolver struct{ *Resolver }
|
type organizationResolver struct{ *Resolver }
|
||||||
type queryResolver struct{ *Resolver }
|
type queryResolver struct{ *Resolver }
|
||||||
|
|||||||
@@ -284,3 +284,30 @@ func (s Service) ListTaskStateTransitions(
|
|||||||
|
|
||||||
return page.NewPage(taskStateTransitions, cursor), nil
|
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