ollama/app/src/index.ts

239 lines
6.6 KiB
TypeScript
Raw Normal View History

import { spawn } from 'child_process'
import { app, autoUpdater, dialog, Tray, Menu, BrowserWindow, nativeTheme } from 'electron'
2023-07-07 17:43:26 +00:00
import Store from 'electron-store'
2023-07-11 17:33:32 +00:00
import winston from 'winston'
import 'winston-daily-rotate-file'
2023-06-23 22:38:22 +00:00
import * as path from 'path'
2023-06-27 16:35:51 +00:00
2023-07-06 21:32:48 +00:00
import { analytics, id } from './telemetry'
2023-07-17 03:25:50 +00:00
import { installed } from './install'
2023-07-06 21:32:48 +00:00
2023-06-27 16:35:51 +00:00
require('@electron/remote/main').initialize()
2023-07-08 17:18:34 +00:00
const store = new Store()
2023-07-05 20:10:30 +00:00
let tray: Tray | null = null
let welcomeWindow: BrowserWindow | null = null
declare const MAIN_WINDOW_WEBPACK_ENTRY: string
2023-07-06 22:05:31 +00:00
2023-07-11 22:51:59 +00:00
const logger = winston.createLogger({
transports: [
new winston.transports.Console(),
new winston.transports.File({
filename: path.join(app.getPath('home'), '.ollama', 'logs', 'server.log'),
maxsize: 1024 * 1024 * 20,
maxFiles: 5,
}),
],
2023-07-17 03:26:12 +00:00
format: winston.format.printf(info => info.message),
2023-07-11 17:33:32 +00:00
})
2023-07-06 20:59:36 +00:00
const SingleInstanceLock = app.requestSingleInstanceLock()
2023-07-06 22:05:31 +00:00
if (!SingleInstanceLock) {
app.quit()
}
2023-07-06 18:41:25 +00:00
function firstRunWindow() {
// Create the browser window.
welcomeWindow = new BrowserWindow({
width: 400,
height: 500,
frame: false,
fullscreenable: false,
resizable: false,
2023-07-17 03:25:11 +00:00
movable: true,
show: false,
webPreferences: {
nodeIntegration: true,
contextIsolation: false,
},
2023-07-17 03:25:11 +00:00
alwaysOnTop: true,
})
require('@electron/remote/main').enable(welcomeWindow.webContents)
// and load the index.html of the app.
welcomeWindow.loadURL(MAIN_WINDOW_WEBPACK_ENTRY)
2023-07-17 03:25:11 +00:00
welcomeWindow.on('ready-to-show', () => welcomeWindow.show())
// for debugging
// welcomeWindow.webContents.openDevTools()
if (process.platform === 'darwin') {
app.dock.hide()
}
}
function createSystemtray() {
let iconPath = nativeTheme.shouldUseDarkColors
? path.join(__dirname, '..', '..', 'assets', 'ollama_icon_16x16Template.png')
: path.join(__dirname, '..', '..', 'assets', 'ollama_outline_icon_16x16Template.png')
2023-07-06 16:45:58 +00:00
if (app.isPackaged) {
iconPath = nativeTheme.shouldUseDarkColors
? path.join(process.resourcesPath, 'ollama_icon_16x16Template.png')
: path.join(process.resourcesPath, 'ollama_outline_icon_16x16Template.png')
2023-07-06 16:45:58 +00:00
}
2023-06-22 16:45:31 +00:00
tray = new Tray(iconPath)
2023-06-22 16:45:31 +00:00
nativeTheme.on('updated', function theThemeHasChanged () {
if (nativeTheme.shouldUseDarkColors) {
app.isPackaged
? tray.setImage(path.join(process.resourcesPath, 'ollama_icon_16x16Template.png'))
: tray.setImage(path.join(__dirname, '..', '..', 'assets', 'ollama_icon_16x16Template.png'))
} else {
app.isPackaged
? tray.setImage(path.join(process.resourcesPath, 'ollama_outline_icon_16x16Template.png'))
: tray.setImage(path.join(__dirname, '..', '..', 'assets', 'ollama_outline_icon_16x16Template.png'))
}
})
2023-07-08 17:18:34 +00:00
const contextMenu = Menu.buildFromTemplate([{ role: 'quit', label: 'Quit Ollama', accelerator: 'Command+Q' }])
2023-06-27 16:35:51 +00:00
2023-07-05 20:10:30 +00:00
tray.setContextMenu(contextMenu)
tray.setToolTip('Ollama')
}
if (require('electron-squirrel-startup')) {
app.quit()
2023-06-25 04:30:02 +00:00
}
2023-07-06 18:32:48 +00:00
function server() {
const binary = app.isPackaged
? path.join(process.resourcesPath, 'ollama')
2023-07-21 16:49:40 +00:00
: path.resolve(process.cwd(), '..', 'ollama')
2023-07-06 18:32:48 +00:00
2023-07-21 16:49:40 +00:00
const proc = spawn(binary, ['serve'])
2023-07-11 23:16:38 +00:00
2023-07-06 18:32:48 +00:00
proc.stdout.on('data', data => {
2023-07-21 16:49:40 +00:00
logger.info(data.toString().trim())
})
2023-07-11 23:16:38 +00:00
2023-07-06 18:32:48 +00:00
proc.stderr.on('data', data => {
2023-07-21 16:49:40 +00:00
logger.error(data.toString().trim())
})
proc.on('exit', (code, signal) => {
2023-07-21 17:09:28 +00:00
if (!code) {
2023-07-21 16:49:40 +00:00
logger.info('Server has stopped.')
setTimeout(server, 5000)
} else {
2023-07-21 16:49:40 +00:00
logger.error(`Server exited with code: ${code}, signal: ${signal}`)
setTimeout(server, 3000)
}
2023-07-21 16:49:40 +00:00
})
app.on('before-quit', () => {
2023-07-21 16:49:40 +00:00
proc.off('exit', server)
proc.kill()
})
2023-07-06 18:32:48 +00:00
}
if (process.platform === 'darwin') {
app.dock.hide()
}
2023-07-06 22:05:31 +00:00
app.on('ready', () => {
if (process.platform === 'darwin') {
if (app.isPackaged) {
if (!app.isInApplicationsFolder()) {
const chosen = dialog.showMessageBoxSync({
type: 'question',
buttons: ['Move to Applications', 'Do Not Move'],
message: 'Ollama works best when run from the Applications directory.',
defaultId: 0,
cancelId: 1,
})
if (chosen === 0) {
try {
app.moveToApplicationsFolder({
conflictHandler: conflictType => {
if (conflictType === 'existsAndRunning') {
dialog.showMessageBoxSync({
type: 'info',
message: 'Cannot move to Applications directory',
detail:
'Another version of Ollama is currently running from your Applications directory. Close it first and try again.',
})
}
return true
},
})
return
} catch (e) {
2023-07-11 17:33:32 +00:00
logger.error(`[Move to Applications] Failed to move to applications folder - ${e.message}}`)
}
}
}
}
2023-07-06 22:05:31 +00:00
}
2023-07-06 22:02:37 +00:00
2023-07-06 22:05:31 +00:00
createSystemtray()
server()
2023-07-17 03:25:11 +00:00
if (store.get('first-time-run') && installed()) {
app.setLoginItemSettings({ openAtLogin: app.getLoginItemSettings().openAtLogin })
2023-07-17 03:25:11 +00:00
return
}
2023-07-17 03:25:11 +00:00
// This is the first run or the CLI is no longer installed
app.setLoginItemSettings({ openAtLogin: true })
firstRunWindow()
2023-07-06 22:05:31 +00:00
})
2023-07-06 18:32:48 +00:00
2023-06-22 16:45:31 +00:00
// Quit when all windows are closed, except on macOS. There, it's common
// for applications and their menu bar to stay active until the user quits
// explicitly with Cmd + Q.
app.on('window-all-closed', () => {
if (process.platform !== 'darwin') {
app.quit()
}
})
// In this file you can include the rest of your app's specific main process
// code. You can also put them in separate files and import them here.
2023-07-06 04:04:06 +00:00
autoUpdater.setFeedURL({
url: `https://ollama.ai/api/update?os=${process.platform}&arch=${process.arch}&version=${app.getVersion()}`,
})
2023-06-27 21:50:50 +00:00
2023-07-06 21:32:48 +00:00
async function heartbeat() {
analytics.track({
anonymousId: id(),
event: 'heartbeat',
2023-07-07 17:44:36 +00:00
properties: {
version: app.getVersion(),
},
2023-07-06 21:32:48 +00:00
})
}
2023-07-06 18:50:46 +00:00
if (app.isPackaged) {
2023-07-06 21:32:48 +00:00
heartbeat()
2023-06-27 21:50:50 +00:00
autoUpdater.checkForUpdates()
2023-07-06 18:50:46 +00:00
setInterval(() => {
2023-07-06 21:32:48 +00:00
heartbeat()
2023-07-06 18:50:46 +00:00
autoUpdater.checkForUpdates()
2023-07-08 17:18:34 +00:00
}, 60 * 60 * 1000)
2023-07-06 18:50:46 +00:00
}
2023-06-28 00:31:02 +00:00
autoUpdater.on('error', e => {
2023-07-11 23:16:38 +00:00
logger.error(`update check failed - ${e.message}`)
})
2023-06-28 00:31:02 +00:00
autoUpdater.on('update-downloaded', (event, releaseNotes, releaseName) => {
dialog
.showMessageBox({
type: 'info',
buttons: ['Restart Now', 'Later'],
title: 'New update available',
message: process.platform === 'win32' ? releaseNotes : releaseName,
detail: 'A new version of Ollama is available. Restart to apply the update.',
})
.then(returnValue => {
if (returnValue.response === 0) autoUpdater.quitAndInstall()
})
})