Skip to content

Commit 7dfa336

Browse files
gearnodegearnode
committed
Style
Signed-off-by: gearnode <bryan@frimin.fr>
1 parent 3fcf5e4 commit 7dfa336

17 files changed

+765
-390
lines changed

‎pkg/probo/get_control.go‎

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/probo/coredata"
22+
"go.gearno.de/kit/pg"
23+
)
24+
25+
func (s Service) GetControl(
26+
ctx context.Context,
27+
controlID gid.GID,
28+
) (*coredata.Control, error) {
29+
control := &coredata.Control{}
30+
31+
err := s.pg.WithConn(
32+
ctx,
33+
func(conn pg.Conn) error {
34+
return control.LoadByID(ctx, conn, s.scope, controlID)
35+
},
36+
)
37+
38+
if err != nil {
39+
return nil, err
40+
}
41+
42+
return control, nil
43+
}

‎pkg/probo/get_evidence.go‎

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/probo/coredata"
22+
"go.gearno.de/kit/pg"
23+
)
24+
25+
func (s Service) GetEvidence(
26+
ctx context.Context,
27+
evidenceID gid.GID,
28+
) (*coredata.Evidence, error) {
29+
evidence := &coredata.Evidence{}
30+
31+
err := s.pg.WithConn(
32+
ctx,
33+
func(conn pg.Conn) error {
34+
return evidence.LoadByID(ctx, conn, s.scope, evidenceID)
35+
},
36+
)
37+
38+
if err != nil {
39+
return nil, err
40+
}
41+
42+
return evidence, nil
43+
}

‎pkg/probo/get_framework.go‎

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/probo/coredata"
22+
"go.gearno.de/kit/pg"
23+
)
24+
25+
func (s Service) GetFramework(
26+
ctx context.Context,
27+
frameworkID gid.GID,
28+
) (*coredata.Framework, error) {
29+
framework := &coredata.Framework{}
30+
31+
err := s.pg.WithConn(
32+
ctx,
33+
func(conn pg.Conn) error {
34+
return framework.LoadByID(ctx, conn, s.scope, frameworkID)
35+
},
36+
)
37+
38+
if err != nil {
39+
return nil, err
40+
}
41+
42+
return framework, nil
43+
}

‎pkg/probo/get_organization.go‎

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/probo/coredata"
22+
"go.gearno.de/kit/pg"
23+
)
24+
25+
func (s Service) GetOrganization(
26+
ctx context.Context,
27+
organizationID gid.GID,
28+
) (*coredata.Organization, error) {
29+
organization := &coredata.Organization{}
30+
31+
err := s.pg.WithConn(
32+
ctx,
33+
func(conn pg.Conn) error {
34+
return organization.LoadByID(
35+
ctx,
36+
conn,
37+
s.scope,
38+
organizationID,
39+
)
40+
},
41+
)
42+
43+
if err != nil {
44+
return nil, err
45+
}
46+
47+
return organization, nil
48+
}

‎pkg/probo/get_people.go‎

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/probo/coredata"
22+
"go.gearno.de/kit/pg"
23+
)
24+
25+
func (s Service) GetPeople(
26+
ctx context.Context,
27+
peopleID gid.GID,
28+
) (*coredata.People, error) {
29+
people := &coredata.People{}
30+
31+
err := s.pg.WithConn(
32+
ctx,
33+
func(conn pg.Conn) error {
34+
return people.LoadByID(ctx, conn, s.scope, peopleID)
35+
},
36+
)
37+
38+
if err != nil {
39+
return nil, err
40+
}
41+
42+
return people, nil
43+
}

‎pkg/probo/get_task.go‎

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/probo/coredata"
22+
"go.gearno.de/kit/pg"
23+
)
24+
25+
func (s Service) GetTask(
26+
ctx context.Context,
27+
taskID gid.GID,
28+
) (*coredata.Task, error) {
29+
task := &coredata.Task{}
30+
31+
err := s.pg.WithConn(
32+
ctx,
33+
func(conn pg.Conn) error {
34+
return task.LoadByID(ctx, conn, s.scope, taskID)
35+
},
36+
)
37+
38+
if err != nil {
39+
return nil, err
40+
}
41+
42+
return task, nil
43+
}

‎pkg/probo/get_vendor.go‎

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/probo/coredata"
22+
"go.gearno.de/kit/pg"
23+
)
24+
25+
func (s Service) GetVendor(
26+
ctx context.Context,
27+
vendorID gid.GID,
28+
) (*coredata.Vendor, error) {
29+
vendor := &coredata.Vendor{}
30+
31+
err := s.pg.WithConn(
32+
ctx,
33+
func(conn pg.Conn) error {
34+
return vendor.LoadByID(ctx, conn, s.scope, vendorID)
35+
},
36+
)
37+
38+
if err != nil {
39+
return nil, err
40+
}
41+
42+
return vendor, nil
43+
}

‎pkg/probo/list_control_state_transitions.go‎

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
2+
//
3+
// Permission to use, copy, modify, and/or distribute this software for any
4+
// purpose with or without fee is hereby granted, provided that the above
5+
// copyright notice and this permission notice appear in all copies.
6+
//
7+
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
8+
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
9+
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
10+
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
11+
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
12+
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
13+
// PERFORMANCE OF THIS SOFTWARE.
14+
15+
package probo
16+
17+
import (
18+
"context"
19+
20+
"github.com/getprobo/probo/pkg/gid"
21+
"github.com/getprobo/probo/pkg/page"
22+
"github.com/getprobo/probo/pkg/probo/coredata"
23+
"go.gearno.de/kit/pg"
24+
)
25+
26+
func (s Service) ListControlStateTransitions(
27+
ctx context.Context,
28+
controlID gid.GID,
29+
cursor *page.Cursor,
30+
) (*page.Page[*coredata.ControlStateTransition], error) {
31+
var controlStateTransitions coredata.ControlStateTransitions
32+
33+
err := s.pg.WithConn(
34+
ctx,
35+
func(conn pg.Conn) error {
36+
return controlStateTransitions.LoadByControlID(
37+
ctx,
38+
conn,
39+
s.scope,
40+
controlID,
41+
cursor,
42+
)
43+
},
44+
)
45+
46+
if err != nil {
47+
return nil, err
48+
}
49+
50+
return page.NewPage(controlStateTransitions, cursor), nil
51+
}

0 commit comments

Comments
 (0)