@@ -1074,6 +1074,10 @@ func (q *querier) DeleteOrganizationMember(ctx context.Context, arg database.Del
10741074 }, q .db .DeleteOrganizationMember )(ctx , arg )
10751075}
10761076
1077+ func (q * querier ) DeleteProvisionerKey (ctx context.Context , id uuid.UUID ) error {
1078+ return deleteQ (q .log , q .auth , q .db .GetProvisionerKeyByID , q .db .DeleteProvisionerKey )(ctx , id )
1079+ }
1080+
10771081func (q * querier ) DeleteReplicasUpdatedBefore (ctx context.Context , updatedAt time.Time ) error {
10781082 if err := q .authorizeContext (ctx , policy .ActionDelete , rbac .ResourceSystem ); err != nil {
10791083 return err
@@ -1671,6 +1675,14 @@ func (q *querier) GetProvisionerJobsCreatedAfter(ctx context.Context, createdAt
16711675 return q .db .GetProvisionerJobsCreatedAfter (ctx , createdAt )
16721676}
16731677
1678+ func (q * querier ) GetProvisionerKeyByID (ctx context.Context , id uuid.UUID ) (database.ProvisionerKey , error ) {
1679+ return fetch (q .log , q .auth , q .db .GetProvisionerKeyByID )(ctx , id )
1680+ }
1681+
1682+ func (q * querier ) GetProvisionerKeyByName (ctx context.Context , name database.GetProvisionerKeyByNameParams ) (database.ProvisionerKey , error ) {
1683+ return fetch (q .log , q .auth , q .db .GetProvisionerKeyByName )(ctx , name )
1684+ }
1685+
16741686func (q * querier ) GetProvisionerLogsAfterID (ctx context.Context , arg database.GetProvisionerLogsAfterIDParams ) ([]database.ProvisionerJobLog , error ) {
16751687 // Authorized read on job lets the actor also read the logs.
16761688 _ , err := q .GetProvisionerJobByID (ctx , arg .JobID )
@@ -2615,6 +2627,10 @@ func (q *querier) InsertProvisionerJobLogs(ctx context.Context, arg database.Ins
26152627 return q .db .InsertProvisionerJobLogs (ctx , arg )
26162628}
26172629
2630+ func (q * querier ) InsertProvisionerKey (ctx context.Context , arg database.InsertProvisionerKeyParams ) (database.ProvisionerKey , error ) {
2631+ return insert (q .log , q .auth , rbac .ResourceProvisionerKeys .InOrg (arg .OrganizationID ).WithID (arg .ID ), q .db .InsertProvisionerKey )(ctx , arg )
2632+ }
2633+
26182634func (q * querier ) InsertReplica (ctx context.Context , arg database.InsertReplicaParams ) (database.Replica , error ) {
26192635 if err := q .authorizeContext (ctx , policy .ActionCreate , rbac .ResourceSystem ); err != nil {
26202636 return database.Replica {}, err
@@ -2843,6 +2859,10 @@ func (q *querier) InsertWorkspaceResourceMetadata(ctx context.Context, arg datab
28432859 return q .db .InsertWorkspaceResourceMetadata (ctx , arg )
28442860}
28452861
2862+ func (q * querier ) ListProvisionerKeysByOrganization (ctx context.Context , organizationID uuid.UUID ) ([]database.ProvisionerKey , error ) {
2863+ return fetchWithPostFilter (q .auth , policy .ActionRead , q .db .ListProvisionerKeysByOrganization )(ctx , organizationID )
2864+ }
2865+
28462866func (q * querier ) ListWorkspaceAgentPortShares (ctx context.Context , workspaceID uuid.UUID ) ([]database.WorkspaceAgentPortShare , error ) {
28472867 workspace , err := q .db .GetWorkspaceByID (ctx , workspaceID )
28482868 if err != nil {
0 commit comments