apps/admin/depends.py

@@ -0,0 +1,15 @@
+# @Time    : 2023/8/15 17:43
+# @Author  : Lan
+# @File    : depends.py
+# @Software: PyCharm
+from typing import Union
+
+from fastapi import Header, HTTPException
+from fastapi.requests import Request
+from core.settings import settings
+
+
+async def admin_required(authorization: Union[str, None] = Header(default=None), request: Request = None):
+    if authorization != settings.admin_token:
+        raise HTTPException(status_code=401, detail='未授权或授权校验失败')
+    return True

apps/admin/views.py

@@ -2,8 +2,9 @@
 # @Author  : Lan
 # @File    : views.py
 # @Software: PyCharm
-from fastapi import APIRouter
+from fastapi import APIRouter, Depends
 
+from apps.admin.depends import admin_required
 from apps.admin.pydantics import IDData
 from apps.base.models import FileCodes
 from core.response import APIResponse
@@ -16,7 +17,12 @@ admin_api = APIRouter(
 )
 
 
-@admin_api.delete('/file/delete')
+@admin_api.post('/login', dependencies=[Depends(admin_required)])
+async def login():
+    return APIResponse()
+
+
+@admin_api.delete('/file/delete', dependencies=[Depends(admin_required)])
 async def file_delete(data: IDData):
     file_code = await FileCodes.get(id=data.id)
     await file_storage.delete_file(file_code)
@@ -24,7 +30,7 @@ async def file_delete(data: IDData):
     return APIResponse()
 
 
-@admin_api.get('/file/list')
+@admin_api.get('/file/list', dependencies=[Depends(admin_required)])
 async def file_list(page: int = 1, size: int = 10):
     data = await FileCodes.all().limit(size).offset((page - 1) * size)
     return APIResponse(detail={
@@ -35,12 +41,12 @@ async def file_list(page: int = 1, size: int = 10):
     })
 
 
-@admin_api.get('/config/get')
+@admin_api.get('/config/get', dependencies=[Depends(admin_required)])
 async def get_config():
     return APIResponse(detail=settings.__dict__)
 
 
-@admin_api.patch('/config/update')
+@admin_api.patch('/config/update', dependencies=[Depends(admin_required)])
 async def update_config(data: dict):
     for k, v in data.items():
         settings.__setattr__(k, v)

apps/base/depends.py

@@ -10,10 +10,6 @@ from fastapi import Header, HTTPException, Request
 from core.response import APIResponse
 
 
-async def admin_required(pwd: Union[str, None] = Header(default=None), request: Request = None):
-    return False
-
-
 class IPRateLimit:
     def __init__(self, count, minutes):
         self.ips = {}

core/settings.py

@@ -17,6 +17,7 @@ default_value = {
     's3_secret_access_key': '',
     's3_bucket_name': '',
     's3_endpoint_url': '',
+    'admin_token': 'FileCodeBox2023',
     'openUpload': 1,
     'uploadSize': 1024 * 1024 * 10,
     'uploadMinute': 1,

fcb-fronted/src/stores/adminData.ts

@@ -0,0 +1,11 @@
+import { defineStore } from 'pinia';
+import {ref} from "vue";
+
+export const useAdminData = defineStore('adminData', () => {
+  const adminPassword = ref(localStorage.getItem('adminPassword') || '');
+  function updateAdminPwd(pwd: string) {
+    adminPassword.value = pwd;
+    localStorage.setItem('adminPassword', pwd);
+  }
+  return { adminPassword,updateAdminPwd };
+});

fcb-fronted/src/utils/request.ts

@@ -1,15 +1,17 @@
 // @ts-ignore
 import axios from "axios";
 import { ElMessage } from "element-plus";
-
 const instance = axios.create({
   baseURL: import.meta.env.DEV ? "http://localhost:12345" : "/",
   timeout: 6000000,
-  headers:{
-    'Authorization':localStorage.getItem('auth')
-  }
 });
-// 对响应进行拦截
+instance.interceptors.request.use(
+  (config: any) => {
+    config.headers= {
+      'Authorization': localStorage.getItem('adminPassword') || '',
+    }
+    return config;
+  });
 instance.interceptors.response.use(
   (response:any) => {
     if (response.data.code === 200) {

fcb-fronted/src/views/Admin/AdminView.vue

@@ -1,5 +1,5 @@
 <template>
-  <el-container style="height: 100vh;width: 100vw;position: relative;user-select: none">
+  <el-container v-if="isLogin" style="height: 100vh;width: 100vw;position: relative;user-select: none">
     <el-header>
       <el-menu mode="horizontal" router :default-active="route.path">
         <el-menu-item v-for="menu in menus" :index="menu.path" :key="menu.path">{{menu.name}}</el-menu-item>
@@ -10,13 +10,32 @@
       <router-view/>
     </el-main>
   </el-container>
+  <el-form size="large" v-else>
+    <el-form-item label="管理密码">
+      <el-input
+          v-model="adminData.adminPassword"
+          placeholder="请输入密码！"
+          type="password"
+      >
+        <template #append>
+          <el-button @click="refreshLoginStatus">登录</el-button>
+        </template>
+      </el-input>
+    </el-form-item>
+  </el-form>
 </template>
 <script setup lang="ts">
 import { useDark, useToggle } from '@vueuse/core';
 import { ref } from "vue";
 const isDark = useDark()
+const isLogin = ref(false);
 const toggleDark = useToggle(isDark)
 import { useRoute } from 'vue-router';
+import { useAdminData } from "@/stores/adminData";
+import { request } from "@/utils/request";
+import { ElMessage } from "element-plus";
+
+const adminData = useAdminData();
 const route = useRoute();
 const menus = ref([
   {
@@ -32,6 +51,22 @@ const menus = ref([
     path: '/admin/about',
   }
 ]);
+const refreshLoginStatus = () => {
+  console.log(adminData.adminPassword);
+  adminData.updateAdminPwd(adminData.adminPassword);
+  request({
+    url: '/admin/login',
+    method: 'post',
+  }).then((res: any) => {
+    if (res.code === 200) {
+      isLogin.value = true;
+      ElMessage.success('登录成功！');
+    } else {
+      ElMessage.error(res.detail);
+    }
+  });
+};
+refreshLoginStatus();
 </script>
 <style lang="scss" scoped>
 </style>

fcb-fronted/src/views/Admin/SettingView.vue

@@ -9,6 +9,9 @@
     <el-form-item size="large" label="关键词" style="letter-spacing: 0.3rem">
       <el-input v-model="config.keywords" />
     </el-form-item>
+    <el-form-item size="large" label="后台密码">
+      <el-input type="password" v-model="config.admin_token" />
+    </el-form-item>
     <el-form-item size="large" label="文件大小">
       <el-input type="number" v-model="config.uploadSize" />
       <template #append>Bit</template>
@@ -80,6 +83,7 @@ const config = ref({
   name: '',
   description: '',
   file_storage: '',
+  admin_token: '',
   keywords: '',
   openUpload: 1,
   uploadSize: 1,