🚧 Store registration preference

client/components/admin/UserManagement.vue

@@ -1,15 +1,40 @@
 <script setup lang="ts">
-const disableRegistration = ref(false);
+import { useMessagesStore } from '~/store/messages';
+
+const { apiUrl } = useApiUrl();
+const { vtFetch } = useVtFetch();
+const messagesStore = useMessagesStore();
+
+const { data, refresh } = useGetServerSettings();
+
+const onRegistrationEnabledChange = async (value: boolean) => {
+  await vtFetch(`${apiUrl.value}admin/server-settings`, {
+    method: 'POST',
+    body: {
+      registrationEnabled: value
+    }
+  });
+  await refresh();
+  await nextTick();
+  messagesStore.createMessage({
+    type: 'info',
+    title: 'Server settings updated',
+    message: `Public registration ${value ? 'enabled' : 'disabled'}`
+  });
+};
 </script>
 
 <template>
-  <div class="user-management">
+  <div v-if="data" class="user-management">
     <ButtonsSwitchButton
-      label="Disable registration"
-      :value="disableRegistration"
-      small-label="Disables registration. Users can only be created by the admin."
+      label="Enable public registration"
+      :value="data.registrationEnabled"
+      small-label="Anyone can create an account."
+      small-label-negative="Accounts can only be created by the admin."
+      @valuechange="onRegistrationEnabledChange"
     />
   </div>
+  <Spinner v-if="!data" />
 </template>
 
 <style lang="scss" scoped>

client/components/buttons/SwitchButton.vue

@@ -18,9 +18,17 @@
     </div>
     <div v-if="label" class="label-container">
       <label :for="`switch-button-${randomId}`" class="label">{{ label }}</label>
-      <label v-if="smallLabel" :for="`switch-button-${randomId}`" class="small-label">{{
-        smallLabel
-      }}</label>
+
+      <label
+        v-if="smallLabel && smallLabelNegative && !value"
+        :for="`switch-button-${randomId}`"
+        class="small-label"
+      >
+        {{ smallLabelNegative }}
+      </label>
+      <label v-else-if="smallLabel" :for="`switch-button-${randomId}`" class="small-label">
+        {{ smallLabel }}
+      </label>
     </div>
   </div>
 </template>
@@ -33,7 +41,13 @@ export default defineComponent({
     label: String,
     smallLabel: {
       type: String,
-      required: false
+      required: false,
+      default: null
+    },
+    smallLabelNegative: {
+      type: String,
+      required: false,
+      default: null
     },
     disabled: Boolean,
     right: {
@@ -176,7 +190,9 @@ export default defineComponent({
           background-color: var(--theme-color);
           display: block;
           position: absolute;
-          transition: background-color 300ms $intro-easing, left 300ms $overshoot-easing;
+          transition:
+            background-color 300ms $intro-easing,
+            left 300ms $overshoot-easing;
         }
       }
     }

client/composables/api/admin.ts

@@ -21,6 +21,19 @@ export const useGetAdminInfo = () => {
   });
 };
 
