From ca72c7c7c70200379cebec3d0f1dff356d08c571 Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Wed, 10 May 2023 08:06:12 +0900
Subject: [PATCH] drive in functions

---
 functions/drive.js    | 7 ++-----
 src/ts/drive/drive.ts | 2 +-
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/functions/drive.js b/functions/drive.js
index fa75c5c6..5575e51b 100644
--- a/functions/drive.js
+++ b/functions/drive.js
@@ -3,16 +3,13 @@ export function onRequest(context) {
     return drive(request, context.env);
 }
 
-const encodedRedirectUri = encodeURIComponent("https://risuai.xyz/")
 
 async function drive(request, env){
 
     const url = new URL(request.url);
+    const encodedRedirectUri = url.host.includes('dev') ? encodeURIComponent("https://risu.pages.dev/") : encodeURIComponent("https://risuai.xyz/")
 
-    const headerE = {
-        "Access-Control-Allow-Origin": "https://risuai.xyz",
-        "Access-Control-Allow-Headers": "*"
-    }
+    const headerE = {}
     
     const params = url.searchParams
     const code = params.get('code')
diff --git a/src/ts/drive/drive.ts b/src/ts/drive/drive.ts
index 18a6b4d6..92d1ecd1 100644
--- a/src/ts/drive/drive.ts
+++ b/src/ts/drive/drive.ts
@@ -45,7 +45,7 @@ export async function checkDriverInit() {
         const code = loc.get('code')
     
         if(code){
-            const res = await fetch(`https://aichandict.xyz/api/drive/access?code=${encodeURIComponent(code)}`)
+            const res = await fetch(`/drive?code=${encodeURIComponent(code)}`)
             if(res.status >= 200 && res.status < 300){
                 const json:{
                     access_token:string,