@@ -1225,46 +1225,6 @@ func (q *querier) authorizeProvisionerJob(ctx context.Context, job database.Prov
1225
1225
return nil
1226
1226
}
1227
1227
1228
- func (q * querier ) UpdateTemplateVersionFlagsByJobID (ctx context.Context , arg database.UpdateTemplateVersionFlagsByJobIDParams ) error {
1229
- // An actor is allowed to update the template version ai task and external agent flag if they are authorized to update the template.
1230
- tv , err := q .db .GetTemplateVersionByJobID (ctx , arg .JobID )
1231
- if err != nil {
1232
- return err
1233
- }
1234
- var obj rbac.Objecter
1235
- if ! tv .TemplateID .Valid {
1236
- obj = rbac .ResourceTemplate .InOrg (tv .OrganizationID )
1237
- } else {
1238
- tpl , err := q .db .GetTemplateByID (ctx , tv .TemplateID .UUID )
1239
- if err != nil {
1240
- return err
1241
- }
1242
- obj = tpl
1243
- }
1244
- if err := q .authorizeContext (ctx , policy .ActionUpdate , obj ); err != nil {
1245
- return err
1246
- }
1247
- return q .db .UpdateTemplateVersionFlagsByJobID (ctx , arg )
1248
- }
1249
-
1250
- func (q * querier ) UpdateWorkspaceBuildFlagsByID (ctx context.Context , arg database.UpdateWorkspaceBuildFlagsByIDParams ) error {
1251
- build , err := q .db .GetWorkspaceBuildByID (ctx , arg .ID )
1252
- if err != nil {
1253
- return err
1254
- }
1255
-
1256
- workspace , err := q .db .GetWorkspaceByID (ctx , build .WorkspaceID )
1257
- if err != nil {
1258
- return err
1259
- }
1260
-
1261
- err = q .authorizeContext (ctx , policy .ActionUpdate , workspace .RBACObject ())
1262
- if err != nil {
1263
- return err
1264
- }
1265
- return q .db .UpdateWorkspaceBuildFlagsByID (ctx , arg )
1266
- }
1267
-
1268
1228
func (q * querier ) AcquireLock (ctx context.Context , id int64 ) error {
1269
1229
return q .db .AcquireLock (ctx , id )
1270
1230
}
@@ -4797,6 +4757,28 @@ func (q *querier) UpdateTemplateVersionExternalAuthProvidersByJobID(ctx context.
4797
4757
return q .db .UpdateTemplateVersionExternalAuthProvidersByJobID (ctx , arg )
4798
4758
}
4799
4759
4760
+ func (q * querier ) UpdateTemplateVersionFlagsByJobID (ctx context.Context , arg database.UpdateTemplateVersionFlagsByJobIDParams ) error {
4761
+ // An actor is allowed to update the template version ai task and external agent flag if they are authorized to update the template.
4762
+ tv , err := q .db .GetTemplateVersionByJobID (ctx , arg .JobID )
4763
+ if err != nil {
4764
+ return err
4765
+ }
4766
+ var obj rbac.Objecter
4767
+ if ! tv .TemplateID .Valid {
4768
+ obj = rbac .ResourceTemplate .InOrg (tv .OrganizationID )
4769
+ } else {
4770
+ tpl , err := q .db .GetTemplateByID (ctx , tv .TemplateID .UUID )
4771
+ if err != nil {
4772
+ return err
4773
+ }
4774
+ obj = tpl
4775
+ }
4776
+ if err := q .authorizeContext (ctx , policy .ActionUpdate , obj ); err != nil {
4777
+ return err
4778
+ }
4779
+ return q .db .UpdateTemplateVersionFlagsByJobID (ctx , arg )
4780
+ }
4781
+
4800
4782
func (q * querier ) UpdateTemplateWorkspacesLastUsedAt (ctx context.Context , arg database.UpdateTemplateWorkspacesLastUsedAtParams ) error {
4801
4783
fetch := func (ctx context.Context , arg database.UpdateTemplateWorkspacesLastUsedAtParams ) (database.Template , error ) {
4802
4784
return q .db .GetTemplateByID (ctx , arg .TemplateID )
@@ -5138,6 +5120,24 @@ func (q *querier) UpdateWorkspaceBuildDeadlineByID(ctx context.Context, arg data
5138
5120
return q .db .UpdateWorkspaceBuildDeadlineByID (ctx , arg )
5139
5121
}
5140
5122
5123
+ func (q * querier ) UpdateWorkspaceBuildFlagsByID (ctx context.Context , arg database.UpdateWorkspaceBuildFlagsByIDParams ) error {
5124
+ build , err := q .db .GetWorkspaceBuildByID (ctx , arg .ID )
5125
+ if err != nil {
5126
+ return err
5127
+ }
5128
+
5129
+ workspace , err := q .db .GetWorkspaceByID (ctx , build .WorkspaceID )
5130
+ if err != nil {
5131
+ return err
5132
+ }
5133
+
5134
+ err = q .authorizeContext (ctx , policy .ActionUpdate , workspace .RBACObject ())
5135
+ if err != nil {
5136
+ return err
5137
+ }
5138
+ return q .db .UpdateWorkspaceBuildFlagsByID (ctx , arg )
5139
+ }
5140
+
5141
5141
func (q * querier ) UpdateWorkspaceBuildProvisionerStateByID (ctx context.Context , arg database.UpdateWorkspaceBuildProvisionerStateByIDParams ) error {
5142
5142
if err := q .authorizeContext (ctx , policy .ActionUpdate , rbac .ResourceSystem ); err != nil {
5143
5143
return err
0 commit comments