@@ -1204,6 +1204,13 @@ func (q *querier) GetActiveWorkspaceBuildsByTemplateID(ctx context.Context, temp
1204
1204
return q .db .GetActiveWorkspaceBuildsByTemplateID (ctx , templateID )
1205
1205
}
1206
1206
1207
+ func (q * querier ) GetAllFrobulators (ctx context.Context ) ([]database.Frobulator , error ) {
1208
+ if err := q .authorizeContext (ctx , policy .ActionRead , rbac .ResourceFrobulator ); err != nil {
1209
+ return nil , err
1210
+ }
1211
+ return q .db .GetAllFrobulators (ctx )
1212
+ }
1213
+
1207
1214
func (q * querier ) GetAllTailnetAgents (ctx context.Context ) ([]database.TailnetAgent , error ) {
1208
1215
if err := q .authorizeContext (ctx , policy .ActionRead , rbac .ResourceTailnetCoordinator ); err != nil {
1209
1216
return []database.TailnetAgent {}, err
@@ -2052,6 +2059,13 @@ func (q *querier) GetUserCount(ctx context.Context) (int64, error) {
2052
2059
return q .db .GetUserCount (ctx )
2053
2060
}
2054
2061
2062
+ func (q * querier ) GetUserFrobulators (ctx context.Context , userID uuid.UUID ) ([]database.Frobulator , error ) {
2063
+ if err := q .authorizeContext (ctx , policy .ActionRead , rbac .ResourceFrobulator .WithOwner (userID .String ())); err != nil {
2064
+ return nil , err
2065
+ }
2066
+ return q .db .GetUserFrobulators (ctx , userID )
2067
+ }
2068
+
2055
2069
func (q * querier ) GetUserLatencyInsights (ctx context.Context , arg database.GetUserLatencyInsightsParams ) ([]database.GetUserLatencyInsightsRow , error ) {
2056
2070
// Used by insights endpoints. Need to check both for auditors and for regular users with template acl perms.
2057
2071
if err := q .authorizeContext (ctx , policy .ActionViewInsights , rbac .ResourceTemplate ); err != nil {
@@ -2537,6 +2551,14 @@ func (q *querier) InsertFile(ctx context.Context, arg database.InsertFileParams)
2537
2551
return insert (q .log , q .auth , rbac .ResourceFile .WithOwner (arg .CreatedBy .String ()), q .db .InsertFile )(ctx , arg )
2538
2552
}
2539
2553
2554
+ func (q * querier ) InsertFrobulator (ctx context.Context , arg database.InsertFrobulatorParams ) error {
2555
+ if err := q .authorizeContext (ctx , policy .ActionCreate , rbac .ResourceFrobulator .WithOwner (arg .UserID .String ())); err != nil {
2556
+ return err
2557
+ }
2558
+
2559
+ return q .db .InsertFrobulator (ctx , arg )
2560
+ }
2561
+
2540
2562
func (q * querier ) InsertGitSSHKey (ctx context.Context , arg database.InsertGitSSHKeyParams ) (database.GitSSHKey , error ) {
2541
2563
return insertWithAction (q .log , q .auth , rbac .ResourceUser .WithOwner (arg .UserID .String ()).WithID (arg .UserID ), policy .ActionUpdatePersonal , q .db .InsertGitSSHKey )(ctx , arg )
2542
2564
}
0 commit comments