index.ts 5.8 KB
Newer Older
1
import { spawn } from 'child_process'
2
3
4
5
6
7
8
9
10
11
12
import {
  app,
  autoUpdater,
  dialog,
  Tray,
  Menu,
  BrowserWindow,
  MenuItemConstructorOptions,
  nativeTheme,
  systemPreferences,
} from 'electron'
Eva Ho's avatar
Eva Ho committed
13
import Store from 'electron-store'
Eva Ho's avatar
Eva Ho committed
14
15
import winston from 'winston'
import 'winston-daily-rotate-file'
Bruce MacDonald's avatar
Bruce MacDonald committed
16
import * as path from 'path'
Jeffrey Morgan's avatar
Jeffrey Morgan committed
17

Jeffrey Morgan's avatar
Jeffrey Morgan committed
18
import { analytics, id } from './telemetry'
Jeffrey Morgan's avatar
Jeffrey Morgan committed
19
import { installed } from './install'
Jeffrey Morgan's avatar
Jeffrey Morgan committed
20
import { MenuItem } from '@electron/remote'
Jeffrey Morgan's avatar
Jeffrey Morgan committed
21

Jeffrey Morgan's avatar
Jeffrey Morgan committed
22
23
require('@electron/remote/main').initialize()

Jeffrey Morgan's avatar
Jeffrey Morgan committed
24
const store = new Store()
25

26
27
28
let welcomeWindow: BrowserWindow | null = null

declare const MAIN_WINDOW_WEBPACK_ENTRY: string
29

30
31
32
33
34
35
36
37
38
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,
    }),
  ],
Jeffrey Morgan's avatar
Jeffrey Morgan committed
39
  format: winston.format.printf(info => info.message),
Eva Ho's avatar
Eva Ho committed
40
41
})

Eva Ho's avatar
Eva Ho committed
42
const SingleInstanceLock = app.requestSingleInstanceLock()
43
44
45
if (!SingleInstanceLock) {
  app.quit()
}
Jeffrey Morgan's avatar
Jeffrey Morgan committed
46

47
48
49
50
51
52
53
54
function firstRunWindow() {
  // Create the browser window.
  welcomeWindow = new BrowserWindow({
    width: 400,
    height: 500,
    frame: false,
    fullscreenable: false,
    resizable: false,
55
56
    movable: true,
    show: false,
57
58
59
60
    webPreferences: {
      nodeIntegration: true,
      contextIsolation: false,
    },
61
    alwaysOnTop: true,
62
63
64
65
66
67
  })

  require('@electron/remote/main').enable(welcomeWindow.webContents)

  // and load the index.html of the app.
  welcomeWindow.loadURL(MAIN_WINDOW_WEBPACK_ENTRY)
68
69
  welcomeWindow.on('ready-to-show', () => welcomeWindow.show())

70
71
  if (process.platform === 'darwin') {
    app.dock.hide()
72
  }
73
74
}

75
let tray: Tray | null = null
Eva Ho's avatar
Eva Ho committed
76

77
function setTray(updateAvailable: boolean) {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
78
79
80
81
82
83
84
85
  const updateItems: MenuItemConstructorOptions[] = [
    { label: 'An update is available', enabled: false },
    {
      label: 'Restart to update',
      click: () => autoUpdater.quitAndInstall(),
    },
    { type: 'separator' },
  ]
86

87
  const menu = Menu.buildFromTemplate([
Jeffrey Morgan's avatar
Jeffrey Morgan committed
88
    ...(updateAvailable ? updateItems : []),
89
90
91
92
93
94
95
96
97
98
99
100
101
102
    { role: 'quit', label: 'Quit Ollama', accelerator: 'Command+Q' },
  ])

  const iconPath = app.isPackaged
    ? updateAvailable
      ? path.join(process.resourcesPath, 'iconUpdateTemplate.png')
      : path.join(process.resourcesPath, 'iconTemplate.png')
    : updateAvailable
    ? path.join(__dirname, '..', '..', 'assets', 'iconUpdateTemplate.png')
    : path.join(__dirname, '..', '..', 'assets', 'iconTemplate.png')

  if (!tray) {
    tray = new Tray(iconPath)
  }
Jeffrey Morgan's avatar
Jeffrey Morgan committed
103

104
105
106
  tray.setToolTip(updateAvailable ? 'An update is available' : 'Ollama')
  tray.setContextMenu(menu)
  tray.setImage(iconPath)
107
108
109
110
}

