fix: gate quick ask with feature toggle (#2331)

This commit is contained in:
Louis 2024-03-12 20:10:59 +07:00 committed by GitHub
parent b9b421a495
commit af5bcea773
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
9 changed files with 74 additions and 7 deletions

View File

@ -4,13 +4,13 @@ import fs from 'fs'
import os from 'os' import os from 'os'
import childProcess from 'child_process' import childProcess from 'child_process'
// TODO: move this to core
const configurationFileName = 'settings.json' const configurationFileName = 'settings.json'
// TODO: do no specify app name in framework module // TODO: do no specify app name in framework module
const defaultJanDataFolder = join(os.homedir(), 'jan') const defaultJanDataFolder = join(os.homedir(), 'jan')
const defaultAppConfig: AppConfiguration = { const defaultAppConfig: AppConfiguration = {
data_folder: defaultJanDataFolder, data_folder: defaultJanDataFolder,
quick_ask: false,
} }
/** /**

View File

@ -1,3 +1,4 @@
export type AppConfiguration = { export type AppConfiguration = {
data_folder: string data_folder: string
quick_ask: boolean
} }

View File

@ -5,7 +5,7 @@ import { join } from 'path'
* Managers * Managers
**/ **/
import { windowManager } from './managers/window' import { windowManager } from './managers/window'
import { log } from '@janhq/core/node' import { getAppConfigurations, log } from '@janhq/core/node'
/** /**
* IPC Handlers * IPC Handlers
@ -95,7 +95,15 @@ app.once('quit', () => {
cleanUpAndQuit() cleanUpAndQuit()
}) })
app.once('window-all-closed', () => {
// Feature Toggle for Quick Ask
if (getAppConfigurations().quick_ask) return
cleanUpAndQuit()
})
function createQuickAskWindow() { function createQuickAskWindow() {
// Feature Toggle for Quick Ask
if (!getAppConfigurations().quick_ask) return
const startUrl = app.isPackaged ? `file://${quickAskPath}` : quickAskUrl const startUrl = app.isPackaged ? `file://${quickAskPath}` : quickAskUrl
windowManager.createQuickAskWindow(preloadPath, startUrl) windowManager.createQuickAskWindow(preloadPath, startUrl)
} }
@ -107,6 +115,9 @@ function createMainWindow() {
function registerGlobalShortcuts() { function registerGlobalShortcuts() {
const ret = registerShortcut(quickAskHotKey, (selectedText: string) => { const ret = registerShortcut(quickAskHotKey, (selectedText: string) => {
// Feature Toggle for Quick Ask
if (!getAppConfigurations().quick_ask) return
if (!windowManager.isQuickAskWindowVisible()) { if (!windowManager.isQuickAskWindowVisible()) {
windowManager.showQuickAskWindow() windowManager.showQuickAskWindow()
windowManager.sendQuickAskSelectedText(selectedText) windowManager.sendQuickAskSelectedText(selectedText)

View File

@ -1,11 +1,15 @@
import { join } from 'path' import { join } from 'path'
import { Tray, app, Menu } from 'electron' import { Tray, app, Menu } from 'electron'
import { windowManager } from '../managers/window' import { windowManager } from '../managers/window'
import { getAppConfigurations } from '@janhq/core/node'
class TrayManager { class TrayManager {
currentTray: Tray | undefined currentTray: Tray | undefined
createSystemTray = () => { createSystemTray = () => {
// Feature Toggle for Quick Ask
if (!getAppConfigurations().quick_ask) return
if (this.currentTray) { if (this.currentTray) {
return return
} }

View File

@ -2,6 +2,7 @@ import { BrowserWindow, app, shell } from 'electron'
import { quickAskWindowConfig } from './quickAskWindowConfig' import { quickAskWindowConfig } from './quickAskWindowConfig'
import { AppEvent } from '@janhq/core' import { AppEvent } from '@janhq/core'
import { mainWindowConfig } from './mainWindowConfig' import { mainWindowConfig } from './mainWindowConfig'
import { getAppConfigurations } from '@janhq/core/node'
/** /**
* Manages the current window instance. * Manages the current window instance.
@ -43,6 +44,9 @@ class WindowManager {
}) })
windowManager.mainWindow?.on('close', function (evt) { windowManager.mainWindow?.on('close', function (evt) {
// Feature Toggle for Quick Ask
if (!getAppConfigurations().quick_ask) return
if (!isAppQuitting) { if (!isAppQuitting) {
evt.preventDefault() evt.preventDefault()
windowManager.hideMainWindow() windowManager.hideMainWindow()

View File

@ -10,7 +10,10 @@ import useGetSystemResources from '@/hooks/useGetSystemResources'
import useModels from '@/hooks/useModels' import useModels from '@/hooks/useModels'
import useThreads from '@/hooks/useThreads' import useThreads from '@/hooks/useThreads'
import { janDataFolderPathAtom } from '@/helpers/atoms/AppConfig.atom' import {
janDataFolderPathAtom,
quickAskEnabledAtom,
} from '@/helpers/atoms/AppConfig.atom'
type Props = { type Props = {
children: ReactNode children: ReactNode
@ -18,6 +21,7 @@ type Props = {
const DataLoader: React.FC<Props> = ({ children }) => { const DataLoader: React.FC<Props> = ({ children }) => {
const setJanDataFolderPath = useSetAtom(janDataFolderPathAtom) const setJanDataFolderPath = useSetAtom(janDataFolderPathAtom)
const setQuickAskEnabled = useSetAtom(quickAskEnabledAtom)
useModels() useModels()
useThreads() useThreads()
@ -29,8 +33,9 @@ const DataLoader: React.FC<Props> = ({ children }) => {
?.getAppConfigurations() ?.getAppConfigurations()
?.then((appConfig: AppConfiguration) => { ?.then((appConfig: AppConfiguration) => {
setJanDataFolderPath(appConfig.data_folder) setJanDataFolderPath(appConfig.data_folder)
setQuickAskEnabled(appConfig.quick_ask)
}) })
}, [setJanDataFolderPath]) }, [setJanDataFolderPath, setQuickAskEnabled])
console.debug('Load Data...') console.debug('Load Data...')

View File

@ -6,6 +6,7 @@ const PROXY_FEATURE_ENABLED = 'proxyFeatureEnabled'
const VULKAN_ENABLED = 'vulkanEnabled' const VULKAN_ENABLED = 'vulkanEnabled'
const IGNORE_SSL = 'ignoreSSLFeature' const IGNORE_SSL = 'ignoreSSLFeature'
const HTTPS_PROXY_FEATURE = 'httpsProxyFeature' const HTTPS_PROXY_FEATURE = 'httpsProxyFeature'
const QUICK_ASK_ENABLED = 'quickAskEnabled'
export const janDataFolderPathAtom = atom('') export const janDataFolderPathAtom = atom('')
@ -19,3 +20,4 @@ export const proxyAtom = atomWithStorage(HTTPS_PROXY_FEATURE, '')
export const ignoreSslAtom = atomWithStorage(IGNORE_SSL, false) export const ignoreSslAtom = atomWithStorage(IGNORE_SSL, false)
export const vulkanEnabledAtom = atomWithStorage(VULKAN_ENABLED, false) export const vulkanEnabledAtom = atomWithStorage(VULKAN_ENABLED, false)
export const quickAskEnabledAtom = atomWithStorage(QUICK_ASK_ENABLED, false)

View File

@ -30,6 +30,7 @@ export default function useFactoryReset() {
// set the default jan data folder to user's home directory // set the default jan data folder to user's home directory
const configuration: AppConfiguration = { const configuration: AppConfiguration = {
data_folder: defaultJanDataFolder, data_folder: defaultJanDataFolder,
quick_ask: appConfiguration?.quick_ask ?? false,
} }
await window.core?.api?.updateAppConfiguration(configuration) await window.core?.api?.updateAppConfiguration(configuration)
} }

View File

@ -2,7 +2,7 @@
import { useEffect, useState, useCallback, ChangeEvent } from 'react' import { useEffect, useState, useCallback, ChangeEvent } from 'react'
import { openExternalUrl, fs } from '@janhq/core' import { openExternalUrl, fs, AppConfiguration } from '@janhq/core'
import { import {
Switch, Switch,
@ -23,7 +23,7 @@ import {
ScrollArea, ScrollArea,
} from '@janhq/uikit' } from '@janhq/uikit'
import { useAtom } from 'jotai' import { useAtom, useAtomValue } from 'jotai'
import { AlertTriangleIcon, AlertCircleIcon } from 'lucide-react' import { AlertTriangleIcon, AlertCircleIcon } from 'lucide-react'
import ShortcutModal from '@/containers/ShortcutModal' import ShortcutModal from '@/containers/ShortcutModal'
@ -42,6 +42,7 @@ import {
proxyAtom, proxyAtom,
proxyEnabledAtom, proxyEnabledAtom,
vulkanEnabledAtom, vulkanEnabledAtom,
quickAskEnabledAtom,
} from '@/helpers/atoms/AppConfig.atom' } from '@/helpers/atoms/AppConfig.atom'
type GPU = { type GPU = {
@ -56,6 +57,8 @@ const Advanced = () => {
) )
const [vulkanEnabled, setVulkanEnabled] = useAtom(vulkanEnabledAtom) const [vulkanEnabled, setVulkanEnabled] = useAtom(vulkanEnabledAtom)
const [proxyEnabled, setProxyEnabled] = useAtom(proxyEnabledAtom) const [proxyEnabled, setProxyEnabled] = useAtom(proxyEnabledAtom)
const quickAskEnabled = useAtomValue(quickAskEnabledAtom)
const [proxy, setProxy] = useAtom(proxyAtom) const [proxy, setProxy] = useAtom(proxyAtom)
const [ignoreSSL, setIgnoreSSL] = useAtom(ignoreSslAtom) const [ignoreSSL, setIgnoreSSL] = useAtom(ignoreSslAtom)
@ -87,6 +90,14 @@ const Advanced = () => {
[setPartialProxy, setProxy] [setPartialProxy, setProxy]
) )
const updateQuickAskEnabled = async (e: boolean) => {
const appConfiguration: AppConfiguration =
await window.core?.api?.getAppConfigurations()
appConfiguration.quick_ask = e
await window.core?.api?.updateAppConfiguration(appConfiguration)
window.core?.api?.relaunch()
}
useEffect(() => { useEffect(() => {
const setUseGpuIfPossible = async () => { const setUseGpuIfPossible = async () => {
const settings = await readSettings() const settings = await readSettings()
@ -361,7 +372,7 @@ const Advanced = () => {
Vulkan Support Vulkan Support
</h6> </h6>
</div> </div>
<p className="text-xs leading-relaxed"> <p className="leading-relaxed">
Enable Vulkan with AMD GPU/APU and Intel Arc GPU for better Enable Vulkan with AMD GPU/APU and Intel Arc GPU for better
model performance (reload needed). model performance (reload needed).
</p> </p>
@ -426,6 +437,34 @@ const Advanced = () => {
/> />
</div> </div>
<div className="flex w-full items-start justify-between border-b border-border py-4 first:pt-0 last:border-none">
<div className="flex-shrink-0 space-y-1.5">
<div className="flex gap-x-2">
<h6 className="text-sm font-semibold capitalize">
Jan Quick Ask
</h6>
</div>
<p className="leading-relaxed">
Enable Quick Ask to be triggered via the default hotkey{' '}
<div className="inline-flex items-center justify-center rounded-full bg-secondary px-1 py-0.5 text-xs font-bold text-muted-foreground">
<span className="font-bold">{isMac ? '⌘' : 'Ctrl'} + J</span>
</div>{' '}
(reload needed).
</p>
</div>
<Switch
checked={quickAskEnabled}
onCheckedChange={() => {
toaster({
title: 'Reload',
description:
'Quick Ask settings updated. Reload now to apply the changes.',
})
updateQuickAskEnabled(!quickAskEnabled)
}}
/>
</div>
{/* Clear log */} {/* Clear log */}
<div className="flex w-full items-start justify-between border-b border-border py-4 first:pt-0 last:border-none"> <div className="flex w-full items-start justify-between border-b border-border py-4 first:pt-0 last:border-none">
<div className="flex-shrink-0 space-y-1.5"> <div className="flex-shrink-0 space-y-1.5">