From e9025f91e86864438661cd253e9c14f2d0a5fce7 Mon Sep 17 00:00:00 2001 From: Cian Johnston Date: Thu, 12 Mar 2026 21:32:57 +0000 Subject: [PATCH] chore(db): remove 23 unused database methods (#22999) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Removes 22 database query methods with no callers outside generated code and the dbauthz wrapper layer (~1,600 lines). **Security keys (6)** — superseded by `cryptokeys` package: `GetAppSecurityKey`, `UpsertAppSecurityKey`, `GetOAuthSigningKey`, `UpsertOAuthSigningKey`, `GetCoordinatorResumeTokenSigningKey`, `UpsertCoordinatorResumeTokenSigningKey` **Superseded queries (4):** - `GetProvisionerJobsByIDs` → `GetProvisionerJobsByIDsWithQueuePosition` - `GetDeploymentDAUs` / `GetTemplateDAUs` → `GetTemplateInsightsByInterval` - `GetWorkspaceBuildParametersByBuildIDs` + its `GetAuthorized...` variant → unused **OAuth2 (2):** `GetOAuth2ProviderAppByRegistrationToken`, `UpdateOAuth2ProviderAppSecretByID` **Chat (4)** — pre-wired with no callers: `GetChatModelConfigByProviderAndModel`, `DeleteChatMessagesByChatID`, `ListChatsByRootID`, `ListChildChatsByParentID` **Other (6):** `DeleteGitSSHKey`, `UpdateUserLinkedID`, `GetFileIDByTemplateVersionID`, `GetTemplateVersionHasAITask`, `InsertUserGroupsByName`, `RemoveUserFromAllGroups` --- coderd/database/dbauthz/dbauthz.go | 185 ------ coderd/database/dbauthz/dbauthz_test.go | 158 ----- coderd/database/dbmetrics/querymetrics.go | 184 ------ coderd/database/dbmock/dbmock.go | 338 ---------- coderd/database/modelqueries.go | 30 - coderd/database/querier.go | 23 - coderd/database/queries.sql.go | 591 ------------------ coderd/database/queries/chatmodelconfigs.sql | 15 - coderd/database/queries/chats.sql | 26 - coderd/database/queries/files.sql | 16 - coderd/database/queries/gitsshkeys.sql | 5 - coderd/database/queries/groupmembers.sql | 25 - coderd/database/queries/oauth2.sql | 7 - coderd/database/queries/provisionerjobs.sql | 8 - coderd/database/queries/siteconfig.sql | 21 - coderd/database/queries/templateversions.sql | 7 - coderd/database/queries/user_links.sql | 8 - .../database/queries/workspaceagentstats.sql | 27 - .../queries/workspacebuildparameters.sql | 14 - 19 files changed, 1688 deletions(-) diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go index 50d4c9c36f..3e57a0cf90 100644 --- a/coderd/database/dbauthz/dbauthz.go +++ b/coderd/database/dbauthz/dbauthz.go @@ -1829,18 +1829,6 @@ func (q *querier) DeleteChatMessagesAfterID(ctx context.Context, arg database.De return q.db.DeleteChatMessagesAfterID(ctx, arg) } -func (q *querier) DeleteChatMessagesByChatID(ctx context.Context, chatID uuid.UUID) error { - // Authorize delete on the parent chat. - chat, err := q.db.GetChatByID(ctx, chatID) - if err != nil { - return err - } - if err := q.authorizeContext(ctx, policy.ActionDelete, chat); err != nil { - return err - } - return q.db.DeleteChatMessagesByChatID(ctx, chatID) -} - func (q *querier) DeleteChatModelConfigByID(ctx context.Context, id uuid.UUID) error { if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceDeploymentConfig); err != nil { return err @@ -1900,10 +1888,6 @@ func (q *querier) DeleteExternalAuthLink(ctx context.Context, arg database.Delet }, q.db.DeleteExternalAuthLink)(ctx, arg) } -func (q *querier) DeleteGitSSHKey(ctx context.Context, userID uuid.UUID) error { - return fetchAndExec(q.log, q.auth, policy.ActionUpdatePersonal, q.db.GetGitSSHKey, q.db.DeleteGitSSHKey)(ctx, userID) -} - func (q *querier) DeleteGroupByID(ctx context.Context, id uuid.UUID) error { return deleteQ(q.log, q.auth, q.db.GetGroupByID, q.db.DeleteGroupByID)(ctx, id) } @@ -2398,13 +2382,6 @@ func (q *querier) GetAnnouncementBanners(ctx context.Context) (string, error) { return q.db.GetAnnouncementBanners(ctx) } -func (q *querier) GetAppSecurityKey(ctx context.Context) (string, error) { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { - return "", err - } - return q.db.GetAppSecurityKey(ctx) -} - func (q *querier) GetApplicationName(ctx context.Context) (string, error) { // No authz checks return q.db.GetApplicationName(ctx) @@ -2543,13 +2520,6 @@ func (q *querier) GetChatModelConfigByID(ctx context.Context, id uuid.UUID) (dat return q.db.GetChatModelConfigByID(ctx, id) } -func (q *querier) GetChatModelConfigByProviderAndModel(ctx context.Context, arg database.GetChatModelConfigByProviderAndModelParams) (database.ChatModelConfig, error) { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceDeploymentConfig); err != nil { - return database.ChatModelConfig{}, err - } - return q.db.GetChatModelConfigByProviderAndModel(ctx, arg) -} - func (q *querier) GetChatModelConfigs(ctx context.Context) ([]database.ChatModelConfig, error) { if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceDeploymentConfig); err != nil { return nil, err @@ -2617,13 +2587,6 @@ func (q *querier) GetConnectionLogsOffset(ctx context.Context, arg database.GetC return q.db.GetAuthorizedConnectionLogsOffset(ctx, arg, prep) } -func (q *querier) GetCoordinatorResumeTokenSigningKey(ctx context.Context) (string, error) { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { - return "", err - } - return q.db.GetCoordinatorResumeTokenSigningKey(ctx) -} - func (q *querier) GetCryptoKeyByFeatureAndSequence(ctx context.Context, arg database.GetCryptoKeyByFeatureAndSequenceParams) (database.CryptoKey, error) { if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceCryptoKey); err != nil { return database.CryptoKey{}, err @@ -2677,14 +2640,6 @@ func (q *querier) GetDefaultProxyConfig(ctx context.Context) (database.GetDefaul return q.db.GetDefaultProxyConfig(ctx) } -// Only used by metrics cache. -func (q *querier) GetDeploymentDAUs(ctx context.Context, tzOffset int32) ([]database.GetDeploymentDAUsRow, error) { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { - return nil, err - } - return q.db.GetDeploymentDAUs(ctx, tzOffset) -} - func (q *querier) GetDeploymentID(ctx context.Context) (string, error) { // No authz checks return q.db.GetDeploymentID(ctx) @@ -2767,22 +2722,6 @@ func (q *querier) GetFileByID(ctx context.Context, id uuid.UUID) (database.File, return file, nil } -func (q *querier) GetFileIDByTemplateVersionID(ctx context.Context, templateVersionID uuid.UUID) (uuid.UUID, error) { - fileID, err := q.db.GetFileIDByTemplateVersionID(ctx, templateVersionID) - if err != nil { - return uuid.Nil, err - } - // This is a kind of weird check, because users will almost never have this - // permission. Since this query is not currently used to provide data in a - // user facing way, it's expected that this query is run as some system - // subject in order to be authorized. - err = q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceFile.WithID(fileID)) - if err != nil { - return uuid.Nil, err - } - return fileID, nil -} - func (q *querier) GetFileTemplates(ctx context.Context, fileID uuid.UUID) ([]database.GetFileTemplatesRow, error) { if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { return nil, err @@ -2991,13 +2930,6 @@ func (q *querier) GetOAuth2ProviderAppByID(ctx context.Context, id uuid.UUID) (d return q.db.GetOAuth2ProviderAppByID(ctx, id) } -func (q *querier) GetOAuth2ProviderAppByRegistrationToken(ctx context.Context, registrationAccessToken []byte) (database.OAuth2ProviderApp, error) { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceOauth2App); err != nil { - return database.OAuth2ProviderApp{}, err - } - return q.db.GetOAuth2ProviderAppByRegistrationToken(ctx, registrationAccessToken) -} - func (q *querier) GetOAuth2ProviderAppCodeByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderAppCode, error) { return fetch(q.log, q.auth, q.db.GetOAuth2ProviderAppCodeByID)(ctx, id) } @@ -3066,13 +2998,6 @@ func (q *querier) GetOAuth2ProviderAppsByUserID(ctx context.Context, userID uuid return q.db.GetOAuth2ProviderAppsByUserID(ctx, userID) } -func (q *querier) GetOAuthSigningKey(ctx context.Context) (string, error) { - if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { - return "", err - } - return q.db.GetOAuthSigningKey(ctx) -} - func (q *querier) GetOrganizationByID(ctx context.Context, id uuid.UUID) (database.Organization, error) { return fetch(q.log, q.auth, q.db.GetOrganizationByID)(ctx, id) } @@ -3312,23 +3237,6 @@ func (q *querier) GetProvisionerJobTimingsByJobID(ctx context.Context, jobID uui return q.db.GetProvisionerJobTimingsByJobID(ctx, jobID) } -func (q *querier) GetProvisionerJobsByIDs(ctx context.Context, ids []uuid.UUID) ([]database.ProvisionerJob, error) { - provisionerJobs, err := q.db.GetProvisionerJobsByIDs(ctx, ids) - if err != nil { - return nil, err - } - orgIDs := make(map[uuid.UUID]struct{}) - for _, job := range provisionerJobs { - orgIDs[job.OrganizationID] = struct{}{} - } - for orgID := range orgIDs { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceProvisionerJobs.InOrg(orgID)); err != nil { - return nil, err - } - } - return provisionerJobs, nil -} - func (q *querier) GetProvisionerJobsByIDsWithQueuePosition(ctx context.Context, ids database.GetProvisionerJobsByIDsWithQueuePositionParams) ([]database.GetProvisionerJobsByIDsWithQueuePositionRow, error) { // TODO: Remove this once we have a proper rbac check for provisioner jobs. // Details in https://github.com/coder/coder/issues/16160 @@ -3535,14 +3443,6 @@ func (q *querier) GetTemplateByOrganizationAndName(ctx context.Context, arg data return fetch(q.log, q.auth, q.db.GetTemplateByOrganizationAndName)(ctx, arg) } -// Only used by metrics cache. -func (q *querier) GetTemplateDAUs(ctx context.Context, arg database.GetTemplateDAUsParams) ([]database.GetTemplateDAUsRow, error) { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { - return nil, err - } - return q.db.GetTemplateDAUs(ctx, arg) -} - func (q *querier) GetTemplateInsights(ctx context.Context, arg database.GetTemplateInsightsParams) (database.GetTemplateInsightsRow, error) { if err := q.authorizeTemplateInsights(ctx, arg.TemplateIDs); err != nil { return database.GetTemplateInsightsRow{}, err @@ -3639,17 +3539,6 @@ func (q *querier) GetTemplateVersionByTemplateIDAndName(ctx context.Context, arg return tv, nil } -func (q *querier) GetTemplateVersionHasAITask(ctx context.Context, id uuid.UUID) (bool, error) { - // If we can successfully call `GetTemplateVersionByID`, then - // we know the actor has sufficient permissions to know if the - // template has an AI task. - if _, err := q.GetTemplateVersionByID(ctx, id); err != nil { - return false, err - } - - return q.db.GetTemplateVersionHasAITask(ctx, id) -} - func (q *querier) GetTemplateVersionParameters(ctx context.Context, templateVersionID uuid.UUID) ([]database.TemplateVersionParameter, error) { // An actor can read template version parameters if they can read the related template. tv, err := q.db.GetTemplateVersionByID(ctx, templateVersionID) @@ -4287,15 +4176,6 @@ func (q *querier) GetWorkspaceBuildParameters(ctx context.Context, workspaceBuil return q.db.GetWorkspaceBuildParameters(ctx, workspaceBuildID) } -func (q *querier) GetWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIDs []uuid.UUID) ([]database.WorkspaceBuildParameter, error) { - prep, err := prepareSQLFilter(ctx, q.auth, policy.ActionRead, rbac.ResourceWorkspace.Type) - if err != nil { - return nil, xerrors.Errorf("(dev error) prepare sql filter: %w", err) - } - - return q.db.GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx, workspaceBuildIDs, prep) -} - func (q *querier) GetWorkspaceBuildProvisionerStateByID(ctx context.Context, buildID uuid.UUID) (database.GetWorkspaceBuildProvisionerStateByIDRow, error) { // Fetching the provisioner state requires Update permission on the template. return fetchWithAction(q.log, q.auth, policy.ActionUpdate, q.db.GetWorkspaceBuildProvisionerStateByID)(ctx, buildID) @@ -4943,16 +4823,6 @@ func (q *querier) InsertUserGroupsByID(ctx context.Context, arg database.InsertU return q.db.InsertUserGroupsByID(ctx, arg) } -func (q *querier) InsertUserGroupsByName(ctx context.Context, arg database.InsertUserGroupsByNameParams) error { - // This will add the user to all named groups. This counts as updating a group. - // NOTE: instead of checking if the user has permission to update each group, we instead - // check if the user has permission to update *a* group in the org. - fetch := func(_ context.Context, arg database.InsertUserGroupsByNameParams) (rbac.Objecter, error) { - return rbac.ResourceGroup.InOrg(arg.OrganizationID), nil - } - return update(q.log, q.auth, fetch, q.db.InsertUserGroupsByName)(ctx, arg) -} - // TODO: Should this be in system.go? func (q *querier) InsertUserLink(ctx context.Context, arg database.InsertUserLinkParams) (database.UserLink, error) { if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceUserObject(arg.UserID)); err != nil { @@ -5217,14 +5087,6 @@ func (q *querier) ListAIBridgeUserPromptsByInterceptionIDs(ctx context.Context, return q.db.ListAIBridgeUserPromptsByInterceptionIDs(ctx, interceptionIDs) } -func (q *querier) ListChatsByRootID(ctx context.Context, rootChatID uuid.UUID) ([]database.Chat, error) { - return fetchWithPostFilter(q.auth, policy.ActionRead, q.db.ListChatsByRootID)(ctx, rootChatID) -} - -func (q *querier) ListChildChatsByParentID(ctx context.Context, parentChatID uuid.UUID) ([]database.Chat, error) { - return fetchWithPostFilter(q.auth, policy.ActionRead, q.db.ListChildChatsByParentID)(ctx, parentChatID) -} - func (q *querier) ListProvisionerKeysByOrganization(ctx context.Context, organizationID uuid.UUID) ([]database.ProvisionerKey, error) { return fetchWithPostFilter(q.auth, policy.ActionRead, q.db.ListProvisionerKeysByOrganization)(ctx, organizationID) } @@ -5336,14 +5198,6 @@ func (q *querier) RegisterWorkspaceProxy(ctx context.Context, arg database.Regis return updateWithReturn(q.log, q.auth, fetch, q.db.RegisterWorkspaceProxy)(ctx, arg) } -func (q *querier) RemoveUserFromAllGroups(ctx context.Context, userID uuid.UUID) error { - // This is a system function to clear user groups in group sync. - if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { - return err - } - return q.db.RemoveUserFromAllGroups(ctx, userID) -} - func (q *querier) RemoveUserFromGroups(ctx context.Context, arg database.RemoveUserFromGroupsParams) ([]uuid.UUID, error) { // This is a system function to clear user groups in group sync. if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { @@ -5678,13 +5532,6 @@ func (q *querier) UpdateOAuth2ProviderAppByID(ctx context.Context, arg database. return q.db.UpdateOAuth2ProviderAppByID(ctx, arg) } -func (q *querier) UpdateOAuth2ProviderAppSecretByID(ctx context.Context, arg database.UpdateOAuth2ProviderAppSecretByIDParams) (database.OAuth2ProviderAppSecret, error) { - if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceOauth2AppSecret); err != nil { - return database.OAuth2ProviderAppSecret{}, err - } - return q.db.UpdateOAuth2ProviderAppSecretByID(ctx, arg) -} - func (q *querier) UpdateOrganization(ctx context.Context, arg database.UpdateOrganizationParams) (database.Organization, error) { fetch := func(ctx context.Context, arg database.UpdateOrganizationParams) (database.Organization, error) { return q.db.GetOrganizationByID(ctx, arg.ID) @@ -6137,13 +5984,6 @@ func (q *querier) UpdateUserLink(ctx context.Context, arg database.UpdateUserLin return fetchAndQuery(q.log, q.auth, policy.ActionUpdatePersonal, fetch, q.db.UpdateUserLink)(ctx, arg) } -func (q *querier) UpdateUserLinkedID(ctx context.Context, arg database.UpdateUserLinkedIDParams) (database.UserLink, error) { - if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { - return database.UserLink{}, err - } - return q.db.UpdateUserLinkedID(ctx, arg) -} - func (q *querier) UpdateUserLoginType(ctx context.Context, arg database.UpdateUserLoginTypeParams) (database.User, error) { if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { return database.User{}, err @@ -6556,13 +6396,6 @@ func (q *querier) UpsertAnnouncementBanners(ctx context.Context, value string) e return q.db.UpsertAnnouncementBanners(ctx, value) } -func (q *querier) UpsertAppSecurityKey(ctx context.Context, data string) error { - if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { - return err - } - return q.db.UpsertAppSecurityKey(ctx, data) -} - func (q *querier) UpsertApplicationName(ctx context.Context, value string) error { if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceDeploymentConfig); err != nil { return err @@ -6615,13 +6448,6 @@ func (q *querier) UpsertConnectionLog(ctx context.Context, arg database.UpsertCo return q.db.UpsertConnectionLog(ctx, arg) } -func (q *querier) UpsertCoordinatorResumeTokenSigningKey(ctx context.Context, value string) error { - if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { - return err - } - return q.db.UpsertCoordinatorResumeTokenSigningKey(ctx, value) -} - func (q *querier) UpsertDefaultProxy(ctx context.Context, arg database.UpsertDefaultProxyParams) error { if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { return err @@ -6671,13 +6497,6 @@ func (q *querier) UpsertOAuth2GithubDefaultEligible(ctx context.Context, eligibl return q.db.UpsertOAuth2GithubDefaultEligible(ctx, eligible) } -func (q *querier) UpsertOAuthSigningKey(ctx context.Context, value string) error { - if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil { - return err - } - return q.db.UpsertOAuthSigningKey(ctx, value) -} - func (q *querier) UpsertPrebuildsSettings(ctx context.Context, value string) error { if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceDeploymentConfig); err != nil { return err @@ -6867,10 +6686,6 @@ func (q *querier) GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx context.Context, return q.GetWorkspacesAndAgentsByOwnerID(ctx, ownerID) } -func (q *querier) GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIDs []uuid.UUID, _ rbac.PreparedAuthorized) ([]database.WorkspaceBuildParameter, error) { - return q.GetWorkspaceBuildParametersByBuildIDs(ctx, workspaceBuildIDs) -} - // GetAuthorizedUsers is not required for dbauthz since GetUsers is already // authenticated. func (q *querier) GetAuthorizedUsers(ctx context.Context, arg database.GetUsersParams, _ rbac.PreparedAuthorized) ([]database.GetUsersRow, error) { diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go index a765909dad..a9cb62ecd5 100644 --- a/coderd/database/dbauthz/dbauthz_test.go +++ b/coderd/database/dbauthz/dbauthz_test.go @@ -401,12 +401,6 @@ func (s *MethodTestSuite) TestChats() { dbm.EXPECT().UnarchiveChatByID(gomock.Any(), chat.ID).Return(nil).AnyTimes() check.Args(chat.ID).Asserts(chat, policy.ActionUpdate).Returns() })) - s.Run("DeleteChatMessagesByChatID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - chat := testutil.Fake(s.T(), faker, database.Chat{}) - dbm.EXPECT().GetChatByID(gomock.Any(), chat.ID).Return(chat, nil).AnyTimes() - dbm.EXPECT().DeleteChatMessagesByChatID(gomock.Any(), chat.ID).Return(nil).AnyTimes() - check.Args(chat.ID).Asserts(chat, policy.ActionDelete).Returns() - })) s.Run("DeleteChatMessagesAfterID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { chat := testutil.Fake(s.T(), faker, database.Chat{}) arg := database.DeleteChatMessagesAfterIDParams{ @@ -514,15 +508,6 @@ func (s *MethodTestSuite) TestChats() { dbm.EXPECT().GetDefaultChatModelConfig(gomock.Any()).Return(config, nil).AnyTimes() check.Asserts(rbac.ResourceDeploymentConfig, policy.ActionRead).Returns(config) })) - s.Run("GetChatModelConfigByProviderAndModel", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - config := testutil.Fake(s.T(), faker, database.ChatModelConfig{}) - args := database.GetChatModelConfigByProviderAndModelParams{ - Provider: config.Provider, - Model: config.Model, - } - dbm.EXPECT().GetChatModelConfigByProviderAndModel(gomock.Any(), args).Return(config, nil).AnyTimes() - check.Args(args).Asserts(rbac.ResourceDeploymentConfig, policy.ActionRead).Returns(config) - })) s.Run("GetChatModelConfigs", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { configA := testutil.Fake(s.T(), faker, database.ChatModelConfig{}) configB := testutil.Fake(s.T(), faker, database.ChatModelConfig{}) @@ -576,20 +561,6 @@ func (s *MethodTestSuite) TestChats() { dbm.EXPECT().GetEnabledChatProviders(gomock.Any()).Return([]database.ChatProvider{providerA, providerB}, nil).AnyTimes() check.Args().Asserts(rbac.ResourceDeploymentConfig, policy.ActionRead).Returns([]database.ChatProvider{providerA, providerB}) })) - s.Run("ListChatsByRootID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - rootChatID := uuid.New() - chatA := testutil.Fake(s.T(), faker, database.Chat{RootChatID: uuid.NullUUID{UUID: rootChatID, Valid: true}}) - chatB := testutil.Fake(s.T(), faker, database.Chat{RootChatID: uuid.NullUUID{UUID: rootChatID, Valid: true}}) - dbm.EXPECT().ListChatsByRootID(gomock.Any(), rootChatID).Return([]database.Chat{chatA, chatB}, nil).AnyTimes() - check.Args(rootChatID).Asserts(chatA, policy.ActionRead, chatB, policy.ActionRead).Returns([]database.Chat{chatA, chatB}) - })) - s.Run("ListChildChatsByParentID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - parentChatID := uuid.New() - chatA := testutil.Fake(s.T(), faker, database.Chat{ParentChatID: uuid.NullUUID{UUID: parentChatID, Valid: true}}) - chatB := testutil.Fake(s.T(), faker, database.Chat{ParentChatID: uuid.NullUUID{UUID: parentChatID, Valid: true}}) - dbm.EXPECT().ListChildChatsByParentID(gomock.Any(), parentChatID).Return([]database.Chat{chatA, chatB}, nil).AnyTimes() - check.Args(parentChatID).Asserts(chatA, policy.ActionRead, chatB, policy.ActionRead).Returns([]database.Chat{chatA, chatB}) - })) s.Run("GetStaleChats", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { threshold := dbtime.Now() chats := []database.Chat{testutil.Fake(s.T(), faker, database.Chat{})} @@ -806,12 +777,6 @@ func (s *MethodTestSuite) TestFile() { dbm.EXPECT().GetFileTemplates(gomock.Any(), f.ID).Return([]database.GetFileTemplatesRow{}, nil).AnyTimes() check.Args(f.ID).Asserts(f, policy.ActionRead).Returns(f) })) - s.Run("GetFileIDByTemplateVersionID", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - tvID := uuid.New() - fileID := uuid.New() - dbm.EXPECT().GetFileIDByTemplateVersionID(gomock.Any(), tvID).Return(fileID, nil).AnyTimes() - check.Args(tvID).Asserts(rbac.ResourceFile.WithID(fileID), policy.ActionRead).Returns(fileID) - })) s.Run("InsertFile", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { u := testutil.Fake(s.T(), faker, database.User{}) ret := testutil.Fake(s.T(), faker, database.File{CreatedBy: u.ID}) @@ -915,16 +880,6 @@ func (s *MethodTestSuite) TestGroup() { check.Args(arg).Asserts(g, policy.ActionUpdate).Returns() })) - s.Run("InsertUserGroupsByName", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - o := testutil.Fake(s.T(), faker, database.Organization{}) - u1 := testutil.Fake(s.T(), faker, database.User{}) - g1 := testutil.Fake(s.T(), faker, database.Group{OrganizationID: o.ID}) - g2 := testutil.Fake(s.T(), faker, database.Group{OrganizationID: o.ID}) - arg := database.InsertUserGroupsByNameParams{OrganizationID: o.ID, UserID: u1.ID, GroupNames: slice.New(g1.Name, g2.Name)} - dbm.EXPECT().InsertUserGroupsByName(gomock.Any(), arg).Return(nil).AnyTimes() - check.Args(arg).Asserts(rbac.ResourceGroup.InOrg(o.ID), policy.ActionUpdate).Returns() - })) - s.Run("InsertUserGroupsByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { o := testutil.Fake(s.T(), faker, database.Organization{}) u1 := testutil.Fake(s.T(), faker, database.User{}) @@ -937,12 +892,6 @@ func (s *MethodTestSuite) TestGroup() { check.Args(arg).Asserts(rbac.ResourceSystem, policy.ActionUpdate).Returns(returns) })) - s.Run("RemoveUserFromAllGroups", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - u1 := testutil.Fake(s.T(), faker, database.User{}) - dbm.EXPECT().RemoveUserFromAllGroups(gomock.Any(), u1.ID).Return(nil).AnyTimes() - check.Args(u1.ID).Asserts(rbac.ResourceSystem, policy.ActionUpdate).Returns() - })) - s.Run("RemoveUserFromGroups", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { o := testutil.Fake(s.T(), faker, database.Organization{}) u1 := testutil.Fake(s.T(), faker, database.User{}) @@ -1099,18 +1048,6 @@ func (s *MethodTestSuite) TestProvisionerJob() { dbm.EXPECT().UpdatePrebuildProvisionerJobWithCancel(gomock.Any(), arg).Return(canceledJobs, nil).AnyTimes() check.Args(arg).Asserts(rbac.ResourcePrebuiltWorkspace, policy.ActionUpdate).Returns(canceledJobs) })) - s.Run("GetProvisionerJobsByIDs", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - org := testutil.Fake(s.T(), faker, database.Organization{}) - org2 := testutil.Fake(s.T(), faker, database.Organization{}) - a := testutil.Fake(s.T(), faker, database.ProvisionerJob{OrganizationID: org.ID}) - b := testutil.Fake(s.T(), faker, database.ProvisionerJob{OrganizationID: org2.ID}) - ids := []uuid.UUID{a.ID, b.ID} - dbm.EXPECT().GetProvisionerJobsByIDs(gomock.Any(), ids).Return([]database.ProvisionerJob{a, b}, nil).AnyTimes() - check.Args(ids).Asserts( - rbac.ResourceProvisionerJobs.InOrg(org.ID), policy.ActionRead, - rbac.ResourceProvisionerJobs.InOrg(org2.ID), policy.ActionRead, - ).OutOfOrder().Returns(slice.New(a, b)) - })) s.Run("GetProvisionerLogsAfterID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { ws := testutil.Fake(s.T(), faker, database.Workspace{}) j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeWorkspaceBuild}) @@ -1574,14 +1511,6 @@ func (s *MethodTestSuite) TestTemplate() { dbm.EXPECT().GetTemplateVersionsCreatedAfter(gomock.Any(), now.Add(-time.Hour)).Return([]database.TemplateVersion{}, nil).AnyTimes() check.Args(now.Add(-time.Hour)).Asserts(rbac.ResourceTemplate.All(), policy.ActionRead) })) - s.Run("GetTemplateVersionHasAITask", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - t := testutil.Fake(s.T(), faker, database.Template{}) - tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: t.ID, Valid: true}}) - dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes() - dbm.EXPECT().GetTemplateByID(gomock.Any(), t.ID).Return(t, nil).AnyTimes() - dbm.EXPECT().GetTemplateVersionHasAITask(gomock.Any(), tv.ID).Return(false, nil).AnyTimes() - check.Args(tv.ID).Asserts(t, policy.ActionRead) - })) s.Run("GetTemplatesWithFilter", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { a := testutil.Fake(s.T(), faker, database.Template{}) arg := database.GetTemplatesWithFilterParams{} @@ -1965,12 +1894,6 @@ func (s *MethodTestSuite) TestUser() { dbm.EXPECT().UpdateUserStatus(gomock.Any(), arg).Return(u, nil).AnyTimes() check.Args(arg).Asserts(u, policy.ActionUpdate).Returns(u) })) - s.Run("DeleteGitSSHKey", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - key := testutil.Fake(s.T(), faker, database.GitSSHKey{}) - dbm.EXPECT().GetGitSSHKey(gomock.Any(), key.UserID).Return(key, nil).AnyTimes() - dbm.EXPECT().DeleteGitSSHKey(gomock.Any(), key.UserID).Return(nil).AnyTimes() - check.Args(key.UserID).Asserts(rbac.ResourceUserObject(key.UserID), policy.ActionUpdatePersonal).Returns() - })) s.Run("GetGitSSHKey", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { key := testutil.Fake(s.T(), faker, database.GitSSHKey{}) dbm.EXPECT().GetGitSSHKey(gomock.Any(), key.UserID).Return(key, nil).AnyTimes() @@ -2225,18 +2148,6 @@ func (s *MethodTestSuite) TestWorkspace() { // No asserts here because SQLFilter. check.Args(ws.OwnerID, emptyPreparedAuthorized{}).Asserts() })) - s.Run("GetWorkspaceBuildParametersByBuildIDs", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - ids := []uuid.UUID{} - dbm.EXPECT().GetAuthorizedWorkspaceBuildParametersByBuildIDs(gomock.Any(), ids, gomock.Any()).Return([]database.WorkspaceBuildParameter{}, nil).AnyTimes() - // no asserts here because SQLFilter - check.Args(ids).Asserts() - })) - s.Run("GetAuthorizedWorkspaceBuildParametersByBuildIDs", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - ids := []uuid.UUID{} - dbm.EXPECT().GetAuthorizedWorkspaceBuildParametersByBuildIDs(gomock.Any(), ids, gomock.Any()).Return([]database.WorkspaceBuildParameter{}, nil).AnyTimes() - // no asserts here because SQLFilter - check.Args(ids, emptyPreparedAuthorized{}).Asserts() - })) s.Run("GetWorkspaceACLByID", s.Mocked(func(dbM *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { ws := testutil.Fake(s.T(), faker, database.Workspace{}) dbM.EXPECT().GetWorkspaceByID(gomock.Any(), ws.ID).Return(ws, nil).AnyTimes() @@ -3398,13 +3309,6 @@ func (s *MethodTestSuite) TestCryptoKeys() { } func (s *MethodTestSuite) TestSystemFunctions() { - s.Run("UpdateUserLinkedID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - u := testutil.Fake(s.T(), faker, database.User{}) - l := testutil.Fake(s.T(), faker, database.UserLink{UserID: u.ID}) - arg := database.UpdateUserLinkedIDParams{UserID: u.ID, LinkedID: l.LinkedID, LoginType: database.LoginTypeGithub} - dbm.EXPECT().UpdateUserLinkedID(gomock.Any(), arg).Return(l, nil).AnyTimes() - check.Args(arg).Asserts(rbac.ResourceSystem, policy.ActionUpdate).Returns(l) - })) s.Run("GetLatestWorkspaceAppStatusByAppID", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { appID := uuid.New() dbm.EXPECT().GetLatestWorkspaceAppStatusByAppID(gomock.Any(), appID).Return(database.WorkspaceAppStatus{}, nil).AnyTimes() @@ -3597,16 +3501,6 @@ func (s *MethodTestSuite) TestSystemFunctions() { Asserts(rbac.ResourceSystem, policy.ActionRead). Returns([]database.WorkspaceAgent{agt}) })) - s.Run("GetProvisionerJobsByIDs", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { - org := testutil.Fake(s.T(), faker, database.Organization{}) - a := testutil.Fake(s.T(), faker, database.ProvisionerJob{OrganizationID: org.ID}) - b := testutil.Fake(s.T(), faker, database.ProvisionerJob{OrganizationID: org.ID}) - ids := []uuid.UUID{a.ID, b.ID} - dbm.EXPECT().GetProvisionerJobsByIDs(gomock.Any(), ids).Return([]database.ProvisionerJob{a, b}, nil).AnyTimes() - check.Args(ids). - Asserts(rbac.ResourceProvisionerJobs.InOrg(org.ID), policy.ActionRead). - Returns(slice.New(a, b)) - })) s.Run("DeleteWorkspaceSubAgentByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) { ws := testutil.Fake(s.T(), faker, database.Workspace{}) agent := testutil.Fake(s.T(), faker, database.WorkspaceAgent{}) @@ -3790,29 +3684,11 @@ func (s *MethodTestSuite) TestSystemFunctions() { dbm.EXPECT().InsertWorkspaceAgentLogSources(gomock.Any(), arg).Return([]database.WorkspaceAgentLogSource{}, nil).AnyTimes() check.Args(arg).Asserts() })) - s.Run("GetTemplateDAUs", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - arg := database.GetTemplateDAUsParams{} - dbm.EXPECT().GetTemplateDAUs(gomock.Any(), arg).Return([]database.GetTemplateDAUsRow{}, nil).AnyTimes() - check.Args(arg).Asserts(rbac.ResourceSystem, policy.ActionRead) - })) s.Run("GetActiveWorkspaceBuildsByTemplateID", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { id := uuid.New() dbm.EXPECT().GetActiveWorkspaceBuildsByTemplateID(gomock.Any(), id).Return([]database.WorkspaceBuild{}, nil).AnyTimes() check.Args(id).Asserts(rbac.ResourceSystem, policy.ActionRead).Returns([]database.WorkspaceBuild{}) })) - s.Run("GetDeploymentDAUs", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - tz := int32(0) - dbm.EXPECT().GetDeploymentDAUs(gomock.Any(), tz).Return([]database.GetDeploymentDAUsRow{}, nil).AnyTimes() - check.Args(tz).Asserts(rbac.ResourceSystem, policy.ActionRead) - })) - s.Run("GetAppSecurityKey", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - dbm.EXPECT().GetAppSecurityKey(gomock.Any()).Return("", sql.ErrNoRows).AnyTimes() - check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Errors(sql.ErrNoRows) - })) - s.Run("UpsertAppSecurityKey", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - dbm.EXPECT().UpsertAppSecurityKey(gomock.Any(), "foo").Return(nil).AnyTimes() - check.Args("foo").Asserts(rbac.ResourceSystem, policy.ActionUpdate) - })) s.Run("GetApplicationName", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { dbm.EXPECT().GetApplicationName(gomock.Any()).Return("foo", nil).AnyTimes() check.Args().Asserts() @@ -3866,22 +3742,6 @@ func (s *MethodTestSuite) TestSystemFunctions() { dbm.EXPECT().GetProvisionerJobsToBeReaped(gomock.Any(), arg).Return([]database.ProvisionerJob{}, nil).AnyTimes() check.Args(arg).Asserts(rbac.ResourceProvisionerJobs, policy.ActionRead) })) - s.Run("UpsertOAuthSigningKey", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - dbm.EXPECT().UpsertOAuthSigningKey(gomock.Any(), "foo").Return(nil).AnyTimes() - check.Args("foo").Asserts(rbac.ResourceSystem, policy.ActionUpdate) - })) - s.Run("GetOAuthSigningKey", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - dbm.EXPECT().GetOAuthSigningKey(gomock.Any()).Return("foo", nil).AnyTimes() - check.Args().Asserts(rbac.ResourceSystem, policy.ActionUpdate) - })) - s.Run("UpsertCoordinatorResumeTokenSigningKey", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - dbm.EXPECT().UpsertCoordinatorResumeTokenSigningKey(gomock.Any(), "foo").Return(nil).AnyTimes() - check.Args("foo").Asserts(rbac.ResourceSystem, policy.ActionUpdate) - })) - s.Run("GetCoordinatorResumeTokenSigningKey", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { - dbm.EXPECT().GetCoordinatorResumeTokenSigningKey(gomock.Any()).Return("foo", nil).AnyTimes() - check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) - })) s.Run("InsertMissingGroups", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) { arg := database.InsertMissingGroupsParams{} dbm.EXPECT().InsertMissingGroups(gomock.Any(), arg).Return([]database.Group{}, xerrors.New("any error")).AnyTimes() @@ -4535,12 +4395,6 @@ func (s *MethodTestSuite) TestOAuth2ProviderApps() { UpdatedAt: app.UpdatedAt, }).Asserts(rbac.ResourceOauth2App, policy.ActionUpdate).Returns(app) })) - s.Run("GetOAuth2ProviderAppByRegistrationToken", s.Subtest(func(db database.Store, check *expects) { - app := dbgen.OAuth2ProviderApp(s.T(), db, database.OAuth2ProviderApp{ - RegistrationAccessToken: []byte("test-token"), - }) - check.Args([]byte("test-token")).Asserts(rbac.ResourceOauth2App, policy.ActionRead).Returns(app) - })) } func (s *MethodTestSuite) TestOAuth2ProviderAppSecrets() { @@ -4585,18 +4439,6 @@ func (s *MethodTestSuite) TestOAuth2ProviderAppSecrets() { AppID: app.ID, }).Asserts(rbac.ResourceOauth2AppSecret, policy.ActionCreate) })) - s.Run("UpdateOAuth2ProviderAppSecretByID", s.Subtest(func(db database.Store, check *expects) { - dbtestutil.DisableForeignKeysAndTriggers(s.T(), db) - app := dbgen.OAuth2ProviderApp(s.T(), db, database.OAuth2ProviderApp{}) - secret := dbgen.OAuth2ProviderAppSecret(s.T(), db, database.OAuth2ProviderAppSecret{ - AppID: app.ID, - }) - secret.LastUsedAt = sql.NullTime{Time: dbtestutil.NowInDefaultTimezone(), Valid: true} - check.Args(database.UpdateOAuth2ProviderAppSecretByIDParams{ - ID: secret.ID, - LastUsedAt: secret.LastUsedAt, - }).Asserts(rbac.ResourceOauth2AppSecret, policy.ActionUpdate).Returns(secret) - })) s.Run("DeleteOAuth2ProviderAppSecretByID", s.Subtest(func(db database.Store, check *expects) { app := dbgen.OAuth2ProviderApp(s.T(), db, database.OAuth2ProviderApp{}) secret := dbgen.OAuth2ProviderAppSecret(s.T(), db, database.OAuth2ProviderAppSecret{ diff --git a/coderd/database/dbmetrics/querymetrics.go b/coderd/database/dbmetrics/querymetrics.go index 27c0d88181..c2286d0602 100644 --- a/coderd/database/dbmetrics/querymetrics.go +++ b/coderd/database/dbmetrics/querymetrics.go @@ -384,14 +384,6 @@ func (m queryMetricsStore) DeleteChatMessagesAfterID(ctx context.Context, arg da return r0 } -func (m queryMetricsStore) DeleteChatMessagesByChatID(ctx context.Context, chatID uuid.UUID) error { - start := time.Now() - r0 := m.s.DeleteChatMessagesByChatID(ctx, chatID) - m.queryLatencies.WithLabelValues("DeleteChatMessagesByChatID").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "DeleteChatMessagesByChatID").Inc() - return r0 -} - func (m queryMetricsStore) DeleteChatModelConfigByID(ctx context.Context, id uuid.UUID) error { start := time.Now() r0 := m.s.DeleteChatModelConfigByID(ctx, id) @@ -448,14 +440,6 @@ func (m queryMetricsStore) DeleteExternalAuthLink(ctx context.Context, arg datab return r0 } -func (m queryMetricsStore) DeleteGitSSHKey(ctx context.Context, userID uuid.UUID) error { - start := time.Now() - r0 := m.s.DeleteGitSSHKey(ctx, userID) - m.queryLatencies.WithLabelValues("DeleteGitSSHKey").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "DeleteGitSSHKey").Inc() - return r0 -} - func (m queryMetricsStore) DeleteGroupByID(ctx context.Context, id uuid.UUID) error { start := time.Now() r0 := m.s.DeleteGroupByID(ctx, id) @@ -951,14 +935,6 @@ func (m queryMetricsStore) GetAnnouncementBanners(ctx context.Context) (string, return r0, r1 } -func (m queryMetricsStore) GetAppSecurityKey(ctx context.Context) (string, error) { - start := time.Now() - r0, r1 := m.s.GetAppSecurityKey(ctx) - m.queryLatencies.WithLabelValues("GetAppSecurityKey").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetAppSecurityKey").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetApplicationName(ctx context.Context) (string, error) { start := time.Now() r0, r1 := m.s.GetApplicationName(ctx) @@ -1071,14 +1047,6 @@ func (m queryMetricsStore) GetChatModelConfigByID(ctx context.Context, id uuid.U return r0, r1 } -func (m queryMetricsStore) GetChatModelConfigByProviderAndModel(ctx context.Context, arg database.GetChatModelConfigByProviderAndModelParams) (database.ChatModelConfig, error) { - start := time.Now() - r0, r1 := m.s.GetChatModelConfigByProviderAndModel(ctx, arg) - m.queryLatencies.WithLabelValues("GetChatModelConfigByProviderAndModel").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetChatModelConfigByProviderAndModel").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetChatModelConfigs(ctx context.Context) ([]database.ChatModelConfig, error) { start := time.Now() r0, r1 := m.s.GetChatModelConfigs(ctx) @@ -1143,14 +1111,6 @@ func (m queryMetricsStore) GetConnectionLogsOffset(ctx context.Context, arg data return r0, r1 } -func (m queryMetricsStore) GetCoordinatorResumeTokenSigningKey(ctx context.Context) (string, error) { - start := time.Now() - r0, r1 := m.s.GetCoordinatorResumeTokenSigningKey(ctx) - m.queryLatencies.WithLabelValues("GetCoordinatorResumeTokenSigningKey").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetCoordinatorResumeTokenSigningKey").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetCryptoKeyByFeatureAndSequence(ctx context.Context, arg database.GetCryptoKeyByFeatureAndSequenceParams) (database.CryptoKey, error) { start := time.Now() r0, r1 := m.s.GetCryptoKeyByFeatureAndSequence(ctx, arg) @@ -1215,14 +1175,6 @@ func (m queryMetricsStore) GetDefaultProxyConfig(ctx context.Context) (database. return r0, r1 } -func (m queryMetricsStore) GetDeploymentDAUs(ctx context.Context, tzOffset int32) ([]database.GetDeploymentDAUsRow, error) { - start := time.Now() - r0, r1 := m.s.GetDeploymentDAUs(ctx, tzOffset) - m.queryLatencies.WithLabelValues("GetDeploymentDAUs").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetDeploymentDAUs").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetDeploymentID(ctx context.Context) (string, error) { start := time.Now() r0, r1 := m.s.GetDeploymentID(ctx) @@ -1319,14 +1271,6 @@ func (m queryMetricsStore) GetFileByID(ctx context.Context, id uuid.UUID) (datab return r0, r1 } -func (m queryMetricsStore) GetFileIDByTemplateVersionID(ctx context.Context, templateVersionID uuid.UUID) (uuid.UUID, error) { - start := time.Now() - r0, r1 := m.s.GetFileIDByTemplateVersionID(ctx, templateVersionID) - m.queryLatencies.WithLabelValues("GetFileIDByTemplateVersionID").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetFileIDByTemplateVersionID").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetFileTemplates(ctx context.Context, fileID uuid.UUID) ([]database.GetFileTemplatesRow, error) { start := time.Now() r0, r1 := m.s.GetFileTemplates(ctx, fileID) @@ -1567,14 +1511,6 @@ func (m queryMetricsStore) GetOAuth2ProviderAppByID(ctx context.Context, id uuid return r0, r1 } -func (m queryMetricsStore) GetOAuth2ProviderAppByRegistrationToken(ctx context.Context, registrationAccessToken []byte) (database.OAuth2ProviderApp, error) { - start := time.Now() - r0, r1 := m.s.GetOAuth2ProviderAppByRegistrationToken(ctx, registrationAccessToken) - m.queryLatencies.WithLabelValues("GetOAuth2ProviderAppByRegistrationToken").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetOAuth2ProviderAppByRegistrationToken").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetOAuth2ProviderAppCodeByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderAppCode, error) { start := time.Now() r0, r1 := m.s.GetOAuth2ProviderAppCodeByID(ctx, id) @@ -1647,14 +1583,6 @@ func (m queryMetricsStore) GetOAuth2ProviderAppsByUserID(ctx context.Context, us return r0, r1 } -func (m queryMetricsStore) GetOAuthSigningKey(ctx context.Context) (string, error) { - start := time.Now() - r0, r1 := m.s.GetOAuthSigningKey(ctx) - m.queryLatencies.WithLabelValues("GetOAuthSigningKey").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetOAuthSigningKey").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetOrganizationByID(ctx context.Context, id uuid.UUID) (database.Organization, error) { start := time.Now() r0, r1 := m.s.GetOrganizationByID(ctx, id) @@ -1855,14 +1783,6 @@ func (m queryMetricsStore) GetProvisionerJobTimingsByJobID(ctx context.Context, return r0, r1 } -func (m queryMetricsStore) GetProvisionerJobsByIDs(ctx context.Context, ids []uuid.UUID) ([]database.ProvisionerJob, error) { - start := time.Now() - r0, r1 := m.s.GetProvisionerJobsByIDs(ctx, ids) - m.queryLatencies.WithLabelValues("GetProvisionerJobsByIDs").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetProvisionerJobsByIDs").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetProvisionerJobsByIDsWithQueuePosition(ctx context.Context, arg database.GetProvisionerJobsByIDsWithQueuePositionParams) ([]database.GetProvisionerJobsByIDsWithQueuePositionRow, error) { start := time.Now() r0, r1 := m.s.GetProvisionerJobsByIDsWithQueuePosition(ctx, arg) @@ -2111,14 +2031,6 @@ func (m queryMetricsStore) GetTemplateByOrganizationAndName(ctx context.Context, return r0, r1 } -func (m queryMetricsStore) GetTemplateDAUs(ctx context.Context, arg database.GetTemplateDAUsParams) ([]database.GetTemplateDAUsRow, error) { - start := time.Now() - r0, r1 := m.s.GetTemplateDAUs(ctx, arg) - m.queryLatencies.WithLabelValues("GetTemplateDAUs").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetTemplateDAUs").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetTemplateInsights(ctx context.Context, arg database.GetTemplateInsightsParams) (database.GetTemplateInsightsRow, error) { start := time.Now() r0, r1 := m.s.GetTemplateInsights(ctx, arg) @@ -2191,14 +2103,6 @@ func (m queryMetricsStore) GetTemplateVersionByTemplateIDAndName(ctx context.Con return r0, r1 } -func (m queryMetricsStore) GetTemplateVersionHasAITask(ctx context.Context, id uuid.UUID) (bool, error) { - start := time.Now() - r0, r1 := m.s.GetTemplateVersionHasAITask(ctx, id) - m.queryLatencies.WithLabelValues("GetTemplateVersionHasAITask").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetTemplateVersionHasAITask").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetTemplateVersionParameters(ctx context.Context, templateVersionID uuid.UUID) ([]database.TemplateVersionParameter, error) { start := time.Now() r0, r1 := m.s.GetTemplateVersionParameters(ctx, templateVersionID) @@ -2711,14 +2615,6 @@ func (m queryMetricsStore) GetWorkspaceBuildParameters(ctx context.Context, work return r0, r1 } -func (m queryMetricsStore) GetWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIds []uuid.UUID) ([]database.WorkspaceBuildParameter, error) { - start := time.Now() - r0, r1 := m.s.GetWorkspaceBuildParametersByBuildIDs(ctx, workspaceBuildIds) - m.queryLatencies.WithLabelValues("GetWorkspaceBuildParametersByBuildIDs").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetWorkspaceBuildParametersByBuildIDs").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetWorkspaceBuildProvisionerStateByID(ctx context.Context, workspaceBuildID uuid.UUID) (database.GetWorkspaceBuildProvisionerStateByIDRow, error) { start := time.Now() r0, r1 := m.s.GetWorkspaceBuildProvisionerStateByID(ctx, workspaceBuildID) @@ -3359,14 +3255,6 @@ func (m queryMetricsStore) InsertUserGroupsByID(ctx context.Context, arg databas return r0, r1 } -func (m queryMetricsStore) InsertUserGroupsByName(ctx context.Context, arg database.InsertUserGroupsByNameParams) error { - start := time.Now() - r0 := m.s.InsertUserGroupsByName(ctx, arg) - m.queryLatencies.WithLabelValues("InsertUserGroupsByName").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "InsertUserGroupsByName").Inc() - return r0 -} - func (m queryMetricsStore) InsertUserLink(ctx context.Context, arg database.InsertUserLinkParams) (database.UserLink, error) { start := time.Now() r0, r1 := m.s.InsertUserLink(ctx, arg) @@ -3575,22 +3463,6 @@ func (m queryMetricsStore) ListAIBridgeUserPromptsByInterceptionIDs(ctx context. return r0, r1 } -func (m queryMetricsStore) ListChatsByRootID(ctx context.Context, rootChatID uuid.UUID) ([]database.Chat, error) { - start := time.Now() - r0, r1 := m.s.ListChatsByRootID(ctx, rootChatID) - m.queryLatencies.WithLabelValues("ListChatsByRootID").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "ListChatsByRootID").Inc() - return r0, r1 -} - -func (m queryMetricsStore) ListChildChatsByParentID(ctx context.Context, parentChatID uuid.UUID) ([]database.Chat, error) { - start := time.Now() - r0, r1 := m.s.ListChildChatsByParentID(ctx, parentChatID) - m.queryLatencies.WithLabelValues("ListChildChatsByParentID").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "ListChildChatsByParentID").Inc() - return r0, r1 -} - func (m queryMetricsStore) ListProvisionerKeysByOrganization(ctx context.Context, organizationID uuid.UUID) ([]database.ProvisionerKey, error) { start := time.Now() r0, r1 := m.s.ListProvisionerKeysByOrganization(ctx, organizationID) @@ -3695,14 +3567,6 @@ func (m queryMetricsStore) RegisterWorkspaceProxy(ctx context.Context, arg datab return r0, r1 } -func (m queryMetricsStore) RemoveUserFromAllGroups(ctx context.Context, userID uuid.UUID) error { - start := time.Now() - r0 := m.s.RemoveUserFromAllGroups(ctx, userID) - m.queryLatencies.WithLabelValues("RemoveUserFromAllGroups").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "RemoveUserFromAllGroups").Inc() - return r0 -} - func (m queryMetricsStore) RemoveUserFromGroups(ctx context.Context, arg database.RemoveUserFromGroupsParams) ([]uuid.UUID, error) { start := time.Now() r0, r1 := m.s.RemoveUserFromGroups(ctx, arg) @@ -3943,14 +3807,6 @@ func (m queryMetricsStore) UpdateOAuth2ProviderAppByID(ctx context.Context, arg return r0, r1 } -func (m queryMetricsStore) UpdateOAuth2ProviderAppSecretByID(ctx context.Context, arg database.UpdateOAuth2ProviderAppSecretByIDParams) (database.OAuth2ProviderAppSecret, error) { - start := time.Now() - r0, r1 := m.s.UpdateOAuth2ProviderAppSecretByID(ctx, arg) - m.queryLatencies.WithLabelValues("UpdateOAuth2ProviderAppSecretByID").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "UpdateOAuth2ProviderAppSecretByID").Inc() - return r0, r1 -} - func (m queryMetricsStore) UpdateOrganization(ctx context.Context, arg database.UpdateOrganizationParams) (database.Organization, error) { start := time.Now() r0, r1 := m.s.UpdateOrganization(ctx, arg) @@ -4238,14 +4094,6 @@ func (m queryMetricsStore) UpdateUserLink(ctx context.Context, arg database.Upda return r0, r1 } -func (m queryMetricsStore) UpdateUserLinkedID(ctx context.Context, arg database.UpdateUserLinkedIDParams) (database.UserLink, error) { - start := time.Now() - r0, r1 := m.s.UpdateUserLinkedID(ctx, arg) - m.queryLatencies.WithLabelValues("UpdateUserLinkedID").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "UpdateUserLinkedID").Inc() - return r0, r1 -} - func (m queryMetricsStore) UpdateUserLoginType(ctx context.Context, arg database.UpdateUserLoginTypeParams) (database.User, error) { start := time.Now() r0, r1 := m.s.UpdateUserLoginType(ctx, arg) @@ -4534,14 +4382,6 @@ func (m queryMetricsStore) UpsertAnnouncementBanners(ctx context.Context, value return r0 } -func (m queryMetricsStore) UpsertAppSecurityKey(ctx context.Context, value string) error { - start := time.Now() - r0 := m.s.UpsertAppSecurityKey(ctx, value) - m.queryLatencies.WithLabelValues("UpsertAppSecurityKey").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "UpsertAppSecurityKey").Inc() - return r0 -} - func (m queryMetricsStore) UpsertApplicationName(ctx context.Context, value string) error { start := time.Now() r0 := m.s.UpsertApplicationName(ctx, value) @@ -4590,14 +4430,6 @@ func (m queryMetricsStore) UpsertConnectionLog(ctx context.Context, arg database return r0, r1 } -func (m queryMetricsStore) UpsertCoordinatorResumeTokenSigningKey(ctx context.Context, value string) error { - start := time.Now() - r0 := m.s.UpsertCoordinatorResumeTokenSigningKey(ctx, value) - m.queryLatencies.WithLabelValues("UpsertCoordinatorResumeTokenSigningKey").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "UpsertCoordinatorResumeTokenSigningKey").Inc() - return r0 -} - func (m queryMetricsStore) UpsertDefaultProxy(ctx context.Context, arg database.UpsertDefaultProxyParams) error { start := time.Now() r0 := m.s.UpsertDefaultProxy(ctx, arg) @@ -4654,14 +4486,6 @@ func (m queryMetricsStore) UpsertOAuth2GithubDefaultEligible(ctx context.Context return r0 } -func (m queryMetricsStore) UpsertOAuthSigningKey(ctx context.Context, value string) error { - start := time.Now() - r0 := m.s.UpsertOAuthSigningKey(ctx, value) - m.queryLatencies.WithLabelValues("UpsertOAuthSigningKey").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "UpsertOAuthSigningKey").Inc() - return r0 -} - func (m queryMetricsStore) UpsertPrebuildsSettings(ctx context.Context, value string) error { start := time.Now() r0 := m.s.UpsertPrebuildsSettings(ctx, value) @@ -4830,14 +4654,6 @@ func (m queryMetricsStore) GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx context return r0, r1 } -func (m queryMetricsStore) GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIDs []uuid.UUID, prepared rbac.PreparedAuthorized) ([]database.WorkspaceBuildParameter, error) { - start := time.Now() - r0, r1 := m.s.GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx, workspaceBuildIDs, prepared) - m.queryLatencies.WithLabelValues("GetAuthorizedWorkspaceBuildParametersByBuildIDs").Observe(time.Since(start).Seconds()) - m.queryCounts.WithLabelValues(httpmw.ExtractHTTPRoute(ctx), httpmw.ExtractHTTPMethod(ctx), "GetAuthorizedWorkspaceBuildParametersByBuildIDs").Inc() - return r0, r1 -} - func (m queryMetricsStore) GetAuthorizedUsers(ctx context.Context, arg database.GetUsersParams, prepared rbac.PreparedAuthorized) ([]database.GetUsersRow, error) { start := time.Now() r0, r1 := m.s.GetAuthorizedUsers(ctx, arg, prepared) diff --git a/coderd/database/dbmock/dbmock.go b/coderd/database/dbmock/dbmock.go index 618c63a770..89c8493c85 100644 --- a/coderd/database/dbmock/dbmock.go +++ b/coderd/database/dbmock/dbmock.go @@ -597,20 +597,6 @@ func (mr *MockStoreMockRecorder) DeleteChatMessagesAfterID(ctx, arg any) *gomock return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteChatMessagesAfterID", reflect.TypeOf((*MockStore)(nil).DeleteChatMessagesAfterID), ctx, arg) } -// DeleteChatMessagesByChatID mocks base method. -func (m *MockStore) DeleteChatMessagesByChatID(ctx context.Context, chatID uuid.UUID) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteChatMessagesByChatID", ctx, chatID) - ret0, _ := ret[0].(error) - return ret0 -} - -// DeleteChatMessagesByChatID indicates an expected call of DeleteChatMessagesByChatID. -func (mr *MockStoreMockRecorder) DeleteChatMessagesByChatID(ctx, chatID any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteChatMessagesByChatID", reflect.TypeOf((*MockStore)(nil).DeleteChatMessagesByChatID), ctx, chatID) -} - // DeleteChatModelConfigByID mocks base method. func (m *MockStore) DeleteChatModelConfigByID(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() @@ -711,20 +697,6 @@ func (mr *MockStoreMockRecorder) DeleteExternalAuthLink(ctx, arg any) *gomock.Ca return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteExternalAuthLink", reflect.TypeOf((*MockStore)(nil).DeleteExternalAuthLink), ctx, arg) } -// DeleteGitSSHKey mocks base method. -func (m *MockStore) DeleteGitSSHKey(ctx context.Context, userID uuid.UUID) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteGitSSHKey", ctx, userID) - ret0, _ := ret[0].(error) - return ret0 -} - -// DeleteGitSSHKey indicates an expected call of DeleteGitSSHKey. -func (mr *MockStoreMockRecorder) DeleteGitSSHKey(ctx, userID any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteGitSSHKey", reflect.TypeOf((*MockStore)(nil).DeleteGitSSHKey), ctx, userID) -} - // DeleteGroupByID mocks base method. func (m *MockStore) DeleteGroupByID(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() @@ -1626,21 +1598,6 @@ func (mr *MockStoreMockRecorder) GetAnnouncementBanners(ctx any) *gomock.Call { return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAnnouncementBanners", reflect.TypeOf((*MockStore)(nil).GetAnnouncementBanners), ctx) } -// GetAppSecurityKey mocks base method. -func (m *MockStore) GetAppSecurityKey(ctx context.Context) (string, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAppSecurityKey", ctx) - ret0, _ := ret[0].(string) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetAppSecurityKey indicates an expected call of GetAppSecurityKey. -func (mr *MockStoreMockRecorder) GetAppSecurityKey(ctx any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAppSecurityKey", reflect.TypeOf((*MockStore)(nil).GetAppSecurityKey), ctx) -} - // GetApplicationName mocks base method. func (m *MockStore) GetApplicationName(ctx context.Context) (string, error) { m.ctrl.T.Helper() @@ -1761,21 +1718,6 @@ func (mr *MockStoreMockRecorder) GetAuthorizedUsers(ctx, arg, prepared any) *gom return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedUsers", reflect.TypeOf((*MockStore)(nil).GetAuthorizedUsers), ctx, arg, prepared) } -// GetAuthorizedWorkspaceBuildParametersByBuildIDs mocks base method. -func (m *MockStore) GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIDs []uuid.UUID, prepared rbac.PreparedAuthorized) ([]database.WorkspaceBuildParameter, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuthorizedWorkspaceBuildParametersByBuildIDs", ctx, workspaceBuildIDs, prepared) - ret0, _ := ret[0].([]database.WorkspaceBuildParameter) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetAuthorizedWorkspaceBuildParametersByBuildIDs indicates an expected call of GetAuthorizedWorkspaceBuildParametersByBuildIDs. -func (mr *MockStoreMockRecorder) GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx, workspaceBuildIDs, prepared any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedWorkspaceBuildParametersByBuildIDs", reflect.TypeOf((*MockStore)(nil).GetAuthorizedWorkspaceBuildParametersByBuildIDs), ctx, workspaceBuildIDs, prepared) -} - // GetAuthorizedWorkspaces mocks base method. func (m *MockStore) GetAuthorizedWorkspaces(ctx context.Context, arg database.GetWorkspacesParams, prepared rbac.PreparedAuthorized) ([]database.GetWorkspacesRow, error) { m.ctrl.T.Helper() @@ -1956,21 +1898,6 @@ func (mr *MockStoreMockRecorder) GetChatModelConfigByID(ctx, id any) *gomock.Cal return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChatModelConfigByID", reflect.TypeOf((*MockStore)(nil).GetChatModelConfigByID), ctx, id) } -// GetChatModelConfigByProviderAndModel mocks base method. -func (m *MockStore) GetChatModelConfigByProviderAndModel(ctx context.Context, arg database.GetChatModelConfigByProviderAndModelParams) (database.ChatModelConfig, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetChatModelConfigByProviderAndModel", ctx, arg) - ret0, _ := ret[0].(database.ChatModelConfig) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetChatModelConfigByProviderAndModel indicates an expected call of GetChatModelConfigByProviderAndModel. -func (mr *MockStoreMockRecorder) GetChatModelConfigByProviderAndModel(ctx, arg any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetChatModelConfigByProviderAndModel", reflect.TypeOf((*MockStore)(nil).GetChatModelConfigByProviderAndModel), ctx, arg) -} - // GetChatModelConfigs mocks base method. func (m *MockStore) GetChatModelConfigs(ctx context.Context) ([]database.ChatModelConfig, error) { m.ctrl.T.Helper() @@ -2091,21 +2018,6 @@ func (mr *MockStoreMockRecorder) GetConnectionLogsOffset(ctx, arg any) *gomock.C return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConnectionLogsOffset", reflect.TypeOf((*MockStore)(nil).GetConnectionLogsOffset), ctx, arg) } -// GetCoordinatorResumeTokenSigningKey mocks base method. -func (m *MockStore) GetCoordinatorResumeTokenSigningKey(ctx context.Context) (string, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCoordinatorResumeTokenSigningKey", ctx) - ret0, _ := ret[0].(string) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetCoordinatorResumeTokenSigningKey indicates an expected call of GetCoordinatorResumeTokenSigningKey. -func (mr *MockStoreMockRecorder) GetCoordinatorResumeTokenSigningKey(ctx any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCoordinatorResumeTokenSigningKey", reflect.TypeOf((*MockStore)(nil).GetCoordinatorResumeTokenSigningKey), ctx) -} - // GetCryptoKeyByFeatureAndSequence mocks base method. func (m *MockStore) GetCryptoKeyByFeatureAndSequence(ctx context.Context, arg database.GetCryptoKeyByFeatureAndSequenceParams) (database.CryptoKey, error) { m.ctrl.T.Helper() @@ -2226,21 +2138,6 @@ func (mr *MockStoreMockRecorder) GetDefaultProxyConfig(ctx any) *gomock.Call { return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDefaultProxyConfig", reflect.TypeOf((*MockStore)(nil).GetDefaultProxyConfig), ctx) } -// GetDeploymentDAUs mocks base method. -func (m *MockStore) GetDeploymentDAUs(ctx context.Context, tzOffset int32) ([]database.GetDeploymentDAUsRow, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDeploymentDAUs", ctx, tzOffset) - ret0, _ := ret[0].([]database.GetDeploymentDAUsRow) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetDeploymentDAUs indicates an expected call of GetDeploymentDAUs. -func (mr *MockStoreMockRecorder) GetDeploymentDAUs(ctx, tzOffset any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentDAUs", reflect.TypeOf((*MockStore)(nil).GetDeploymentDAUs), ctx, tzOffset) -} - // GetDeploymentID mocks base method. func (m *MockStore) GetDeploymentID(ctx context.Context) (string, error) { m.ctrl.T.Helper() @@ -2421,21 +2318,6 @@ func (mr *MockStoreMockRecorder) GetFileByID(ctx, id any) *gomock.Call { return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileByID", reflect.TypeOf((*MockStore)(nil).GetFileByID), ctx, id) } -// GetFileIDByTemplateVersionID mocks base method. -func (m *MockStore) GetFileIDByTemplateVersionID(ctx context.Context, templateVersionID uuid.UUID) (uuid.UUID, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetFileIDByTemplateVersionID", ctx, templateVersionID) - ret0, _ := ret[0].(uuid.UUID) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetFileIDByTemplateVersionID indicates an expected call of GetFileIDByTemplateVersionID. -func (mr *MockStoreMockRecorder) GetFileIDByTemplateVersionID(ctx, templateVersionID any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileIDByTemplateVersionID", reflect.TypeOf((*MockStore)(nil).GetFileIDByTemplateVersionID), ctx, templateVersionID) -} - // GetFileTemplates mocks base method. func (m *MockStore) GetFileTemplates(ctx context.Context, fileID uuid.UUID) ([]database.GetFileTemplatesRow, error) { m.ctrl.T.Helper() @@ -2886,21 +2768,6 @@ func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppByID(ctx, id any) *gomock.C return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppByID), ctx, id) } -// GetOAuth2ProviderAppByRegistrationToken mocks base method. -func (m *MockStore) GetOAuth2ProviderAppByRegistrationToken(ctx context.Context, registrationAccessToken []byte) (database.OAuth2ProviderApp, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppByRegistrationToken", ctx, registrationAccessToken) - ret0, _ := ret[0].(database.OAuth2ProviderApp) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetOAuth2ProviderAppByRegistrationToken indicates an expected call of GetOAuth2ProviderAppByRegistrationToken. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppByRegistrationToken(ctx, registrationAccessToken any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppByRegistrationToken", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppByRegistrationToken), ctx, registrationAccessToken) -} - // GetOAuth2ProviderAppCodeByID mocks base method. func (m *MockStore) GetOAuth2ProviderAppCodeByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderAppCode, error) { m.ctrl.T.Helper() @@ -3036,21 +2903,6 @@ func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppsByUserID(ctx, userID any) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppsByUserID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppsByUserID), ctx, userID) } -// GetOAuthSigningKey mocks base method. -func (m *MockStore) GetOAuthSigningKey(ctx context.Context) (string, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuthSigningKey", ctx) - ret0, _ := ret[0].(string) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetOAuthSigningKey indicates an expected call of GetOAuthSigningKey. -func (mr *MockStoreMockRecorder) GetOAuthSigningKey(ctx any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuthSigningKey", reflect.TypeOf((*MockStore)(nil).GetOAuthSigningKey), ctx) -} - // GetOrganizationByID mocks base method. func (m *MockStore) GetOrganizationByID(ctx context.Context, id uuid.UUID) (database.Organization, error) { m.ctrl.T.Helper() @@ -3426,21 +3278,6 @@ func (mr *MockStoreMockRecorder) GetProvisionerJobTimingsByJobID(ctx, jobID any) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobTimingsByJobID", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobTimingsByJobID), ctx, jobID) } -// GetProvisionerJobsByIDs mocks base method. -func (m *MockStore) GetProvisionerJobsByIDs(ctx context.Context, ids []uuid.UUID) ([]database.ProvisionerJob, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerJobsByIDs", ctx, ids) - ret0, _ := ret[0].([]database.ProvisionerJob) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetProvisionerJobsByIDs indicates an expected call of GetProvisionerJobsByIDs. -func (mr *MockStoreMockRecorder) GetProvisionerJobsByIDs(ctx, ids any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobsByIDs", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobsByIDs), ctx, ids) -} - // GetProvisionerJobsByIDsWithQueuePosition mocks base method. func (m *MockStore) GetProvisionerJobsByIDsWithQueuePosition(ctx context.Context, arg database.GetProvisionerJobsByIDsWithQueuePositionParams) ([]database.GetProvisionerJobsByIDsWithQueuePositionRow, error) { m.ctrl.T.Helper() @@ -3906,21 +3743,6 @@ func (mr *MockStoreMockRecorder) GetTemplateByOrganizationAndName(ctx, arg any) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateByOrganizationAndName", reflect.TypeOf((*MockStore)(nil).GetTemplateByOrganizationAndName), ctx, arg) } -// GetTemplateDAUs mocks base method. -func (m *MockStore) GetTemplateDAUs(ctx context.Context, arg database.GetTemplateDAUsParams) ([]database.GetTemplateDAUsRow, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateDAUs", ctx, arg) - ret0, _ := ret[0].([]database.GetTemplateDAUsRow) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetTemplateDAUs indicates an expected call of GetTemplateDAUs. -func (mr *MockStoreMockRecorder) GetTemplateDAUs(ctx, arg any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateDAUs", reflect.TypeOf((*MockStore)(nil).GetTemplateDAUs), ctx, arg) -} - // GetTemplateGroupRoles mocks base method. func (m *MockStore) GetTemplateGroupRoles(ctx context.Context, id uuid.UUID) ([]database.TemplateGroup, error) { m.ctrl.T.Helper() @@ -4086,21 +3908,6 @@ func (mr *MockStoreMockRecorder) GetTemplateVersionByTemplateIDAndName(ctx, arg return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionByTemplateIDAndName", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionByTemplateIDAndName), ctx, arg) } -// GetTemplateVersionHasAITask mocks base method. -func (m *MockStore) GetTemplateVersionHasAITask(ctx context.Context, id uuid.UUID) (bool, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionHasAITask", ctx, id) - ret0, _ := ret[0].(bool) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetTemplateVersionHasAITask indicates an expected call of GetTemplateVersionHasAITask. -func (mr *MockStoreMockRecorder) GetTemplateVersionHasAITask(ctx, id any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionHasAITask", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionHasAITask), ctx, id) -} - // GetTemplateVersionParameters mocks base method. func (m *MockStore) GetTemplateVersionParameters(ctx context.Context, templateVersionID uuid.UUID) ([]database.TemplateVersionParameter, error) { m.ctrl.T.Helper() @@ -5061,21 +4868,6 @@ func (mr *MockStoreMockRecorder) GetWorkspaceBuildParameters(ctx, workspaceBuild return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildParameters", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildParameters), ctx, workspaceBuildID) } -// GetWorkspaceBuildParametersByBuildIDs mocks base method. -func (m *MockStore) GetWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIds []uuid.UUID) ([]database.WorkspaceBuildParameter, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildParametersByBuildIDs", ctx, workspaceBuildIds) - ret0, _ := ret[0].([]database.WorkspaceBuildParameter) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetWorkspaceBuildParametersByBuildIDs indicates an expected call of GetWorkspaceBuildParametersByBuildIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildParametersByBuildIDs(ctx, workspaceBuildIds any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildParametersByBuildIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildParametersByBuildIDs), ctx, workspaceBuildIds) -} - // GetWorkspaceBuildProvisionerStateByID mocks base method. func (m *MockStore) GetWorkspaceBuildProvisionerStateByID(ctx context.Context, workspaceBuildID uuid.UUID) (database.GetWorkspaceBuildProvisionerStateByIDRow, error) { m.ctrl.T.Helper() @@ -6280,20 +6072,6 @@ func (mr *MockStoreMockRecorder) InsertUserGroupsByID(ctx, arg any) *gomock.Call return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserGroupsByID", reflect.TypeOf((*MockStore)(nil).InsertUserGroupsByID), ctx, arg) } -// InsertUserGroupsByName mocks base method. -func (m *MockStore) InsertUserGroupsByName(ctx context.Context, arg database.InsertUserGroupsByNameParams) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertUserGroupsByName", ctx, arg) - ret0, _ := ret[0].(error) - return ret0 -} - -// InsertUserGroupsByName indicates an expected call of InsertUserGroupsByName. -func (mr *MockStoreMockRecorder) InsertUserGroupsByName(ctx, arg any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserGroupsByName", reflect.TypeOf((*MockStore)(nil).InsertUserGroupsByName), ctx, arg) -} - // InsertUserLink mocks base method. func (m *MockStore) InsertUserLink(ctx context.Context, arg database.InsertUserLinkParams) (database.UserLink, error) { m.ctrl.T.Helper() @@ -6709,36 +6487,6 @@ func (mr *MockStoreMockRecorder) ListAuthorizedAIBridgeModels(ctx, arg, prepared return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAuthorizedAIBridgeModels", reflect.TypeOf((*MockStore)(nil).ListAuthorizedAIBridgeModels), ctx, arg, prepared) } -// ListChatsByRootID mocks base method. -func (m *MockStore) ListChatsByRootID(ctx context.Context, rootChatID uuid.UUID) ([]database.Chat, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListChatsByRootID", ctx, rootChatID) - ret0, _ := ret[0].([]database.Chat) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ListChatsByRootID indicates an expected call of ListChatsByRootID. -func (mr *MockStoreMockRecorder) ListChatsByRootID(ctx, rootChatID any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListChatsByRootID", reflect.TypeOf((*MockStore)(nil).ListChatsByRootID), ctx, rootChatID) -} - -// ListChildChatsByParentID mocks base method. -func (m *MockStore) ListChildChatsByParentID(ctx context.Context, parentChatID uuid.UUID) ([]database.Chat, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListChildChatsByParentID", ctx, parentChatID) - ret0, _ := ret[0].([]database.Chat) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ListChildChatsByParentID indicates an expected call of ListChildChatsByParentID. -func (mr *MockStoreMockRecorder) ListChildChatsByParentID(ctx, parentChatID any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListChildChatsByParentID", reflect.TypeOf((*MockStore)(nil).ListChildChatsByParentID), ctx, parentChatID) -} - // ListProvisionerKeysByOrganization mocks base method. func (m *MockStore) ListProvisionerKeysByOrganization(ctx context.Context, organizationID uuid.UUID) ([]database.ProvisionerKey, error) { m.ctrl.T.Helper() @@ -6962,20 +6710,6 @@ func (mr *MockStoreMockRecorder) RegisterWorkspaceProxy(ctx, arg any) *gomock.Ca return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterWorkspaceProxy", reflect.TypeOf((*MockStore)(nil).RegisterWorkspaceProxy), ctx, arg) } -// RemoveUserFromAllGroups mocks base method. -func (m *MockStore) RemoveUserFromAllGroups(ctx context.Context, userID uuid.UUID) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RemoveUserFromAllGroups", ctx, userID) - ret0, _ := ret[0].(error) - return ret0 -} - -// RemoveUserFromAllGroups indicates an expected call of RemoveUserFromAllGroups. -func (mr *MockStoreMockRecorder) RemoveUserFromAllGroups(ctx, userID any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveUserFromAllGroups", reflect.TypeOf((*MockStore)(nil).RemoveUserFromAllGroups), ctx, userID) -} - // RemoveUserFromGroups mocks base method. func (m *MockStore) RemoveUserFromGroups(ctx context.Context, arg database.RemoveUserFromGroupsParams) ([]uuid.UUID, error) { m.ctrl.T.Helper() @@ -7417,21 +7151,6 @@ func (mr *MockStoreMockRecorder) UpdateOAuth2ProviderAppByID(ctx, arg any) *gomo return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).UpdateOAuth2ProviderAppByID), ctx, arg) } -// UpdateOAuth2ProviderAppSecretByID mocks base method. -func (m *MockStore) UpdateOAuth2ProviderAppSecretByID(ctx context.Context, arg database.UpdateOAuth2ProviderAppSecretByIDParams) (database.OAuth2ProviderAppSecret, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateOAuth2ProviderAppSecretByID", ctx, arg) - ret0, _ := ret[0].(database.OAuth2ProviderAppSecret) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// UpdateOAuth2ProviderAppSecretByID indicates an expected call of UpdateOAuth2ProviderAppSecretByID. -func (mr *MockStoreMockRecorder) UpdateOAuth2ProviderAppSecretByID(ctx, arg any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOAuth2ProviderAppSecretByID", reflect.TypeOf((*MockStore)(nil).UpdateOAuth2ProviderAppSecretByID), ctx, arg) -} - // UpdateOrganization mocks base method. func (m *MockStore) UpdateOrganization(ctx context.Context, arg database.UpdateOrganizationParams) (database.Organization, error) { m.ctrl.T.Helper() @@ -7946,21 +7665,6 @@ func (mr *MockStoreMockRecorder) UpdateUserLink(ctx, arg any) *gomock.Call { return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLink", reflect.TypeOf((*MockStore)(nil).UpdateUserLink), ctx, arg) } -// UpdateUserLinkedID mocks base method. -func (m *MockStore) UpdateUserLinkedID(ctx context.Context, arg database.UpdateUserLinkedIDParams) (database.UserLink, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserLinkedID", ctx, arg) - ret0, _ := ret[0].(database.UserLink) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// UpdateUserLinkedID indicates an expected call of UpdateUserLinkedID. -func (mr *MockStoreMockRecorder) UpdateUserLinkedID(ctx, arg any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLinkedID", reflect.TypeOf((*MockStore)(nil).UpdateUserLinkedID), ctx, arg) -} - // UpdateUserLoginType mocks base method. func (m *MockStore) UpdateUserLoginType(ctx context.Context, arg database.UpdateUserLoginTypeParams) (database.User, error) { m.ctrl.T.Helper() @@ -8479,20 +8183,6 @@ func (mr *MockStoreMockRecorder) UpsertAnnouncementBanners(ctx, value any) *gomo return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertAnnouncementBanners", reflect.TypeOf((*MockStore)(nil).UpsertAnnouncementBanners), ctx, value) } -// UpsertAppSecurityKey mocks base method. -func (m *MockStore) UpsertAppSecurityKey(ctx context.Context, value string) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertAppSecurityKey", ctx, value) - ret0, _ := ret[0].(error) - return ret0 -} - -// UpsertAppSecurityKey indicates an expected call of UpsertAppSecurityKey. -func (mr *MockStoreMockRecorder) UpsertAppSecurityKey(ctx, value any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertAppSecurityKey", reflect.TypeOf((*MockStore)(nil).UpsertAppSecurityKey), ctx, value) -} - // UpsertApplicationName mocks base method. func (m *MockStore) UpsertApplicationName(ctx context.Context, value string) error { m.ctrl.T.Helper() @@ -8581,20 +8271,6 @@ func (mr *MockStoreMockRecorder) UpsertConnectionLog(ctx, arg any) *gomock.Call return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertConnectionLog", reflect.TypeOf((*MockStore)(nil).UpsertConnectionLog), ctx, arg) } -// UpsertCoordinatorResumeTokenSigningKey mocks base method. -func (m *MockStore) UpsertCoordinatorResumeTokenSigningKey(ctx context.Context, value string) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertCoordinatorResumeTokenSigningKey", ctx, value) - ret0, _ := ret[0].(error) - return ret0 -} - -// UpsertCoordinatorResumeTokenSigningKey indicates an expected call of UpsertCoordinatorResumeTokenSigningKey. -func (mr *MockStoreMockRecorder) UpsertCoordinatorResumeTokenSigningKey(ctx, value any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertCoordinatorResumeTokenSigningKey", reflect.TypeOf((*MockStore)(nil).UpsertCoordinatorResumeTokenSigningKey), ctx, value) -} - // UpsertDefaultProxy mocks base method. func (m *MockStore) UpsertDefaultProxy(ctx context.Context, arg database.UpsertDefaultProxyParams) error { m.ctrl.T.Helper() @@ -8693,20 +8369,6 @@ func (mr *MockStoreMockRecorder) UpsertOAuth2GithubDefaultEligible(ctx, eligible return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertOAuth2GithubDefaultEligible", reflect.TypeOf((*MockStore)(nil).UpsertOAuth2GithubDefaultEligible), ctx, eligible) } -// UpsertOAuthSigningKey mocks base method. -func (m *MockStore) UpsertOAuthSigningKey(ctx context.Context, value string) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertOAuthSigningKey", ctx, value) - ret0, _ := ret[0].(error) - return ret0 -} - -// UpsertOAuthSigningKey indicates an expected call of UpsertOAuthSigningKey. -func (mr *MockStoreMockRecorder) UpsertOAuthSigningKey(ctx, value any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertOAuthSigningKey", reflect.TypeOf((*MockStore)(nil).UpsertOAuthSigningKey), ctx, value) -} - // UpsertPrebuildsSettings mocks base method. func (m *MockStore) UpsertPrebuildsSettings(ctx context.Context, value string) error { m.ctrl.T.Helper() diff --git a/coderd/database/modelqueries.go b/coderd/database/modelqueries.go index 5f09cbc7cf..dbad0a1329 100644 --- a/coderd/database/modelqueries.go +++ b/coderd/database/modelqueries.go @@ -235,7 +235,6 @@ func (q *sqlQuerier) GetTemplateGroupRoles(ctx context.Context, id uuid.UUID) ([ type workspaceQuerier interface { GetAuthorizedWorkspaces(ctx context.Context, arg GetWorkspacesParams, prepared rbac.PreparedAuthorized) ([]GetWorkspacesRow, error) GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx context.Context, ownerID uuid.UUID, prepared rbac.PreparedAuthorized) ([]GetWorkspacesAndAgentsByOwnerIDRow, error) - GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIDs []uuid.UUID, prepared rbac.PreparedAuthorized) ([]WorkspaceBuildParameter, error) } // GetAuthorizedWorkspaces returns all workspaces that the user is authorized to access. @@ -391,35 +390,6 @@ func (q *sqlQuerier) GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx context.Conte return items, nil } -func (q *sqlQuerier) GetAuthorizedWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIDs []uuid.UUID, prepared rbac.PreparedAuthorized) ([]WorkspaceBuildParameter, error) { - authorizedFilter, err := prepared.CompileToSQL(ctx, rbac.ConfigWorkspaces()) - if err != nil { - return nil, xerrors.Errorf("compile authorized filter: %w", err) - } - - filtered, err := insertAuthorizedFilter(getWorkspaceBuildParametersByBuildIDs, fmt.Sprintf(" AND %s", authorizedFilter)) - if err != nil { - return nil, xerrors.Errorf("insert authorized filter: %w", err) - } - - query := fmt.Sprintf("-- name: GetAuthorizedWorkspaceBuildParametersByBuildIDs :many\n%s", filtered) - rows, err := q.db.QueryContext(ctx, query, pq.Array(workspaceBuildIDs)) - if err != nil { - return nil, err - } - defer rows.Close() - - var items []WorkspaceBuildParameter - for rows.Next() { - var i WorkspaceBuildParameter - if err := rows.Scan(&i.WorkspaceBuildID, &i.Name, &i.Value); err != nil { - return nil, err - } - items = append(items, i) - } - return items, nil -} - type userQuerier interface { GetAuthorizedUsers(ctx context.Context, arg GetUsersParams, prepared rbac.PreparedAuthorized) ([]GetUsersRow, error) } diff --git a/coderd/database/querier.go b/coderd/database/querier.go index 32c883b0a6..8ebd3c001d 100644 --- a/coderd/database/querier.go +++ b/coderd/database/querier.go @@ -96,7 +96,6 @@ type sqlcQuerier interface { DeleteAllWebpushSubscriptions(ctx context.Context) error DeleteApplicationConnectAPIKeysByUserID(ctx context.Context, userID uuid.UUID) error DeleteChatMessagesAfterID(ctx context.Context, arg DeleteChatMessagesAfterIDParams) error - DeleteChatMessagesByChatID(ctx context.Context, chatID uuid.UUID) error DeleteChatModelConfigByID(ctx context.Context, id uuid.UUID) error DeleteChatProviderByID(ctx context.Context, id uuid.UUID) error DeleteChatQueuedMessage(ctx context.Context, arg DeleteChatQueuedMessageParams) error @@ -104,7 +103,6 @@ type sqlcQuerier interface { DeleteCustomRole(ctx context.Context, arg DeleteCustomRoleParams) error DeleteExpiredAPIKeys(ctx context.Context, arg DeleteExpiredAPIKeysParams) (int64, error) DeleteExternalAuthLink(ctx context.Context, arg DeleteExternalAuthLinkParams) error - DeleteGitSSHKey(ctx context.Context, userID uuid.UUID) error DeleteGroupByID(ctx context.Context, id uuid.UUID) error DeleteGroupMemberFromGroup(ctx context.Context, arg DeleteGroupMemberFromGroupParams) error DeleteLicense(ctx context.Context, id int32) (int32, error) @@ -202,7 +200,6 @@ type sqlcQuerier interface { // rows we delete. Stale rows are excluded from the sum but still deleted. GetAndResetBoundaryUsageSummary(ctx context.Context, maxStalenessMs int64) (GetAndResetBoundaryUsageSummaryRow, error) GetAnnouncementBanners(ctx context.Context) (string, error) - GetAppSecurityKey(ctx context.Context) (string, error) GetApplicationName(ctx context.Context) (string, error) // GetAuditLogsBefore retrieves `row_limit` number of audit logs before the provided // ID. @@ -226,7 +223,6 @@ type sqlcQuerier interface { GetChatMessagesByChatID(ctx context.Context, arg GetChatMessagesByChatIDParams) ([]ChatMessage, error) GetChatMessagesForPromptByChatID(ctx context.Context, chatID uuid.UUID) ([]ChatMessage, error) GetChatModelConfigByID(ctx context.Context, id uuid.UUID) (ChatModelConfig, error) - GetChatModelConfigByProviderAndModel(ctx context.Context, arg GetChatModelConfigByProviderAndModelParams) (ChatModelConfig, error) GetChatModelConfigs(ctx context.Context) ([]ChatModelConfig, error) GetChatProviderByID(ctx context.Context, id uuid.UUID) (ChatProvider, error) GetChatProviderByProvider(ctx context.Context, provider string) (ChatProvider, error) @@ -235,7 +231,6 @@ type sqlcQuerier interface { GetChatSystemPrompt(ctx context.Context) (string, error) GetChatsByOwnerID(ctx context.Context, arg GetChatsByOwnerIDParams) ([]Chat, error) GetConnectionLogsOffset(ctx context.Context, arg GetConnectionLogsOffsetParams) ([]GetConnectionLogsOffsetRow, error) - GetCoordinatorResumeTokenSigningKey(ctx context.Context) (string, error) GetCryptoKeyByFeatureAndSequence(ctx context.Context, arg GetCryptoKeyByFeatureAndSequenceParams) (CryptoKey, error) GetCryptoKeys(ctx context.Context) ([]CryptoKey, error) GetCryptoKeysByFeature(ctx context.Context, feature CryptoKeyFeature) ([]CryptoKey, error) @@ -244,7 +239,6 @@ type sqlcQuerier interface { GetDefaultChatModelConfig(ctx context.Context) (ChatModelConfig, error) GetDefaultOrganization(ctx context.Context) (Organization, error) GetDefaultProxyConfig(ctx context.Context) (GetDefaultProxyConfigRow, error) - GetDeploymentDAUs(ctx context.Context, tzOffset int32) ([]GetDeploymentDAUsRow, error) GetDeploymentID(ctx context.Context) (string, error) GetDeploymentWorkspaceAgentStats(ctx context.Context, createdAt time.Time) (GetDeploymentWorkspaceAgentStatsRow, error) GetDeploymentWorkspaceAgentUsageStats(ctx context.Context, createdAt time.Time) (GetDeploymentWorkspaceAgentUsageStatsRow, error) @@ -257,7 +251,6 @@ type sqlcQuerier interface { GetFailedWorkspaceBuildsByTemplateID(ctx context.Context, arg GetFailedWorkspaceBuildsByTemplateIDParams) ([]GetFailedWorkspaceBuildsByTemplateIDRow, error) GetFileByHashAndCreator(ctx context.Context, arg GetFileByHashAndCreatorParams) (File, error) GetFileByID(ctx context.Context, id uuid.UUID) (File, error) - GetFileIDByTemplateVersionID(ctx context.Context, templateVersionID uuid.UUID) (uuid.UUID, error) // Get all templates that use a file. GetFileTemplates(ctx context.Context, fileID uuid.UUID) ([]GetFileTemplatesRow, error) // Fetches inbox notifications for a user filtered by templates and targets @@ -306,7 +299,6 @@ type sqlcQuerier interface { // RFC 7591/7592 Dynamic Client Registration queries GetOAuth2ProviderAppByClientID(ctx context.Context, id uuid.UUID) (OAuth2ProviderApp, error) GetOAuth2ProviderAppByID(ctx context.Context, id uuid.UUID) (OAuth2ProviderApp, error) - GetOAuth2ProviderAppByRegistrationToken(ctx context.Context, registrationAccessToken []byte) (OAuth2ProviderApp, error) GetOAuth2ProviderAppCodeByID(ctx context.Context, id uuid.UUID) (OAuth2ProviderAppCode, error) GetOAuth2ProviderAppCodeByPrefix(ctx context.Context, secretPrefix []byte) (OAuth2ProviderAppCode, error) GetOAuth2ProviderAppSecretByID(ctx context.Context, id uuid.UUID) (OAuth2ProviderAppSecret, error) @@ -316,7 +308,6 @@ type sqlcQuerier interface { GetOAuth2ProviderAppTokenByPrefix(ctx context.Context, hashPrefix []byte) (OAuth2ProviderAppToken, error) GetOAuth2ProviderApps(ctx context.Context) ([]OAuth2ProviderApp, error) GetOAuth2ProviderAppsByUserID(ctx context.Context, userID uuid.UUID) ([]GetOAuth2ProviderAppsByUserIDRow, error) - GetOAuthSigningKey(ctx context.Context) (string, error) GetOrganizationByID(ctx context.Context, id uuid.UUID) (Organization, error) GetOrganizationByName(ctx context.Context, arg GetOrganizationByNameParams) (Organization, error) GetOrganizationIDsByMemberIDs(ctx context.Context, ids []uuid.UUID) ([]GetOrganizationIDsByMemberIDsRow, error) @@ -372,7 +363,6 @@ type sqlcQuerier interface { // Blocks until the row is available for update. GetProvisionerJobByIDWithLock(ctx context.Context, id uuid.UUID) (ProvisionerJob, error) GetProvisionerJobTimingsByJobID(ctx context.Context, jobID uuid.UUID) ([]ProvisionerJobTiming, error) - GetProvisionerJobsByIDs(ctx context.Context, ids []uuid.UUID) ([]ProvisionerJob, error) GetProvisionerJobsByIDsWithQueuePosition(ctx context.Context, arg GetProvisionerJobsByIDsWithQueuePositionParams) ([]GetProvisionerJobsByIDsWithQueuePositionRow, error) GetProvisionerJobsByOrganizationAndStatusWithQueuePositionAndProvisioner(ctx context.Context, arg GetProvisionerJobsByOrganizationAndStatusWithQueuePositionAndProvisionerParams) ([]GetProvisionerJobsByOrganizationAndStatusWithQueuePositionAndProvisionerRow, error) GetProvisionerJobsCreatedAfter(ctx context.Context, createdAt time.Time) ([]ProvisionerJob, error) @@ -430,7 +420,6 @@ type sqlcQuerier interface { GetTemplateAverageBuildTime(ctx context.Context, templateID uuid.NullUUID) (GetTemplateAverageBuildTimeRow, error) GetTemplateByID(ctx context.Context, id uuid.UUID) (Template, error) GetTemplateByOrganizationAndName(ctx context.Context, arg GetTemplateByOrganizationAndNameParams) (Template, error) - GetTemplateDAUs(ctx context.Context, arg GetTemplateDAUsParams) ([]GetTemplateDAUsRow, error) // GetTemplateInsights returns the aggregate user-produced usage of all // workspaces in a given timeframe. The template IDs, active users, and // usage_seconds all reflect any usage in the template, including apps. @@ -462,7 +451,6 @@ type sqlcQuerier interface { GetTemplateVersionByID(ctx context.Context, id uuid.UUID) (TemplateVersion, error) GetTemplateVersionByJobID(ctx context.Context, jobID uuid.UUID) (TemplateVersion, error) GetTemplateVersionByTemplateIDAndName(ctx context.Context, arg GetTemplateVersionByTemplateIDAndNameParams) (TemplateVersion, error) - GetTemplateVersionHasAITask(ctx context.Context, id uuid.UUID) (bool, error) GetTemplateVersionParameters(ctx context.Context, templateVersionID uuid.UUID) ([]TemplateVersionParameter, error) GetTemplateVersionTerraformValues(ctx context.Context, templateVersionID uuid.UUID) (TemplateVersionTerraformValue, error) GetTemplateVersionVariables(ctx context.Context, templateVersionID uuid.UUID) ([]TemplateVersionVariable, error) @@ -555,7 +543,6 @@ type sqlcQuerier interface { // Also checks if all agents are ready and returns the worst status. GetWorkspaceBuildMetricsByResourceID(ctx context.Context, id uuid.UUID) (GetWorkspaceBuildMetricsByResourceIDRow, error) GetWorkspaceBuildParameters(ctx context.Context, workspaceBuildID uuid.UUID) ([]WorkspaceBuildParameter, error) - GetWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIds []uuid.UUID) ([]WorkspaceBuildParameter, error) // Fetches the provisioner state of a workspace build, joined through to the // template so that dbauthz can enforce policy.ActionUpdate on the template. // Provisioner state contains sensitive Terraform state and should only be @@ -666,8 +653,6 @@ type sqlcQuerier interface { // InsertUserGroupsByID adds a user to all provided groups, if they exist. // If there is a conflict, the user is already a member InsertUserGroupsByID(ctx context.Context, arg InsertUserGroupsByIDParams) ([]uuid.UUID, error) - // InsertUserGroupsByName adds a user to all provided groups, if they exist. - InsertUserGroupsByName(ctx context.Context, arg InsertUserGroupsByNameParams) error InsertUserLink(ctx context.Context, arg InsertUserLinkParams) (UserLink, error) InsertVolumeResourceMonitor(ctx context.Context, arg InsertVolumeResourceMonitorParams) (WorkspaceAgentVolumeResourceMonitor, error) InsertWebpushSubscription(ctx context.Context, arg InsertWebpushSubscriptionParams) (WebpushSubscription, error) @@ -696,8 +681,6 @@ type sqlcQuerier interface { ListAIBridgeTokenUsagesByInterceptionIDs(ctx context.Context, interceptionIds []uuid.UUID) ([]AIBridgeTokenUsage, error) ListAIBridgeToolUsagesByInterceptionIDs(ctx context.Context, interceptionIds []uuid.UUID) ([]AIBridgeToolUsage, error) ListAIBridgeUserPromptsByInterceptionIDs(ctx context.Context, interceptionIds []uuid.UUID) ([]AIBridgeUserPrompt, error) - ListChatsByRootID(ctx context.Context, rootChatID uuid.UUID) ([]Chat, error) - ListChildChatsByParentID(ctx context.Context, parentChatID uuid.UUID) ([]Chat, error) ListProvisionerKeysByOrganization(ctx context.Context, organizationID uuid.UUID) ([]ProvisionerKey, error) ListProvisionerKeysByOrganizationExcludeReserved(ctx context.Context, organizationID uuid.UUID) ([]ProvisionerKey, error) ListTasks(ctx context.Context, arg ListTasksParams) ([]Task, error) @@ -717,7 +700,6 @@ type sqlcQuerier interface { PopNextQueuedMessage(ctx context.Context, chatID uuid.UUID) (ChatQueuedMessage, error) ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate(ctx context.Context, templateID uuid.UUID) error RegisterWorkspaceProxy(ctx context.Context, arg RegisterWorkspaceProxyParams) (WorkspaceProxy, error) - RemoveUserFromAllGroups(ctx context.Context, userID uuid.UUID) error RemoveUserFromGroups(ctx context.Context, arg RemoveUserFromGroupsParams) ([]uuid.UUID, error) RevokeDBCryptKey(ctx context.Context, activeKeyDigest string) error // Note that this selects from the CTE, not the original table. The CTE is named @@ -763,7 +745,6 @@ type sqlcQuerier interface { UpdateNotificationTemplateMethodByID(ctx context.Context, arg UpdateNotificationTemplateMethodByIDParams) (NotificationTemplate, error) UpdateOAuth2ProviderAppByClientID(ctx context.Context, arg UpdateOAuth2ProviderAppByClientIDParams) (OAuth2ProviderApp, error) UpdateOAuth2ProviderAppByID(ctx context.Context, arg UpdateOAuth2ProviderAppByIDParams) (OAuth2ProviderApp, error) - UpdateOAuth2ProviderAppSecretByID(ctx context.Context, arg UpdateOAuth2ProviderAppSecretByIDParams) (OAuth2ProviderAppSecret, error) UpdateOrganization(ctx context.Context, arg UpdateOrganizationParams) (Organization, error) UpdateOrganizationDeletedByID(ctx context.Context, arg UpdateOrganizationDeletedByIDParams) error UpdateOrganizationWorkspaceSharingSettings(ctx context.Context, arg UpdateOrganizationWorkspaceSharingSettingsParams) (Organization, error) @@ -803,7 +784,6 @@ type sqlcQuerier interface { UpdateUserHashedPassword(ctx context.Context, arg UpdateUserHashedPasswordParams) error UpdateUserLastSeenAt(ctx context.Context, arg UpdateUserLastSeenAtParams) (User, error) UpdateUserLink(ctx context.Context, arg UpdateUserLinkParams) (UserLink, error) - UpdateUserLinkedID(ctx context.Context, arg UpdateUserLinkedIDParams) (UserLink, error) UpdateUserLoginType(ctx context.Context, arg UpdateUserLoginTypeParams) (User, error) UpdateUserNotificationPreferences(ctx context.Context, arg UpdateUserNotificationPreferencesParams) (int64, error) UpdateUserProfile(ctx context.Context, arg UpdateUserProfileParams) (User, error) @@ -841,7 +821,6 @@ type sqlcQuerier interface { UpdateWorkspacesDormantDeletingAtByTemplateID(ctx context.Context, arg UpdateWorkspacesDormantDeletingAtByTemplateIDParams) ([]WorkspaceTable, error) UpdateWorkspacesTTLByTemplateID(ctx context.Context, arg UpdateWorkspacesTTLByTemplateIDParams) error UpsertAnnouncementBanners(ctx context.Context, value string) error - UpsertAppSecurityKey(ctx context.Context, value string) error UpsertApplicationName(ctx context.Context, value string) error // Upserts boundary usage statistics for a replica. On INSERT (new period), uses // delta values for unique counts (only data since last flush). On UPDATE, uses @@ -852,7 +831,6 @@ type sqlcQuerier interface { UpsertChatDiffStatusReference(ctx context.Context, arg UpsertChatDiffStatusReferenceParams) (ChatDiffStatus, error) UpsertChatSystemPrompt(ctx context.Context, value string) error UpsertConnectionLog(ctx context.Context, arg UpsertConnectionLogParams) (ConnectionLog, error) - UpsertCoordinatorResumeTokenSigningKey(ctx context.Context, value string) error // The default proxy is implied and not actually stored in the database. // So we need to store it's configuration here for display purposes. // The functional values are immutable and controlled implicitly. @@ -864,7 +842,6 @@ type sqlcQuerier interface { UpsertNotificationReportGeneratorLog(ctx context.Context, arg UpsertNotificationReportGeneratorLogParams) error UpsertNotificationsSettings(ctx context.Context, value string) error UpsertOAuth2GithubDefaultEligible(ctx context.Context, eligible bool) error - UpsertOAuthSigningKey(ctx context.Context, value string) error UpsertPrebuildsSettings(ctx context.Context, value string) error UpsertProvisionerDaemon(ctx context.Context, arg UpsertProvisionerDaemonParams) (ProvisionerDaemon, error) UpsertRuntimeConfig(ctx context.Context, arg UpsertRuntimeConfigParams) error diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go index 874e61ff5d..5b84ad764d 100644 --- a/coderd/database/queries.sql.go +++ b/coderd/database/queries.sql.go @@ -2360,50 +2360,6 @@ func (q *sqlQuerier) GetChatModelConfigByID(ctx context.Context, id uuid.UUID) ( return i, err } -const getChatModelConfigByProviderAndModel = `-- name: GetChatModelConfigByProviderAndModel :one -SELECT - id, provider, model, display_name, created_by, updated_by, enabled, is_default, deleted, deleted_at, created_at, updated_at, context_limit, compression_threshold, options -FROM - chat_model_configs -WHERE - provider = $1::text - AND model = $2::text - AND deleted = FALSE -ORDER BY - updated_at DESC, - created_at DESC, - id DESC -LIMIT 1 -` - -type GetChatModelConfigByProviderAndModelParams struct { - Provider string `db:"provider" json:"provider"` - Model string `db:"model" json:"model"` -} - -func (q *sqlQuerier) GetChatModelConfigByProviderAndModel(ctx context.Context, arg GetChatModelConfigByProviderAndModelParams) (ChatModelConfig, error) { - row := q.db.QueryRowContext(ctx, getChatModelConfigByProviderAndModel, arg.Provider, arg.Model) - var i ChatModelConfig - err := row.Scan( - &i.ID, - &i.Provider, - &i.Model, - &i.DisplayName, - &i.CreatedBy, - &i.UpdatedBy, - &i.Enabled, - &i.IsDefault, - &i.Deleted, - &i.DeletedAt, - &i.CreatedAt, - &i.UpdatedAt, - &i.ContextLimit, - &i.CompressionThreshold, - &i.Options, - ) - return i, err -} - const getChatModelConfigs = `-- name: GetChatModelConfigs :many SELECT id, provider, model, display_name, created_by, updated_by, enabled, is_default, deleted, deleted_at, created_at, updated_at, context_limit, compression_threshold, options @@ -3200,18 +3156,6 @@ func (q *sqlQuerier) DeleteChatMessagesAfterID(ctx context.Context, arg DeleteCh return err } -const deleteChatMessagesByChatID = `-- name: DeleteChatMessagesByChatID :exec -DELETE FROM - chat_messages -WHERE - chat_id = $1::uuid -` - -func (q *sqlQuerier) DeleteChatMessagesByChatID(ctx context.Context, chatID uuid.UUID) error { - _, err := q.db.ExecContext(ctx, deleteChatMessagesByChatID, chatID) - return err -} - const deleteChatQueuedMessage = `-- name: DeleteChatQueuedMessage :exec DELETE FROM chat_queued_messages WHERE id = $1 AND chat_id = $2 ` @@ -3957,106 +3901,6 @@ func (q *sqlQuerier) InsertChatQueuedMessage(ctx context.Context, arg InsertChat return i, err } -const listChatsByRootID = `-- name: ListChatsByRootID :many -SELECT - id, owner_id, workspace_id, title, status, worker_id, started_at, heartbeat_at, created_at, updated_at, parent_chat_id, root_chat_id, last_model_config_id, archived, last_error -FROM - chats -WHERE - root_chat_id = $1::uuid -ORDER BY - created_at ASC -` - -func (q *sqlQuerier) ListChatsByRootID(ctx context.Context, rootChatID uuid.UUID) ([]Chat, error) { - rows, err := q.db.QueryContext(ctx, listChatsByRootID, rootChatID) - if err != nil { - return nil, err - } - defer rows.Close() - var items []Chat - for rows.Next() { - var i Chat - if err := rows.Scan( - &i.ID, - &i.OwnerID, - &i.WorkspaceID, - &i.Title, - &i.Status, - &i.WorkerID, - &i.StartedAt, - &i.HeartbeatAt, - &i.CreatedAt, - &i.UpdatedAt, - &i.ParentChatID, - &i.RootChatID, - &i.LastModelConfigID, - &i.Archived, - &i.LastError, - ); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - -const listChildChatsByParentID = `-- name: ListChildChatsByParentID :many -SELECT - id, owner_id, workspace_id, title, status, worker_id, started_at, heartbeat_at, created_at, updated_at, parent_chat_id, root_chat_id, last_model_config_id, archived, last_error -FROM - chats -WHERE - parent_chat_id = $1::uuid -ORDER BY - created_at ASC -` - -func (q *sqlQuerier) ListChildChatsByParentID(ctx context.Context, parentChatID uuid.UUID) ([]Chat, error) { - rows, err := q.db.QueryContext(ctx, listChildChatsByParentID, parentChatID) - if err != nil { - return nil, err - } - defer rows.Close() - var items []Chat - for rows.Next() { - var i Chat - if err := rows.Scan( - &i.ID, - &i.OwnerID, - &i.WorkspaceID, - &i.Title, - &i.Status, - &i.WorkerID, - &i.StartedAt, - &i.HeartbeatAt, - &i.CreatedAt, - &i.UpdatedAt, - &i.ParentChatID, - &i.RootChatID, - &i.LastModelConfigID, - &i.Archived, - &i.LastError, - ); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - const popNextQueuedMessage = `-- name: PopNextQueuedMessage :one DELETE FROM chat_queued_messages WHERE id = ( @@ -5581,30 +5425,6 @@ func (q *sqlQuerier) GetFileByID(ctx context.Context, id uuid.UUID) (File, error return i, err } -const getFileIDByTemplateVersionID = `-- name: GetFileIDByTemplateVersionID :one -SELECT - files.id -FROM - files -JOIN - provisioner_jobs ON - provisioner_jobs.storage_method = 'file' - AND provisioner_jobs.file_id = files.id -JOIN - template_versions ON template_versions.job_id = provisioner_jobs.id -WHERE - template_versions.id = $1 -LIMIT - 1 -` - -func (q *sqlQuerier) GetFileIDByTemplateVersionID(ctx context.Context, templateVersionID uuid.UUID) (uuid.UUID, error) { - row := q.db.QueryRowContext(ctx, getFileIDByTemplateVersionID, templateVersionID) - var id uuid.UUID - err := row.Scan(&id) - return id, err -} - const getFileTemplates = `-- name: GetFileTemplates :many SELECT files.id AS file_id, @@ -5711,18 +5531,6 @@ func (q *sqlQuerier) InsertFile(ctx context.Context, arg InsertFileParams) (File return i, err } -const deleteGitSSHKey = `-- name: DeleteGitSSHKey :exec -DELETE FROM - gitsshkeys -WHERE - user_id = $1 -` - -func (q *sqlQuerier) DeleteGitSSHKey(ctx context.Context, userID uuid.UUID) error { - _, err := q.db.ExecContext(ctx, deleteGitSSHKey, userID) - return err -} - const getGitSSHKey = `-- name: GetGitSSHKey :one SELECT user_id, created_at, updated_at, private_key, public_key @@ -6047,49 +5855,6 @@ func (q *sqlQuerier) InsertUserGroupsByID(ctx context.Context, arg InsertUserGro return items, nil } -const insertUserGroupsByName = `-- name: InsertUserGroupsByName :exec -WITH groups AS ( - SELECT - id - FROM - groups - WHERE - groups.organization_id = $2 AND - groups.name = ANY($3 :: text []) -) -INSERT INTO - group_members (user_id, group_id) -SELECT - $1, - groups.id -FROM - groups -` - -type InsertUserGroupsByNameParams struct { - UserID uuid.UUID `db:"user_id" json:"user_id"` - OrganizationID uuid.UUID `db:"organization_id" json:"organization_id"` - GroupNames []string `db:"group_names" json:"group_names"` -} - -// InsertUserGroupsByName adds a user to all provided groups, if they exist. -func (q *sqlQuerier) InsertUserGroupsByName(ctx context.Context, arg InsertUserGroupsByNameParams) error { - _, err := q.db.ExecContext(ctx, insertUserGroupsByName, arg.UserID, arg.OrganizationID, pq.Array(arg.GroupNames)) - return err -} - -const removeUserFromAllGroups = `-- name: RemoveUserFromAllGroups :exec -DELETE FROM - group_members -WHERE - user_id = $1 -` - -func (q *sqlQuerier) RemoveUserFromAllGroups(ctx context.Context, userID uuid.UUID) error { - _, err := q.db.ExecContext(ctx, removeUserFromAllGroups, userID) - return err -} - const removeUserFromGroups = `-- name: RemoveUserFromGroups :many DELETE FROM group_members @@ -9074,44 +8839,6 @@ func (q *sqlQuerier) GetOAuth2ProviderAppByID(ctx context.Context, id uuid.UUID) return i, err } -const getOAuth2ProviderAppByRegistrationToken = `-- name: GetOAuth2ProviderAppByRegistrationToken :one -SELECT id, created_at, updated_at, name, icon, callback_url, redirect_uris, client_type, dynamically_registered, client_id_issued_at, client_secret_expires_at, grant_types, response_types, token_endpoint_auth_method, scope, contacts, client_uri, logo_uri, tos_uri, policy_uri, jwks_uri, jwks, software_id, software_version, registration_access_token, registration_client_uri FROM oauth2_provider_apps WHERE registration_access_token = $1 -` - -func (q *sqlQuerier) GetOAuth2ProviderAppByRegistrationToken(ctx context.Context, registrationAccessToken []byte) (OAuth2ProviderApp, error) { - row := q.db.QueryRowContext(ctx, getOAuth2ProviderAppByRegistrationToken, registrationAccessToken) - var i OAuth2ProviderApp - err := row.Scan( - &i.ID, - &i.CreatedAt, - &i.UpdatedAt, - &i.Name, - &i.Icon, - &i.CallbackURL, - pq.Array(&i.RedirectUris), - &i.ClientType, - &i.DynamicallyRegistered, - &i.ClientIDIssuedAt, - &i.ClientSecretExpiresAt, - pq.Array(&i.GrantTypes), - pq.Array(&i.ResponseTypes), - &i.TokenEndpointAuthMethod, - &i.Scope, - pq.Array(&i.Contacts), - &i.ClientUri, - &i.LogoUri, - &i.TosUri, - &i.PolicyUri, - &i.JwksUri, - &i.Jwks, - &i.SoftwareID, - &i.SoftwareVersion, - &i.RegistrationAccessToken, - &i.RegistrationClientUri, - ) - return i, err -} - const getOAuth2ProviderAppCodeByID = `-- name: GetOAuth2ProviderAppCodeByID :one SELECT id, created_at, expires_at, secret_prefix, hashed_secret, user_id, app_id, resource_uri, code_challenge, code_challenge_method, state_hash, redirect_uri FROM oauth2_provider_app_codes WHERE id = $1 ` @@ -9951,32 +9678,6 @@ func (q *sqlQuerier) UpdateOAuth2ProviderAppByID(ctx context.Context, arg Update return i, err } -const updateOAuth2ProviderAppSecretByID = `-- name: UpdateOAuth2ProviderAppSecretByID :one -UPDATE oauth2_provider_app_secrets SET - last_used_at = $2 -WHERE id = $1 RETURNING id, created_at, last_used_at, hashed_secret, display_secret, app_id, secret_prefix -` - -type UpdateOAuth2ProviderAppSecretByIDParams struct { - ID uuid.UUID `db:"id" json:"id"` - LastUsedAt sql.NullTime `db:"last_used_at" json:"last_used_at"` -} - -func (q *sqlQuerier) UpdateOAuth2ProviderAppSecretByID(ctx context.Context, arg UpdateOAuth2ProviderAppSecretByIDParams) (OAuth2ProviderAppSecret, error) { - row := q.db.QueryRowContext(ctx, updateOAuth2ProviderAppSecretByID, arg.ID, arg.LastUsedAt) - var i OAuth2ProviderAppSecret - err := row.Scan( - &i.ID, - &i.CreatedAt, - &i.LastUsedAt, - &i.HashedSecret, - &i.DisplaySecret, - &i.AppID, - &i.SecretPrefix, - ) - return i, err -} - const deleteOrganizationMember = `-- name: DeleteOrganizationMember :exec DELETE FROM @@ -12845,60 +12546,6 @@ func (q *sqlQuerier) GetProvisionerJobTimingsByJobID(ctx context.Context, jobID return items, nil } -const getProvisionerJobsByIDs = `-- name: GetProvisionerJobsByIDs :many -SELECT - id, created_at, updated_at, started_at, canceled_at, completed_at, error, organization_id, initiator_id, provisioner, storage_method, type, input, worker_id, file_id, tags, error_code, trace_metadata, job_status, logs_length, logs_overflowed -FROM - provisioner_jobs -WHERE - id = ANY($1 :: uuid [ ]) -` - -func (q *sqlQuerier) GetProvisionerJobsByIDs(ctx context.Context, ids []uuid.UUID) ([]ProvisionerJob, error) { - rows, err := q.db.QueryContext(ctx, getProvisionerJobsByIDs, pq.Array(ids)) - if err != nil { - return nil, err - } - defer rows.Close() - var items []ProvisionerJob - for rows.Next() { - var i ProvisionerJob - if err := rows.Scan( - &i.ID, - &i.CreatedAt, - &i.UpdatedAt, - &i.StartedAt, - &i.CanceledAt, - &i.CompletedAt, - &i.Error, - &i.OrganizationID, - &i.InitiatorID, - &i.Provisioner, - &i.StorageMethod, - &i.Type, - &i.Input, - &i.WorkerID, - &i.FileID, - &i.Tags, - &i.ErrorCode, - &i.TraceMetadata, - &i.JobStatus, - &i.LogsLength, - &i.LogsOverflowed, - ); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - const getProvisionerJobsByIDsWithQueuePosition = `-- name: GetProvisionerJobsByIDsWithQueuePosition :many WITH filtered_provisioner_jobs AS ( -- Step 1: Filter provisioner_jobs @@ -14702,17 +14349,6 @@ func (q *sqlQuerier) GetAnnouncementBanners(ctx context.Context) (string, error) return value, err } -const getAppSecurityKey = `-- name: GetAppSecurityKey :one -SELECT value FROM site_configs WHERE key = 'app_signing_key' -` - -func (q *sqlQuerier) GetAppSecurityKey(ctx context.Context) (string, error) { - row := q.db.QueryRowContext(ctx, getAppSecurityKey) - var value string - err := row.Scan(&value) - return value, err -} - const getApplicationName = `-- name: GetApplicationName :one SELECT value FROM site_configs WHERE key = 'application_name' ` @@ -14736,17 +14372,6 @@ func (q *sqlQuerier) GetChatSystemPrompt(ctx context.Context) (string, error) { return chat_system_prompt, err } -const getCoordinatorResumeTokenSigningKey = `-- name: GetCoordinatorResumeTokenSigningKey :one -SELECT value FROM site_configs WHERE key = 'coordinator_resume_token_signing_key' -` - -func (q *sqlQuerier) GetCoordinatorResumeTokenSigningKey(ctx context.Context) (string, error) { - row := q.db.QueryRowContext(ctx, getCoordinatorResumeTokenSigningKey) - var value string - err := row.Scan(&value) - return value, err -} - const getDERPMeshKey = `-- name: GetDERPMeshKey :one SELECT value FROM site_configs WHERE key = 'derp_mesh_key' ` @@ -14850,17 +14475,6 @@ func (q *sqlQuerier) GetOAuth2GithubDefaultEligible(ctx context.Context) (bool, return column_1, err } -const getOAuthSigningKey = `-- name: GetOAuthSigningKey :one -SELECT value FROM site_configs WHERE key = 'oauth_signing_key' -` - -func (q *sqlQuerier) GetOAuthSigningKey(ctx context.Context) (string, error) { - row := q.db.QueryRowContext(ctx, getOAuthSigningKey) - var value string - err := row.Scan(&value) - return value, err -} - const getPrebuildsSettings = `-- name: GetPrebuildsSettings :one SELECT COALESCE((SELECT value FROM site_configs WHERE key = 'prebuilds_settings'), '{}') :: text AS prebuilds_settings @@ -14930,16 +14544,6 @@ func (q *sqlQuerier) UpsertAnnouncementBanners(ctx context.Context, value string return err } -const upsertAppSecurityKey = `-- name: UpsertAppSecurityKey :exec -INSERT INTO site_configs (key, value) VALUES ('app_signing_key', $1) -ON CONFLICT (key) DO UPDATE set value = $1 WHERE site_configs.key = 'app_signing_key' -` - -func (q *sqlQuerier) UpsertAppSecurityKey(ctx context.Context, value string) error { - _, err := q.db.ExecContext(ctx, upsertAppSecurityKey, value) - return err -} - const upsertApplicationName = `-- name: UpsertApplicationName :exec INSERT INTO site_configs (key, value) VALUES ('application_name', $1) ON CONFLICT (key) DO UPDATE SET value = $1 WHERE site_configs.key = 'application_name' @@ -14960,16 +14564,6 @@ func (q *sqlQuerier) UpsertChatSystemPrompt(ctx context.Context, value string) e return err } -const upsertCoordinatorResumeTokenSigningKey = `-- name: UpsertCoordinatorResumeTokenSigningKey :exec -INSERT INTO site_configs (key, value) VALUES ('coordinator_resume_token_signing_key', $1) -ON CONFLICT (key) DO UPDATE set value = $1 WHERE site_configs.key = 'coordinator_resume_token_signing_key' -` - -func (q *sqlQuerier) UpsertCoordinatorResumeTokenSigningKey(ctx context.Context, value string) error { - _, err := q.db.ExecContext(ctx, upsertCoordinatorResumeTokenSigningKey, value) - return err -} - const upsertDefaultProxy = `-- name: UpsertDefaultProxy :exec INSERT INTO site_configs (key, value) VALUES @@ -15055,16 +14649,6 @@ func (q *sqlQuerier) UpsertOAuth2GithubDefaultEligible(ctx context.Context, elig return err } -const upsertOAuthSigningKey = `-- name: UpsertOAuthSigningKey :exec -INSERT INTO site_configs (key, value) VALUES ('oauth_signing_key', $1) -ON CONFLICT (key) DO UPDATE set value = $1 WHERE site_configs.key = 'oauth_signing_key' -` - -func (q *sqlQuerier) UpsertOAuthSigningKey(ctx context.Context, value string) error { - _, err := q.db.ExecContext(ctx, upsertOAuthSigningKey, value) - return err -} - const upsertPrebuildsSettings = `-- name: UpsertPrebuildsSettings :exec INSERT INTO site_configs (key, value) VALUES ('prebuilds_settings', $1) ON CONFLICT (key) DO UPDATE SET value = $1 WHERE site_configs.key = 'prebuilds_settings' @@ -17383,21 +16967,6 @@ func (q *sqlQuerier) GetTemplateVersionByTemplateIDAndName(ctx context.Context, return i, err } -const getTemplateVersionHasAITask = `-- name: GetTemplateVersionHasAITask :one -SELECT EXISTS ( - SELECT 1 - FROM template_versions - WHERE id = $1 AND has_ai_task = TRUE -) -` - -func (q *sqlQuerier) GetTemplateVersionHasAITask(ctx context.Context, id uuid.UUID) (bool, error) { - row := q.db.QueryRowContext(ctx, getTemplateVersionHasAITask, id) - var exists bool - err := row.Scan(&exists) - return exists, err -} - const getTemplateVersionsByIDs = `-- name: GetTemplateVersionsByIDs :many SELECT id, template_id, organization_id, created_at, updated_at, name, readme, job_id, created_by, external_auth_providers, message, archived, source_example_id, has_ai_task, has_external_agent, created_by_avatar_url, created_by_username, created_by_name @@ -18502,38 +18071,6 @@ func (q *sqlQuerier) UpdateUserLink(ctx context.Context, arg UpdateUserLinkParam return i, err } -const updateUserLinkedID = `-- name: UpdateUserLinkedID :one -UPDATE - user_links -SET - linked_id = $1 -WHERE - user_id = $2 AND login_type = $3 RETURNING user_id, login_type, linked_id, oauth_access_token, oauth_refresh_token, oauth_expiry, oauth_access_token_key_id, oauth_refresh_token_key_id, claims -` - -type UpdateUserLinkedIDParams struct { - LinkedID string `db:"linked_id" json:"linked_id"` - UserID uuid.UUID `db:"user_id" json:"user_id"` - LoginType LoginType `db:"login_type" json:"login_type"` -} - -func (q *sqlQuerier) UpdateUserLinkedID(ctx context.Context, arg UpdateUserLinkedIDParams) (UserLink, error) { - row := q.db.QueryRowContext(ctx, updateUserLinkedID, arg.LinkedID, arg.UserID, arg.LoginType) - var i UserLink - err := row.Scan( - &i.UserID, - &i.LoginType, - &i.LinkedID, - &i.OAuthAccessToken, - &i.OAuthRefreshToken, - &i.OAuthExpiry, - &i.OAuthAccessTokenKeyID, - &i.OAuthRefreshTokenKeyID, - &i.Claims, - ) - return i, err -} - const createUserSecret = `-- name: CreateUserSecret :one INSERT INTO user_secrets ( id, @@ -22150,48 +21687,6 @@ func (q *sqlQuerier) DeleteOldWorkspaceAgentStats(ctx context.Context) error { return err } -const getDeploymentDAUs = `-- name: GetDeploymentDAUs :many -SELECT - (created_at at TIME ZONE cast($1::integer as text))::date as date, - user_id -FROM - workspace_agent_stats -WHERE - connection_count > 0 -GROUP BY - date, user_id -ORDER BY - date ASC -` - -type GetDeploymentDAUsRow struct { - Date time.Time `db:"date" json:"date"` - UserID uuid.UUID `db:"user_id" json:"user_id"` -} - -func (q *sqlQuerier) GetDeploymentDAUs(ctx context.Context, tzOffset int32) ([]GetDeploymentDAUsRow, error) { - rows, err := q.db.QueryContext(ctx, getDeploymentDAUs, tzOffset) - if err != nil { - return nil, err - } - defer rows.Close() - var items []GetDeploymentDAUsRow - for rows.Next() { - var i GetDeploymentDAUsRow - if err := rows.Scan(&i.Date, &i.UserID); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - const getDeploymentWorkspaceAgentStats = `-- name: GetDeploymentWorkspaceAgentStats :one WITH stats AS ( SELECT @@ -22330,54 +21825,6 @@ func (q *sqlQuerier) GetDeploymentWorkspaceAgentUsageStats(ctx context.Context, return i, err } -const getTemplateDAUs = `-- name: GetTemplateDAUs :many -SELECT - (created_at at TIME ZONE cast($2::integer as text))::date as date, - user_id -FROM - workspace_agent_stats -WHERE - template_id = $1 AND - connection_count > 0 -GROUP BY - date, user_id -ORDER BY - date ASC -` - -type GetTemplateDAUsParams struct { - TemplateID uuid.UUID `db:"template_id" json:"template_id"` - TzOffset int32 `db:"tz_offset" json:"tz_offset"` -} - -type GetTemplateDAUsRow struct { - Date time.Time `db:"date" json:"date"` - UserID uuid.UUID `db:"user_id" json:"user_id"` -} - -func (q *sqlQuerier) GetTemplateDAUs(ctx context.Context, arg GetTemplateDAUsParams) ([]GetTemplateDAUsRow, error) { - rows, err := q.db.QueryContext(ctx, getTemplateDAUs, arg.TemplateID, arg.TzOffset) - if err != nil { - return nil, err - } - defer rows.Close() - var items []GetTemplateDAUsRow - for rows.Next() { - var i GetTemplateDAUsRow - if err := rows.Scan(&i.Date, &i.UserID); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - const getWorkspaceAgentStats = `-- name: GetWorkspaceAgentStats :many WITH agent_stats AS ( SELECT @@ -23589,44 +23036,6 @@ func (q *sqlQuerier) GetWorkspaceBuildParameters(ctx context.Context, workspaceB return items, nil } -const getWorkspaceBuildParametersByBuildIDs = `-- name: GetWorkspaceBuildParametersByBuildIDs :many -SELECT - workspace_build_parameters.workspace_build_id, workspace_build_parameters.name, workspace_build_parameters.value -FROM - workspace_build_parameters -JOIN - workspace_builds ON workspace_builds.id = workspace_build_parameters.workspace_build_id -JOIN - workspaces ON workspaces.id = workspace_builds.workspace_id -WHERE - workspace_build_parameters.workspace_build_id = ANY($1 :: uuid[]) - -- Authorize Filter clause will be injected below in GetAuthorizedWorkspaceBuildParametersByBuildIDs - -- @authorize_filter -` - -func (q *sqlQuerier) GetWorkspaceBuildParametersByBuildIDs(ctx context.Context, workspaceBuildIds []uuid.UUID) ([]WorkspaceBuildParameter, error) { - rows, err := q.db.QueryContext(ctx, getWorkspaceBuildParametersByBuildIDs, pq.Array(workspaceBuildIds)) - if err != nil { - return nil, err - } - defer rows.Close() - var items []WorkspaceBuildParameter - for rows.Next() { - var i WorkspaceBuildParameter - if err := rows.Scan(&i.WorkspaceBuildID, &i.Name, &i.Value); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - const insertWorkspaceBuildParameters = `-- name: InsertWorkspaceBuildParameters :exec INSERT INTO workspace_build_parameters (workspace_build_id, name, value) diff --git a/coderd/database/queries/chatmodelconfigs.sql b/coderd/database/queries/chatmodelconfigs.sql index a24891d427..ec719760ad 100644 --- a/coderd/database/queries/chatmodelconfigs.sql +++ b/coderd/database/queries/chatmodelconfigs.sql @@ -16,21 +16,6 @@ WHERE is_default = TRUE AND deleted = FALSE; --- name: GetChatModelConfigByProviderAndModel :one -SELECT - * -FROM - chat_model_configs -WHERE - provider = @provider::text - AND model = @model::text - AND deleted = FALSE -ORDER BY - updated_at DESC, - created_at DESC, - id DESC -LIMIT 1; - -- name: GetChatModelConfigs :many SELECT * diff --git a/coderd/database/queries/chats.sql b/coderd/database/queries/chats.sql index dcb3b8e124..80435f24dd 100644 --- a/coderd/database/queries/chats.sql +++ b/coderd/database/queries/chats.sql @@ -5,12 +5,6 @@ WHERE id = @id OR root_chat_id = @id; -- name: UnarchiveChatByID :exec UPDATE chats SET archived = false, updated_at = NOW() WHERE id = @id::uuid; --- name: DeleteChatMessagesByChatID :exec -DELETE FROM - chat_messages -WHERE - chat_id = @chat_id::uuid; - -- name: DeleteChatMessagesAfterID :exec DELETE FROM chat_messages @@ -141,26 +135,6 @@ LIMIT -- Default to 50 to prevent accidental excessively large queries. COALESCE(NULLIF(@limit_opt :: int, 0), 50); --- name: ListChildChatsByParentID :many -SELECT - * -FROM - chats -WHERE - parent_chat_id = @parent_chat_id::uuid -ORDER BY - created_at ASC; - --- name: ListChatsByRootID :many -SELECT - * -FROM - chats -WHERE - root_chat_id = @root_chat_id::uuid -ORDER BY - created_at ASC; - -- name: InsertChat :one INSERT INTO chats ( owner_id, diff --git a/coderd/database/queries/files.sql b/coderd/database/queries/files.sql index 1e5892e425..cdf6e37ce0 100644 --- a/coderd/database/queries/files.sql +++ b/coderd/database/queries/files.sql @@ -8,22 +8,6 @@ WHERE LIMIT 1; --- name: GetFileIDByTemplateVersionID :one -SELECT - files.id -FROM - files -JOIN - provisioner_jobs ON - provisioner_jobs.storage_method = 'file' - AND provisioner_jobs.file_id = files.id -JOIN - template_versions ON template_versions.job_id = provisioner_jobs.id -WHERE - template_versions.id = @template_version_id -LIMIT - 1; - -- name: GetFileByHashAndCreator :one SELECT diff --git a/coderd/database/queries/gitsshkeys.sql b/coderd/database/queries/gitsshkeys.sql index 4365e3349b..a9b4353dd4 100644 --- a/coderd/database/queries/gitsshkeys.sql +++ b/coderd/database/queries/gitsshkeys.sql @@ -30,8 +30,3 @@ WHERE RETURNING *; --- name: DeleteGitSSHKey :exec -DELETE FROM - gitsshkeys -WHERE - user_id = $1; diff --git a/coderd/database/queries/groupmembers.sql b/coderd/database/queries/groupmembers.sql index 7de8dbe4e4..858a5937a3 100644 --- a/coderd/database/queries/groupmembers.sql +++ b/coderd/database/queries/groupmembers.sql @@ -31,25 +31,6 @@ WHERE group_id = @group_id user_is_system = false END; --- InsertUserGroupsByName adds a user to all provided groups, if they exist. --- name: InsertUserGroupsByName :exec -WITH groups AS ( - SELECT - id - FROM - groups - WHERE - groups.organization_id = @organization_id AND - groups.name = ANY(@group_names :: text []) -) -INSERT INTO - group_members (user_id, group_id) -SELECT - @user_id, - groups.id -FROM - groups; - -- InsertUserGroupsByID adds a user to all provided groups, if they exist. -- name: InsertUserGroupsByID :many WITH groups AS ( @@ -71,12 +52,6 @@ FROM ON CONFLICT DO NOTHING RETURNING group_id; --- name: RemoveUserFromAllGroups :exec -DELETE FROM - group_members -WHERE - user_id = @user_id; - -- name: RemoveUserFromGroups :many DELETE FROM group_members diff --git a/coderd/database/queries/oauth2.sql b/coderd/database/queries/oauth2.sql index 30d3e08027..e7162b5ab1 100644 --- a/coderd/database/queries/oauth2.sql +++ b/coderd/database/queries/oauth2.sql @@ -115,11 +115,6 @@ INSERT INTO oauth2_provider_app_secrets ( $6 ) RETURNING *; --- name: UpdateOAuth2ProviderAppSecretByID :one -UPDATE oauth2_provider_app_secrets SET - last_used_at = $2 -WHERE id = $1 RETURNING *; - -- name: DeleteOAuth2ProviderAppSecretByID :exec DELETE FROM oauth2_provider_app_secrets WHERE id = $1; @@ -249,5 +244,3 @@ WHERE id = $1 RETURNING *; -- name: DeleteOAuth2ProviderAppByClientID :exec DELETE FROM oauth2_provider_apps WHERE id = $1; --- name: GetOAuth2ProviderAppByRegistrationToken :one -SELECT * FROM oauth2_provider_apps WHERE registration_access_token = $1; diff --git a/coderd/database/queries/provisionerjobs.sql b/coderd/database/queries/provisionerjobs.sql index a11c18e9ee..f57f907631 100644 --- a/coderd/database/queries/provisionerjobs.sql +++ b/coderd/database/queries/provisionerjobs.sql @@ -67,14 +67,6 @@ WHERE id = $1 FOR UPDATE; --- name: GetProvisionerJobsByIDs :many -SELECT - * -FROM - provisioner_jobs -WHERE - id = ANY(@ids :: uuid [ ]); - -- name: GetProvisionerJobsByIDsWithQueuePosition :many WITH filtered_provisioner_jobs AS ( -- Step 1: Filter provisioner_jobs diff --git a/coderd/database/queries/siteconfig.sql b/coderd/database/queries/siteconfig.sql index 63b8076a46..129f0271b9 100644 --- a/coderd/database/queries/siteconfig.sql +++ b/coderd/database/queries/siteconfig.sql @@ -57,27 +57,6 @@ ON CONFLICT (key) DO UPDATE SET value = $1 WHERE site_configs.key = 'application -- name: GetApplicationName :one SELECT value FROM site_configs WHERE key = 'application_name'; --- name: GetAppSecurityKey :one -SELECT value FROM site_configs WHERE key = 'app_signing_key'; - --- name: UpsertAppSecurityKey :exec -INSERT INTO site_configs (key, value) VALUES ('app_signing_key', $1) -ON CONFLICT (key) DO UPDATE set value = $1 WHERE site_configs.key = 'app_signing_key'; - --- name: GetOAuthSigningKey :one -SELECT value FROM site_configs WHERE key = 'oauth_signing_key'; - --- name: UpsertOAuthSigningKey :exec -INSERT INTO site_configs (key, value) VALUES ('oauth_signing_key', $1) -ON CONFLICT (key) DO UPDATE set value = $1 WHERE site_configs.key = 'oauth_signing_key'; - --- name: GetCoordinatorResumeTokenSigningKey :one -SELECT value FROM site_configs WHERE key = 'coordinator_resume_token_signing_key'; - --- name: UpsertCoordinatorResumeTokenSigningKey :exec -INSERT INTO site_configs (key, value) VALUES ('coordinator_resume_token_signing_key', $1) -ON CONFLICT (key) DO UPDATE set value = $1 WHERE site_configs.key = 'coordinator_resume_token_signing_key'; - -- name: GetHealthSettings :one SELECT COALESCE((SELECT value FROM site_configs WHERE key = 'health_settings'), '{}') :: text AS health_settings diff --git a/coderd/database/queries/templateversions.sql b/coderd/database/queries/templateversions.sql index 128b2e5f58..e68383aa06 100644 --- a/coderd/database/queries/templateversions.sql +++ b/coderd/database/queries/templateversions.sql @@ -226,13 +226,6 @@ WHERE template_versions.id IN (archived_versions.id) RETURNING template_versions.id; --- name: GetTemplateVersionHasAITask :one -SELECT EXISTS ( - SELECT 1 - FROM template_versions - WHERE id = $1 AND has_ai_task = TRUE -); - -- name: UpdateTemplateVersionFlagsByJobID :exec UPDATE template_versions diff --git a/coderd/database/queries/user_links.sql b/coderd/database/queries/user_links.sql index 43e7fad64e..b352e80840 100644 --- a/coderd/database/queries/user_links.sql +++ b/coderd/database/queries/user_links.sql @@ -37,14 +37,6 @@ INSERT INTO VALUES ( $1, $2, $3, $4, $5, $6, $7, $8, $9 ) RETURNING *; --- name: UpdateUserLinkedID :one -UPDATE - user_links -SET - linked_id = $1 -WHERE - user_id = $2 AND login_type = $3 RETURNING *; - -- name: UpdateUserLink :one UPDATE user_links diff --git a/coderd/database/queries/workspaceagentstats.sql b/coderd/database/queries/workspaceagentstats.sql index ea4c14127d..28c17d8271 100644 --- a/coderd/database/queries/workspaceagentstats.sql +++ b/coderd/database/queries/workspaceagentstats.sql @@ -40,33 +40,6 @@ SELECT unnest(@connection_median_latency_ms :: double precision[]) AS connection_median_latency_ms, unnest(@usage :: boolean[]) AS usage; --- name: GetTemplateDAUs :many -SELECT - (created_at at TIME ZONE cast(@tz_offset::integer as text))::date as date, - user_id -FROM - workspace_agent_stats -WHERE - template_id = $1 AND - connection_count > 0 -GROUP BY - date, user_id -ORDER BY - date ASC; - --- name: GetDeploymentDAUs :many -SELECT - (created_at at TIME ZONE cast(@tz_offset::integer as text))::date as date, - user_id -FROM - workspace_agent_stats -WHERE - connection_count > 0 -GROUP BY - date, user_id -ORDER BY - date ASC; - -- name: DeleteOldWorkspaceAgentStats :exec DELETE FROM workspace_agent_stats diff --git a/coderd/database/queries/workspacebuildparameters.sql b/coderd/database/queries/workspacebuildparameters.sql index b639a553ef..2c09a84614 100644 --- a/coderd/database/queries/workspacebuildparameters.sql +++ b/coderd/database/queries/workspacebuildparameters.sql @@ -42,17 +42,3 @@ FROM ( ORDER BY created_at DESC, name LIMIT 100; --- name: GetWorkspaceBuildParametersByBuildIDs :many -SELECT - workspace_build_parameters.* -FROM - workspace_build_parameters -JOIN - workspace_builds ON workspace_builds.id = workspace_build_parameters.workspace_build_id -JOIN - workspaces ON workspaces.id = workspace_builds.workspace_id -WHERE - workspace_build_parameters.workspace_build_id = ANY(@workspace_build_ids :: uuid[]) - -- Authorize Filter clause will be injected below in GetAuthorizedWorkspaceBuildParametersByBuildIDs - -- @authorize_filter -;