@@ -1074,6 +1074,10 @@ func (q *querier) DeleteOrganizationMember(ctx context.Context, arg database.Del
1074
1074
}, q .db .DeleteOrganizationMember )(ctx , arg )
1075
1075
}
1076
1076
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
+
1077
1081
func (q * querier ) DeleteReplicasUpdatedBefore (ctx context.Context , updatedAt time.Time ) error {
1078
1082
if err := q .authorizeContext (ctx , policy .ActionDelete , rbac .ResourceSystem ); err != nil {
1079
1083
return err
@@ -1671,6 +1675,14 @@ func (q *querier) GetProvisionerJobsCreatedAfter(ctx context.Context, createdAt
1671
1675
return q .db .GetProvisionerJobsCreatedAfter (ctx , createdAt )
1672
1676
}
1673
1677
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
+
1674
1686
func (q * querier ) GetProvisionerLogsAfterID (ctx context.Context , arg database.GetProvisionerLogsAfterIDParams ) ([]database.ProvisionerJobLog , error ) {
1675
1687
// Authorized read on job lets the actor also read the logs.
1676
1688
_ , err := q .GetProvisionerJobByID (ctx , arg .JobID )
@@ -2615,6 +2627,10 @@ func (q *querier) InsertProvisionerJobLogs(ctx context.Context, arg database.Ins
2615
2627
return q .db .InsertProvisionerJobLogs (ctx , arg )
2616
2628
}
2617
2629
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
+
2618
2634
func (q * querier ) InsertReplica (ctx context.Context , arg database.InsertReplicaParams ) (database.Replica , error ) {
2619
2635
if err := q .authorizeContext (ctx , policy .ActionCreate , rbac .ResourceSystem ); err != nil {
2620
2636
return database.Replica {}, err
@@ -2843,6 +2859,10 @@ func (q *querier) InsertWorkspaceResourceMetadata(ctx context.Context, arg datab
2843
2859
return q .db .InsertWorkspaceResourceMetadata (ctx , arg )
2844
2860
}
2845
2861
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
+
2846
2866
func (q * querier ) ListWorkspaceAgentPortShares (ctx context.Context , workspaceID uuid.UUID ) ([]database.WorkspaceAgentPortShare , error ) {
2847
2867
workspace , err := q .db .GetWorkspaceByID (ctx , workspaceID )
2848
2868
if err != nil {
0 commit comments