if (require('electron-squirrel-startup')) {
  app.quit()
Jeffrey Morgan's avatar
Jeffrey Morgan committed
111
112
}

Jeffrey Morgan's avatar
Jeffrey Morgan committed
113
114
function server() {
  const binary = app.isPackaged
Eva Ho's avatar
Eva Ho committed
115
116
    ? path.join(process.resourcesPath, 'ollama')
    : path.resolve(process.cwd(), '..', 'ollama')
Jeffrey Morgan's avatar
Jeffrey Morgan committed
117

Eva Ho's avatar
Eva Ho committed
118
  const proc = spawn(binary, ['serve'])
Jeffrey Morgan's avatar
Jeffrey Morgan committed
119

Jeffrey Morgan's avatar
Jeffrey Morgan committed
120
  proc.stdout.on('data', data => {
Eva Ho's avatar
Eva Ho committed
121
122
    logger.info(data.toString().trim())
  })
Jeffrey Morgan's avatar
Jeffrey Morgan committed
123

Jeffrey Morgan's avatar
Jeffrey Morgan committed
124
  proc.stderr.on('data', data => {
Eva Ho's avatar
Eva Ho committed
125
126
    logger.error(data.toString().trim())
  })
127

Eva Ho's avatar
Eva Ho committed
128
  function restart() {
hoyyeva's avatar
hoyyeva committed
129
    setTimeout(server, 3000)
Eva Ho's avatar
Eva Ho committed
130
131
  }

Eva Ho's avatar
Eva Ho committed
132
  proc.on('exit', restart)
133

134
  app.on('before-quit', () => {
Eva Ho's avatar
Eva Ho committed
135
    proc.off('exit', restart)
Eva Ho's avatar
Eva Ho committed
136
137
    proc.kill()
  })
Jeffrey Morgan's avatar
Jeffrey Morgan committed
138
139
}

140
141
if (process.platform === 'darwin') {
  app.dock.hide()
142
143
}

144
app.on('ready', () => {
145
146
147
148
149
150
151
152
153
  if (app.isPackaged) {
    heartbeat()
    autoUpdater.checkForUpdates()
    setInterval(() => {
      heartbeat()
      autoUpdater.checkForUpdates()
    }, 60 * 60 * 1000)
  }

Jeffrey Morgan's avatar
Jeffrey Morgan committed
154
155
  setTray(false)

156
  if (process.platform === 'darwin') {
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
    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) {
Eva Ho's avatar
Eva Ho committed
184
            logger.error(`[Move to Applications] Failed to move to applications folder - ${e.message}}`)
185
          }
186
187
188
        }
      }
    }
189
  }
Jeffrey Morgan's avatar
Jeffrey Morgan committed
190

191
  server()
192

193
  if (store.get('first-time-run') && installed()) {
194
    app.setLoginItemSettings({ openAtLogin: app.getLoginItemSettings().openAtLogin })
195
    return
196
  }
197
198
199
200

  // This is the first run or the CLI is no longer installed
  app.setLoginItemSettings({ openAtLogin: true })
  firstRunWindow()
201
})
Jeffrey Morgan's avatar
Jeffrey Morgan committed
202

Jeffrey Morgan's avatar
Jeffrey Morgan committed
203
204
205
206
207
208
209
210
211
212
213
// 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.
Jeffrey Morgan's avatar
Jeffrey Morgan committed
214
215
216
autoUpdater.setFeedURL({
  url: `https://ollama.ai/api/update?os=${process.platform}&arch=${process.arch}&version=${app.getVersion()}`,
})
217

Jeffrey Morgan's avatar
Jeffrey Morgan committed
218
219
220
221
async function heartbeat() {
  analytics.track({
    anonymousId: id(),
    event: 'heartbeat',
Jeffrey Morgan's avatar
Jeffrey Morgan committed
222
223
224
    properties: {
      version: app.getVersion(),
    },
Jeffrey Morgan's avatar
Jeffrey Morgan committed
225
226
227
  })
}

228
autoUpdater.on('error', e => {
229
  console.error(`update check failed - ${e.message}`)
230
231
})

232
233
autoUpdater.on('update-downloaded', () => {
  setTray(true)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
234
})