index.ts 5.6 KB
Newer Older
1
2
import { spawn } from 'child_process'
import { app, autoUpdater, dialog, Tray, Menu, BrowserWindow } from 'electron'
Eva Ho's avatar
Eva Ho committed
3
import Store from 'electron-store'
Eva Ho's avatar
Eva Ho committed
4
5
import winston from 'winston'
import 'winston-daily-rotate-file'
Bruce MacDonald's avatar
Bruce MacDonald committed
6
import * as path from 'path'
Jeffrey Morgan's avatar
Jeffrey Morgan committed
7

Jeffrey Morgan's avatar
Jeffrey Morgan committed
8
9
import { analytics, id } from './telemetry'

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

Jeffrey Morgan's avatar
Jeffrey Morgan committed
12
const store = new Store()
13
let tray: Tray | null = null
14
15
16
let welcomeWindow: BrowserWindow | null = null

declare const MAIN_WINDOW_WEBPACK_ENTRY: string
17

18
19
20
21
22
23
24
25
26
27
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,
    }),
  ],
  format: winston.format.printf(info => `${info.message}`),
Eva Ho's avatar
Eva Ho committed
28
29
})

Eva Ho's avatar
Eva Ho committed
30
const SingleInstanceLock = app.requestSingleInstanceLock()
31
32
33
if (!SingleInstanceLock) {
  app.quit()
}
Jeffrey Morgan's avatar
Jeffrey Morgan committed
34

35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
function firstRunWindow() {
  // Create the browser window.
  welcomeWindow = new BrowserWindow({
    width: 400,
    height: 500,
    frame: false,
    fullscreenable: false,
    resizable: false,
    movable: false,
    transparent: true,
    webPreferences: {
      nodeIntegration: true,
      contextIsolation: false,
    },
  })

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

  // and load the index.html of the app.
  welcomeWindow.loadURL(MAIN_WINDOW_WEBPACK_ENTRY)

  // for debugging
  // welcomeWindow.webContents.openDevTools()

  if (process.platform === 'darwin') {
    app.dock.hide()
61
  }
62
63
64
}

function createSystemtray() {
65
  let iconPath = path.join(__dirname, '..', '..', 'assets', 'ollama_icon_16x16Template.png')
Eva Ho's avatar
Eva Ho committed
66
67

  if (app.isPackaged) {
68
    iconPath = path.join(process.resourcesPath, 'ollama_icon_16x16Template.png')
Eva Ho's avatar
Eva Ho committed
69
  }
Jeffrey Morgan's avatar
Jeffrey Morgan committed
70

71
  tray = new Tray(iconPath)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
72

Jeffrey Morgan's avatar
Jeffrey Morgan committed
73
  const contextMenu = Menu.buildFromTemplate([{ role: 'quit', label: 'Quit Ollama', accelerator: 'Command+Q' }])
Jeffrey Morgan's avatar
Jeffrey Morgan committed
74

75
76
77
78
79
80
  tray.setContextMenu(contextMenu)
  tray.setToolTip('Ollama')
}

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

Jeffrey Morgan's avatar
Jeffrey Morgan committed
83
84
function server() {
  const binary = app.isPackaged
85
86
    ? path.join(process.resourcesPath, 'ollama')
    : path.resolve(process.cwd(), '..', 'ollama')
Jeffrey Morgan's avatar
Jeffrey Morgan committed
87

Jeffrey Morgan's avatar
Jeffrey Morgan committed
88
89
  const proc = spawn(binary, ['serve'])

Jeffrey Morgan's avatar
Jeffrey Morgan committed
90
  proc.stdout.on('data', data => {
91
    logger.info(data.toString().trim())
Jeffrey Morgan's avatar
Jeffrey Morgan committed
92
  })
Jeffrey Morgan's avatar
Jeffrey Morgan committed
93

Jeffrey Morgan's avatar
Jeffrey Morgan committed
94
  proc.stderr.on('data', data => {
95
    logger.error(data.toString().trim())
Jeffrey Morgan's avatar
Jeffrey Morgan committed
96
97
  })

98
  proc.on('exit', () => {
99
100
    logger.info('Restarting the server...')
    server()
101
102
103
  })

  proc.on('disconnect', () => {
104
105
    logger.info('Server disconnected. Reconnecting...')
    server()
106
107
  })

Jeffrey Morgan's avatar
Jeffrey Morgan committed
108
109
110
111
112
  process.on('exit', () => {
    proc.kill()
  })
}

113
114
if (process.platform === 'darwin') {
  app.dock.hide()
115
116
}

117
118
app.on('ready', () => {
  if (process.platform === 'darwin') {
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
    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
146
            logger.error(`[Move to Applications] Failed to move to applications folder - ${e.message}}`)
147
          }
148
149
150
        }
      }
    }
151
  }
Jeffrey Morgan's avatar
Jeffrey Morgan committed
152

153
  createSystemtray()
154
  server()
155

156
157
158
159
  if (!store.has('first-time-run')) {
    // This is the first run
    app.setLoginItemSettings({ openAtLogin: true })
    firstRunWindow()
160
    store.set('first-time-run', true)
161
162
163
164
  } else {
    // The app has been run before
    app.setLoginItemSettings({ openAtLogin: app.getLoginItemSettings().openAtLogin })
  }
165
})
Jeffrey Morgan's avatar
Jeffrey Morgan committed
166

Jeffrey Morgan's avatar
Jeffrey Morgan committed
167
168
169
170
171
172
173
174
175
176
177
// 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
178
179
180
autoUpdater.setFeedURL({
  url: `https://ollama.ai/api/update?os=${process.platform}&arch=${process.arch}&version=${app.getVersion()}`,
})
181

Jeffrey Morgan's avatar
Jeffrey Morgan committed
182
183
184
185
async function heartbeat() {
  analytics.track({
    anonymousId: id(),
    event: 'heartbeat',
Jeffrey Morgan's avatar
Jeffrey Morgan committed
186
187
188
    properties: {
      version: app.getVersion(),
    },
Jeffrey Morgan's avatar
Jeffrey Morgan committed
189
190
191
  })
}

192
if (app.isPackaged) {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
193
  heartbeat()
194
  autoUpdater.checkForUpdates()
195
  setInterval(() => {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
196
    heartbeat()
197
    autoUpdater.checkForUpdates()
Jeffrey Morgan's avatar
Jeffrey Morgan committed
198
  }, 60 * 60 * 1000)
199
}
Jeffrey Morgan's avatar
Jeffrey Morgan committed
200

201
autoUpdater.on('error', e => {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
202
  logger.error(`update check failed - ${e.message}`)
203
204
})

Jeffrey Morgan's avatar
Jeffrey Morgan committed
205
206
207
208
209
210
211
212
213
214
215
216
217
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()
    })
})