chore: add schema version to workspace setting

proto/store/workspace_setting.proto

@@ -27,7 +27,10 @@ message WorkspaceSetting {
 }
 
 message WorkspaceBasicSetting {
+  // The secret key for workspace. Mainly used for session management.
   string secret_key = 1;
+  // The current schema version of database.
+  string schema_version = 2;
 }
 
 message WorkspaceGeneralSetting {

store/migrator.go

@@ -14,6 +14,7 @@ import (
 
 	"github.com/pkg/errors"
 
+	storepb "github.com/usememos/memos/proto/gen/store"
 	"github.com/usememos/memos/server/version"
 )
 
@@ -96,11 +97,15 @@ func (s *Store) Migrate(ctx context.Context) error {
 			slog.Info("end migrate")
 
 			// Upsert the current schema version to migration_history.
+			// TODO: retire using migration history later.
 			if _, err = s.driver.UpsertMigrationHistory(ctx, &UpsertMigrationHistory{
 				Version: schemaVersion,
 			}); err != nil {
 				return errors.Wrapf(err, "failed to upsert migration history with version: %s", schemaVersion)
 			}
+			if err := s.updateCurrentSchemaVersion(ctx, schemaVersion); err != nil {
+				return errors.Wrap(err, "failed to update current schema version")
+			}
 		}
 	} else if s.Profile.Mode == "demo" {
 		// In demo mode, we should seed the database.
@@ -112,6 +117,7 @@ func (s *Store) Migrate(ctx context.Context) error {
 }
 
 func (s *Store) preMigrate(ctx context.Context) error {
+	// TODO: using schema version in basic setting instead of migration history.
 	migrationHistoryList, err := s.driver.FindMigrationHistoryList(ctx, &FindMigrationHistory{})
 	// If any error occurs or no migration history found, apply the latest schema.
 	if err != nil || len(migrationHistoryList) == 0 {
@@ -141,11 +147,15 @@ func (s *Store) preMigrate(ctx context.Context) error {
 			return errors.Wrap(err, "failed to commit transaction")
 		}
 
+		// TODO: using schema version in basic setting instead of migration history.
 		if _, err := s.driver.UpsertMigrationHistory(ctx, &UpsertMigrationHistory{
 			Version: schemaVersion,
 		}); err != nil {
 			return errors.Wrap(err, "failed to upsert migration history")
 		}
+		if err := s.updateCurrentSchemaVersion(ctx, schemaVersion); err != nil {
+			return errors.Wrap(err, "failed to update current schema version")
+		}
 	}
 	if s.Profile.Mode == "prod" {
 		if err := s.normalizedMigrationHistoryList(ctx); err != nil {
@@ -295,3 +305,18 @@ func (s *Store) normalizedMigrationHistoryList(ctx context.Context) error {
 	}
 	return tx.Commit()
 }
+
+func (s *Store) updateCurrentSchemaVersion(ctx context.Context, schemaVersion string) error {
+	workspaceBasicSetting, err := s.GetWorkspaceBasicSetting(ctx)
+	if err != nil {
+		return errors.Wrap(err, "failed to get workspace basic setting")
+	}
+	workspaceBasicSetting.SchemaVersion = schemaVersion
+	if _, err := s.UpsertWorkspaceSetting(ctx, &storepb.WorkspaceSetting{
+		Key:   storepb.WorkspaceSettingKey_BASIC,
+		Value: &storepb.WorkspaceSetting_BasicSetting{BasicSetting: workspaceBasicSetting},
+	}); err != nil {
+		return errors.Wrap(err, "failed to upsert workspace setting")
+	}
+	return nil
+}