+export const useGetServerSettings = () => {
+  const { apiUrl } = useApiUrl();
+  const { vtFetch } = useVtFetch();
+
+  return useLazyAsyncData(
+    'server-settings',
+    () => vtFetch<ApiDto<'ServerSettingsDto'>>(`${apiUrl.value}admin/server-settings`),
+    {
+      server: false
+    }
+  );
+};
+
 export const useGetBlockedVideos = () => {
   const { apiUrl } = useApiUrl();
 

server/src/admin/admin.controller.ts

@@ -16,6 +16,7 @@ import { Private } from 'server/auth/decorators/private.decorator';
 import { InfoDto } from './dto/info.dto';
 import { UserDto } from 'server/user/user.dto';
 import { UserprofileDto } from 'server/user/dto/userprofile.dto';
+import { ServerSettingsDto } from './dto/server-settings.dto';
 
 @ApiTags('Admin')
 @UseGuards(AdminGuard)
@@ -30,6 +31,16 @@ export class AdminController {
     return this.adminService.getInfo();
   }
 
+  @Get('server-settings')
+  getServerSettings(): Promise<ServerSettingsDto> {
+    return this.adminService.getServerSettings();
+  }
+
+  @Post('server-settings')
+  updateServerSettings(@Body() serverSettings: ServerSettingsDto): Promise<ServerSettingsDto> {
+    return this.adminService.updateServerSettings(serverSettings);
+  }
+
   @Get('logs')
   getLogs(): Promise<LogsDto> {
     return this.adminService.getLogs();

server/src/admin/admin.module.ts

@@ -7,6 +7,7 @@ import { AdminService } from './admin.service';
 import { AdminController } from './admin.controller';
 import { BlockedVideo, BlockedVideoSchema } from './schemas/blocked-video';
 import { UserModule } from '../user/user.module';
+import { ServerSettings, ServerSettingsSchema } from './schemas/server-settings';
 
 const moduleMetadata: ModuleMetadata = {
   imports: [
@@ -19,6 +20,11 @@ const moduleMetadata: ModuleMetadata = {
         name: BlockedVideo.name,
         schema: BlockedVideoSchema,
         collection: 'blocked-videos'
+      },
+      {
+        name: ServerSettings.name,
+        schema: ServerSettingsSchema,
+        collection: 'server-settings'
       }
     ]),
     UserModule

server/src/admin/admin.service.ts

@@ -13,13 +13,17 @@ import { proxyEnabled } from 'server/common/proxyAgent';
 import { UserprofileDto } from '../user/dto/userprofile.dto';
 import { UserService } from '../user/user.service';
 import { UserDto } from '../user/user.dto';
+import { ServerSettingsDto } from './dto/server-settings.dto';
+import { ServerSettings } from './schemas/server-settings';
 
 @Injectable()
 export class AdminService {
   constructor(
     private configService: ConfigService,
     @InjectModel(BlockedVideo.name)
-    private readonly blockedVideoModel: Model<BlockedVideo>,
+    private readonly BlockedVideoModel: Model<BlockedVideo>,
+    @InjectModel(ServerSettings.name)
+    private readonly ServerSettingsModel: Model<ServerSettings>,
     private userService: UserService
   ) {}
 
@@ -96,6 +100,35 @@ export class AdminService {
     return createdUser;
   }
 
+  defaultSettings: ServerSettingsDto = {
+    registrationEnabled: true
+  };
+
+  async getServerSettings(): Promise<ServerSettingsDto> {
+    const serverSettings = await this.ServerSettingsModel.findOne({ version: 1 }).exec();
+    if (!serverSettings) {
+      return this.defaultSettings;
+    }
+    return serverSettings;
+  }
+
+  async updateServerSettings(serverSettings: ServerSettingsDto): Promise<ServerSettingsDto> {
+    const settingsToSet: ServerSettingsDto = {
+      ...this.defaultSettings,
+      ...serverSettings
+    };
+
+    const newSettings = await this.ServerSettingsModel.findOneAndUpdate(
+      { version: 1 },
+      { version: 1, ...settingsToSet },
+      {
+        upsert: true
+      }
+    ).exec();
+
+    return newSettings;
+  }
+
   async dowloadLogFile(logFile: string): Promise<StreamableFile> {
     let logFolder = resolve(__dirname, '../logs');
     if (this.configService.get('VIEWTUBE_BASE_DIR')) {
@@ -117,12 +150,12 @@ export class AdminService {
   }
 
   async getAllBlockedVideoIds(): Promise<string[]> {
-    const blockedVideos = await this.blockedVideoModel.find().exec();
+    const blockedVideos = await this.BlockedVideoModel.find().exec();
     return blockedVideos.map(video => video.videoId);
   }
 
   async isVideoBlocked(videoId: string): Promise<boolean> {
-    const blockedVideo = await this.blockedVideoModel.findOne({ videoId }).exec();
+    const blockedVideo = await this.BlockedVideoModel.findOne({ videoId }).exec();
     if (!blockedVideo) {
       return false;
     }
@@ -134,7 +167,7 @@ export class AdminService {
     if (!videoIdToBlock) {
       throw new BadRequestException();
     }
-    const blockedVideo = await this.blockedVideoModel.findOneAndUpdate(
+    const blockedVideo = await this.BlockedVideoModel.findOneAndUpdate(
       { videoId },
       { videoId },
       { new: true, upsert: true }
@@ -143,6 +176,6 @@ export class AdminService {
   }
 
   async unblockVideoId(videoId: string): Promise<void> {
-    await this.blockedVideoModel.findOneAndDelete({ videoId });
+    await this.BlockedVideoModel.findOneAndDelete({ videoId });
   }
 }

server/src/admin/dto/server-settings.dto.ts

@@ -0,0 +1,3 @@
+export class ServerSettingsDto {
+  registrationEnabled: boolean;
+}

server/src/admin/schemas/server-settings.ts

@@ -0,0 +1,14 @@
+import { Document } from 'mongoose';
+import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
+import { ServerSettingsDto } from '../dto/server-settings.dto';
+
+@Schema()
+export class ServerSettings extends Document implements ServerSettingsDto {
+  @Prop()
+  version: number;
+
+  @Prop()
+  registrationEnabled: boolean;
+}
+
+export const ServerSettingsSchema = SchemaFactory.createForClass(ServerSettings);