Works with deep links

whathes · Mar 29, 2023 · 6b744fb · 6b744fb
1 parent 5c03e9d
commit 6b744fb
Show file tree

Hide file tree

Showing 3 changed files with 84 additions and 85 deletions.
diff --git a/src/main/auth.ts b/src/main/auth.ts
@@ -7,6 +7,7 @@ import {
     BrowserWindow,
     ipcMain,
     IpcMainInvokeEvent,
+    webContents,
 } from 'electron'
 import { API_ROOT } from '../utils'
 import crypto from 'crypto'
@@ -17,11 +18,14 @@ const store = new Store()
 
 let win: BrowserWindow | null = null
 
-// const auth0Domain = 'cursor.us.auth0.com'
-// const clientId = 'KbZUR41cY7W6zRSdpSUJ7I7mLYBKOCmB'
-
 const auth0Domain = 'cursor.us.auth0.com'
 const clientId = 'KbZUR41cY7W6zRSdpSUJ7I7mLYBKOCmB'
+const cursorHome = 'https://cursor.so'
+
+// Test domain/client
+// const auth0Domain = 'dev-27d5cph2nbetfllb.us.auth0.com'
+// const clientId = 'OzaBXLClY5CAGxNzUhQ2vlknpi07tGuE'
+// const cursorHome = 'http://localhost:4000'
 
 let accessToken: string | null = null
 let profile: any | null = null
@@ -38,12 +42,15 @@ const redirectUri = AUTH0_CALLBACK_URL
 const DUMMY_URL = `${API_ROOT}/dummy/*`
 const API_AUDIENCE = `https://${auth0Domain}/api/v2/`
 
-const loginUrl = `${API_ROOT}/auth/login`
-const signUpUrl = `${API_ROOT}/auth/signUp`
-const logoutUrl = `${API_ROOT}/auth/logout`
-const settingsUrl = `${API_ROOT}/auth/settings`
+// These are routes that exist on our homepage
+const loginUrl = `${cursorHome}/api/auth/loginDeep`
+const signUpUrl = `${cursorHome}/api/auth/loginDeep`
+const settingsUrl = `${cursorHome}/settings`
 const supportUrl = `${API_ROOT}/auth/support`
