No need to login with Stable-Diffusion page and Invite page

api/handler/power_log_handler.go

@@ -33,6 +33,8 @@ func (h *PowerLogHandler) List(c *gin.Context) {
 	}
 
 	session := h.DB.Session(&gorm.Session{})
+	userId := h.GetLoginUserId(c)
+	session = session.Where("user_id", userId)
 	if data.Model != "" {
 		session = session.Where("model", data.Model)
 	}

database/update-v4.0.0.sql

@@ -67,4 +67,8 @@ ALTER TABLE `chatgpt_api_keys` CHANGE `use_proxy` `proxy_url` VARCHAR(100) NULL
 UPDATE chatgpt_api_keys  set proxy_url='';
 
 -- 重置系统配置，系统配置的数据结构变了，旧数据解析会失败。
-UPDATE `chatgpt_configs` SET `config_json` = '{\"title\":\"ChatPlus AI 智能助手\",\"admin_title\":\"ChatPlus 控制台\",\"logo\":\"http://localhost:5678/static/upload/2024/3/1710732653645531.png\",\"init_power\":100,\"daily_power\":10,\"invite_power\":10,\"vip_month_power\":1000,\"register_ways\":[\"mobile\",\"username\",\"email\"],\"enabled_register\":true,\"reward_img\":\"http://localhost:5678/static/upload/2024/3/1710753716309668.jpg\",\"enabled_reward\":true,\"power_price\":0.1,\"order_pay_timeout\":1800,\"default_models\":[11,7,1,10,12,19,18,17],\"mj_power\":20,\"sd_power\":5,\"dall_power\":15,\"wechat_card_url\":\"/images/wx.png\",\"enable_context\":true,\"context_deep\":4}' WHERE `chatgpt_configs`.`id` = 1;
+UPDATE `chatgpt_configs` SET `config_json` = '{\"title\":\"ChatPlus AI 智能助手\",\"admin_title\":\"ChatPlus 控制台\",\"logo\":\"http://localhost:5678/static/upload/2024/3/1710732653645531.png\",\"init_power\":100,\"daily_power\":10,\"invite_power\":10,\"vip_month_power\":1000,\"register_ways\":[\"mobile\",\"username\",\"email\"],\"enabled_register\":true,\"reward_img\":\"http://localhost:5678/static/upload/2024/3/1710753716309668.jpg\",\"enabled_reward\":true,\"power_price\":0.1,\"order_pay_timeout\":1800,\"default_models\":[11,7,1,10,12,19,18,17],\"mj_power\":20,\"sd_power\":5,\"dall_power\":15,\"wechat_card_url\":\"/images/wx.png\",\"enable_context\":true,\"context_deep\":4}' WHERE `chatgpt_configs`.`id` = 1;
+
+-- 重置用户默认模型
+UPDATE `chatgpt_users` set chat_models_json = '[1]';
+

web/src/views/ImageMj.vue

@@ -471,16 +471,7 @@
 
 <script setup>
 import {nextTick, onMounted, onUnmounted, ref} from "vue"
-import {
-  ChromeFilled,
-  Delete,
-  DeleteFilled,
-  DocumentCopy,
-  InfoFilled,
-  Picture,
-  Plus,
-  Refresh
-} from "@element-plus/icons-vue";
+import {ChromeFilled, Delete, DocumentCopy, InfoFilled, Picture, Plus, Refresh} from "@element-plus/icons-vue";
 import Compressor from "compressorjs";
 import {httpGet, httpPost} from "@/utils/http";
 import {ElMessage, ElMessageBox, ElNotification} from "element-plus";
@@ -489,7 +480,7 @@ import Clipboard from "clipboard";
 import {checkSession} from "@/action/session";
 import {useRouter} from "vue-router";
 import {getSessionId} from "@/store/session";
-import {isMobile, removeArrayItem} from "@/utils/libs";
+import {removeArrayItem} from "@/utils/libs";
 import LoginDialog from "@/components/LoginDialog.vue";
 
 const listBoxHeight = ref(window.innerHeight - 40)
@@ -673,7 +664,6 @@ onMounted(() => {
 })
 
 // 初始化数据
-
 const initData = () => {
   checkSession().then(user => {
     power.value = user['power']
@@ -683,7 +673,6 @@ const initData = () => {
     fetchRunningJobs()
     fetchFinishJobs(1)
     connect()
-
   }).catch(() => {
 
   });

web/src/views/Invitation.vue

@@ -84,6 +84,8 @@
         </div>
       </div>
     </div>
+
+    <login-dialog :show="showLoginDialog" @hide="showLoginDialog =  false" @success="initData"/>
   </div>
 </template>
 
@@ -95,7 +97,7 @@ import {ElMessage} from "element-plus";
 import Clipboard from "clipboard";
 import InviteList from "@/components/InviteList.vue";
 import {checkSession} from "@/action/session";
-import {useRouter} from "vue-router";
+import LoginDialog from "@/components/LoginDialog.vue";
 
 const inviteURL = ref("")
 const qrImg = ref("")
@@ -104,10 +106,24 @@ const inviteImgCalls = ref(0)
 const hits = ref(0)
 const regNum = ref(0)
 const rate = ref(0)
-const router = useRouter()
 const isLogin = ref(false)
+const showLoginDialog = ref(true)
 
 onMounted(() => {
+  initData()
+
+  // 复制链接
+  const clipboard = new Clipboard('.copy-link');
+  clipboard.on('success', () => {
+    ElMessage.success('复制成功！');
+  })
+
+  clipboard.on('error', () => {
+    ElMessage.error('复制失败！');
+  })
+})
+
+const initData = () => {
   checkSession().then(() => {
     isLogin.value = true
     httpGet("/api/invite/code").then(res => {
@@ -136,19 +152,8 @@ onMounted(() => {
       ElMessage.error("获取系统配置失败：" + e.message)
     })
   }).catch(() => {
-    router.push('/login')
   });
-
-  // 复制链接
-  const clipboard = new Clipboard('.copy-link');
-  clipboard.on('success', () => {
-    ElMessage.success('复制成功！');
-  })
-
-  clipboard.on('error', () => {
-    ElMessage.error('复制失败！');
-  })
-})
+}
 </script>
 
 <style lang="stylus" scoped>