From 2bb3937d0d65a9fd86784118b5dfdae59aeeb886 Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Thu, 3 Aug 2023 13:03:23 +0900
Subject: [PATCH] [feat] local request warn

---
 src/lib/Setting/Pages/BotSettings.svelte | 28 ++++++++++++++----------
 src/lib/Setting/Settings.svelte          |  4 ++--
 src/ts/process/request.ts                |  7 ++++++
 src/ts/storage/globalApi.ts              | 11 ++++++++++
 4 files changed, 36 insertions(+), 14 deletions(-)

diff --git a/src/lib/Setting/Pages/BotSettings.svelte b/src/lib/Setting/Pages/BotSettings.svelte
index b0134ef6..598f414b 100644
--- a/src/lib/Setting/Pages/BotSettings.svelte
+++ b/src/lib/Setting/Pages/BotSettings.svelte
@@ -231,15 +231,17 @@
     {/if}
 {/if}
 {#if advancedBotSettings}
-    <span class="text-textcolor">{language.mainPrompt} <Help key="mainprompt"/></span>
-    <TextAreaInput fullwidth autocomplete="off" height={"32"} bind:value={$DataBase.mainPrompt}></TextAreaInput>
-    <span class="text-textcolor2 mb-6 text-sm mt-2">{tokens.mainPrompt} {language.tokens}</span>
-    <span class="text-textcolor">{language.jailbreakPrompt} <Help key="jailbreak"/></span>
-    <TextAreaInput fullwidth autocomplete="off" height={"32"} bind:value={$DataBase.jailbreak}></TextAreaInput>
-    <span class="text-textcolor2 mb-6 text-sm mt-2">{tokens.jailbreak} {language.tokens}</span>
-    <span class="text-textcolor">{language.globalNote} <Help key="globalNote"/></span>
-    <TextAreaInput fullwidth autocomplete="off" height={"32"} bind:value={$DataBase.globalNote}></TextAreaInput>
-    <span class="text-textcolor2 mb-6 text-sm mt-2">{tokens.globalNote} {language.tokens}</span>    
+    {#if !$DataBase.promptTemplate}
+        <span class="text-textcolor">{language.mainPrompt} <Help key="mainprompt"/></span>
+        <TextAreaInput fullwidth autocomplete="off" height={"32"} bind:value={$DataBase.mainPrompt}></TextAreaInput>
+        <span class="text-textcolor2 mb-6 text-sm mt-2">{tokens.mainPrompt} {language.tokens}</span>
+        <span class="text-textcolor">{language.jailbreakPrompt} <Help key="jailbreak"/></span>
+        <TextAreaInput fullwidth autocomplete="off" height={"32"} bind:value={$DataBase.jailbreak}></TextAreaInput>
+        <span class="text-textcolor2 mb-6 text-sm mt-2">{tokens.jailbreak} {language.tokens}</span>
+        <span class="text-textcolor">{language.globalNote} <Help key="globalNote"/></span>
+        <TextAreaInput fullwidth autocomplete="off" height={"32"} bind:value={$DataBase.globalNote}></TextAreaInput>
+        <span class="text-textcolor2 mb-6 text-sm mt-2">{tokens.globalNote} {language.tokens}</span>
+    {/if}    
 
     <span class="text-textcolor">{language.maxContextSize}</span>
     <NumberInput min={0} max={getModelMaxContext($DataBase.aiModel)} marginBottom={true} bind:value={$DataBase.maxContext}/>
@@ -414,9 +416,11 @@
         {/each}
     </table>
 
-    <div class="flex items-center mt-4">
-        <Check bind:check={$DataBase.promptPreprocess} name={language.promptPreprocess}/>
-    </div>
+    {#if !$DataBase.promptTemplate}
+        <div class="flex items-center mt-4">
+            <Check bind:check={$DataBase.promptPreprocess} name={language.promptPreprocess}/>
+        </div>
+    {/if}
     <div class="flex items-center mt-4">
         {#if $DataBase.promptTemplate}
             <Check check={true} name={language.usePromptTemplate} onChange={()=>{
diff --git a/src/lib/Setting/Settings.svelte b/src/lib/Setting/Settings.svelte
index 989b425e..4c71b271 100644
--- a/src/lib/Setting/Settings.svelte
+++ b/src/lib/Setting/Settings.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-    import { AccessibilityIcon, ActivityIcon, AlignLeft, BookIcon, BotIcon, BoxIcon, CodeIcon, ContactIcon, DiamondIcon, FolderIcon, LanguagesIcon, MonitorIcon, Sailboat, UserIcon, XCircleIcon } from "lucide-svelte";
+    import { AccessibilityIcon, ActivityIcon, AlignLeft, BookIcon, BotIcon, BoxIcon, CodeIcon, ContactIcon, DiamondIcon, FolderIcon, LanguagesIcon, MonitorIcon, Sailboat, ScrollTextIcon, UserIcon, XCircleIcon } from "lucide-svelte";
     import { language } from "src/lang";
     import DisplaySettings from "./Pages/DisplaySettings.svelte";
     import UserSettings from "./Pages/UserSettings.svelte";
@@ -48,7 +48,7 @@
                     on:click={() => {
                         selected = 13
                 }}>
-                        <BotIcon />
+                        <ScrollTextIcon />
                         <span>{language.prompt}</span>
                     </button>
                 {/if}
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 7a232ecb..f37ff8f0 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -213,6 +213,13 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
             let throughProxi = (!isTauri) && (!isNodeServer) && (!db.usePlainFetch)
             if(db.useStreaming && arg.useStreaming){
                 body.stream = true
+                let urlHost = new URL(replacerURL).host
+                if(urlHost.includes("localhost") || urlHost.includes("172.0.0.1") || urlHost.includes("0.0.0.0")){
+                    return {
+                        type: 'fail',
+                        result: 'You are trying local request on streaming. this is not allowed dude to browser/os security policy. turn off streaming.',
+                    }
+                }
                 const da =  (throughProxi)
                     ? await fetch(hubURL + `/proxy2`, {
                         body: JSON.stringify(body),
diff --git a/src/ts/storage/globalApi.ts b/src/ts/storage/globalApi.ts
index 327bed53..8ed6e5dd 100644
--- a/src/ts/storage/globalApi.ts
+++ b/src/ts/storage/globalApi.ts
@@ -511,6 +511,17 @@ export async function globalFetch(url:string, arg:{
     
         const urlHost = (new URL(url)).hostname
         let forcePlainFetch = (knownHostes.includes(urlHost) && (!isTauri)) || db.usePlainFetch || arg.plainFetchForce
+
+        console.log(urlHost)
+        //check if the url is a local url like localhost
+        if(urlHost.includes("localhost") || urlHost.includes("172.0.0.1") || urlHost.includes("0.0.0.0")){
+            return {
+                ok: false,
+                data: 'You are trying local request on web version. this is not allowed dude to browser security policy. use the desktop version instead, or use tunneling service like ngrok and set the cors to allow all.',
+                headers: {}
+            }
+        }
+
         if(forcePlainFetch){
             try {
                 let headers = arg.headers ?? {}