-const payUrl = `${API_ROOT}/auth/pay`
+
+// These are api routes
+const logoutUrl = `${API_ROOT}/api/auth/logout`
+const payUrl = `${API_ROOT}/api/auth/checkoutDeep`
 
 const storeWrapper = {
     get: async (key: string) => {
@@ -79,34 +86,6 @@ function sha256(buffer: Buffer) {
     return crypto.createHash('sha256').update(buffer).digest()
 }
 
-export function getAuthenticationURL() {
-    verifier = base64URLEncode(crypto.randomBytes(32))
-    const challenge = base64URLEncode(sha256(Buffer.from(verifier)))
-
-    const state = Math.random().toString(36).substring(2, 18)
-    const scope = 'openid profile offline_access'
-    const responseType = 'code'
-
-    const queryParams = {
-        audience: API_AUDIENCE,
-        client_id: clientId,
-        redirect_uri: AUTH0_CALLBACK_URL,
-        state,
-        scope,
-        response_type: responseType,
-        code_challenge: challenge,
-        code_challenge_method: 'S256', // SHA-256
-    }
-
-    return {
-        url: `${loginUrl}?${new URLSearchParams(
-            queryParams
-        )}`,
-        state,
-        verifier,
-    }
-}
-
 export async function stripeUrlRequest(window: BrowserWindow) {
     const response = await fetch(`${API_ROOT}/auth/create-checkout-session`, {
         method: 'POST',
@@ -124,8 +103,9 @@ export async function stripeUrlRequest(window: BrowserWindow) {
     window.loadURL(newUrl)
 }
 
-export async function refreshTokens(event: IpcMainInvokeEvent) {
+export async function refreshTokens(event?: IpcMainInvokeEvent) {
     const refreshToken = await storeWrapper.get('refreshToken')
+    console.log('retrieving refreshToken', refreshToken)
 
     if (refreshToken) {
         const refreshOptions = {
@@ -136,20 +116,25 @@ export async function refreshTokens(event: IpcMainInvokeEvent) {
                 client_id: clientId,
                 refresh_token: refreshToken,
                 // audience: API_AUDIENCE,
-                state: 'thisisatest',
+                // state: 'thisisatest',
             }),
         }
         try {
             const response = await fetch(
                 `https://${auth0Domain}/oauth/token`,
                 refreshOptions
             )
-            const data = (await response.json()) as {
+            console.log('resp', response)
+            console.log(response.status)
+            const origData = await response.json()
+            console.log('Orig data', origData)
+            const data = origData as {
                 access_token: string
                 id_token: string
             }
 
             accessToken = data.access_token
+            console.log('GETTING BACK PROFILE', data.id_token)
             profile = jwtDecode(data.id_token)
         } catch (error) {
             // await logout(parentWindow)
@@ -161,57 +146,34 @@ export async function refreshTokens(event: IpcMainInvokeEvent) {
     }
 
     console.log('UPDATING AUTH STATUS IN refresh tokens')
-    event.sender.send('updateAuthStatus', { accessToken, profile })
+    if (event) {
+        event.sender.send('updateAuthStatus', { accessToken, profile })
+    }
 }
 
-export async function loadTokens(
+export async function setupTokens(
     callbackURL: string,
-    window: BrowserWindow
+    // window: BrowserWindow
 ) {
     const urlParts = url.parse(callbackURL, true)
     const query = urlParts.query
-
-    const exchangeOptions = {
-        grant_type: 'authorization_code',
-        client_id: clientId,
-        code: query.code,
-        redirect_uri: redirectUri,
-        code_verifier: verifier,
-        // audience: API_AUDIENCE,
-    }
-
-    const options = {
-        method: 'POST',
-        headers: {
-            'content-type': 'application/json',
-        },
-        body: JSON.stringify(exchangeOptions),
+    const host = urlParts.host
+    //
+    if (host?.toLowerCase() === 'changetokens') {
+        console.log('settings access and refresh')
+        accessToken = query.accessToken as string
+        refreshToken = query.refreshToken as string
+        console.log('storing refreshToken', refreshToken)
+
+        await storeWrapper.set('refreshToken', refreshToken)
     }
+    // Get the profile id from this
+    await refreshTokens()
+    await loadStripeProfile()
+    webContents.getAllWebContents().forEach((wc) => {
+        wc.send('updateAuthStatus', { accessToken, profile, stripeProfile })
+    })
 
-    try {
-        const response = await fetch(
-            `https://${auth0Domain}/oauth/token`,
-            options
-        )
-
-        const data = (await response.json()) as {
-            access_token: string
-            id_token: string
-            refresh_token?: string
-        }
-        accessToken = data.access_token
-        profile = jwtDecode(data.id_token)
-        refreshToken = data.refresh_token!
-
-        if (refreshToken) {
-            await storeWrapper.set('refreshToken', refreshToken)
-        }
-    } catch (error) {
-        await logout(window)
-        // destroyAuthWin(window)
-
-        throw error
-    }
 }
 
 export async function loadStripeProfile() {

diff --git a/src/main/main.ts b/src/main/main.ts
@@ -35,23 +35,59 @@ import { setupStoreHandlers } from './storeHandler'
 import { resourcesDir } from './utils'
 import { setupIndex } from './indexer'
 
-import { authPackage, refreshTokens } from './auth'
+import { authPackage, setupTokens } from './auth'
 
 import { setupTerminal } from './terminal'
 import todesktop from '@todesktop/runtime'
 todesktop.init()
 
 const store = new Store()
 
+let main_window: Electron.BrowserWindow
+
+if (process.defaultApp) {
+    if (process.argv.length >= 2) {
+        app.setAsDefaultProtocolClient('electron-fiddle', process.execPath, [path.resolve(process.argv[1])])
+    }
+} else {
+    app.setAsDefaultProtocolClient('electron-fiddle')
+}
+
+const gotTheLock = app.requestSingleInstanceLock()
+
+if (!gotTheLock) {
+  app.quit()
+} else {
+  app.on('second-instance', (event, commandLine, workingDirectory) => {
+    // Someone tried to run a second instance, we should focus our window.
+    if (main_window) {
+      if (main_window.isMinimized()) main_window.restore()
+      main_window.focus()
+    }
+    console.log('second instance')
+    const url = commandLine.pop()?.slice(0,-1)
+    // dialog.showErrorBox('Welcome Back (in app already)', `You arrived from: ${url}`)
+    if (url) {
+        setupTokens(url)
+    }
+  })
+}
+
 type Event = IpcMainInvokeEvent
 process.env['ELECTRON_DISABLE_SECURITY_WARNINGS'] = 'true'
 
 // Handle creating/removing shortcuts on Windows when installing/uninstalling.
 if (require('electron-squirrel-startup')) {
     app.quit()
 }
+
+app.on('open-url', (event, url) => {
+    // dialog.showErrorBox('Welcome (first time i think)', `You arrived from: ${url}`)
+    if (url) {
+        setupTokens(url)
+    }
+})
 
-let main_window: Electron.BrowserWindow
 
 // Remove holded defaults
 if (process.platform === 'darwin')

diff --git a/src/utils.ts b/src/utils.ts
@@ -1,4 +1,5 @@
-export const API_ROOT = 'https://aicursor.com'
+// export const API_ROOT = 'https://aicursor.com'
+export const API_ROOT = 'http://localhost:8000'
 
 export class NoAuthRateLimitError extends Error {
     constructor(