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

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

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

Jeffrey Morgan's avatar
Jeffrey Morgan committed
11
const store = new Store()
12
let tray: Tray | null = null
13

Eva Ho's avatar
Eva Ho committed
14
const SingleInstanceLock = app.requestSingleInstanceLock()
15
16
17
if (!SingleInstanceLock) {
  app.quit()
}
Jeffrey Morgan's avatar
Jeffrey Morgan committed
18

19
const createSystemtray = () => {
20
  let iconPath = path.join(__dirname, '..', '..', 'assets', 'ollama_icon_16x16Template.png')
Eva Ho's avatar
Eva Ho committed
21
22

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

26
  tray = new Tray(iconPath)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
27

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

30
31
32
33
34
35
36
  tray.setContextMenu(contextMenu)
  tray.setToolTip('Ollama')
}

// Handle creating/removing shortcuts on Windows when installing/uninstalling.
if (require('electron-squirrel-startup')) {
  app.quit()
Jeffrey Morgan's avatar
Jeffrey Morgan committed
37
38
}

Jeffrey Morgan's avatar
Jeffrey Morgan committed
39
40
const ollama = path.join(process.resourcesPath, 'ollama')

Jeffrey Morgan's avatar
Jeffrey Morgan committed
41
42
// if the app is packaged then run the server
if (app.isPackaged) {
Bruce MacDonald's avatar
Bruce MacDonald committed
43
  // Start the executable
44
45
  console.log(`Starting server`)
  const proc = spawn(ollama, ['serve'])
Jeffrey Morgan's avatar
Jeffrey Morgan committed
46
  proc.stdout.on('data', data => {
Bruce MacDonald's avatar
Bruce MacDonald committed
47
48
    console.log(`server: ${data}`)
  })
Jeffrey Morgan's avatar
Jeffrey Morgan committed
49
  proc.stderr.on('data', data => {
Bruce MacDonald's avatar
Bruce MacDonald committed
50
51
52
    console.error(`server: ${data}`)
  })

Jeffrey Morgan's avatar
Jeffrey Morgan committed
53
54
55
  process.on('exit', () => {
    proc.kill()
  })
Jeffrey Morgan's avatar
Jeffrey Morgan committed
56
57
}

Jeffrey Morgan's avatar
Jeffrey Morgan committed
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
function server() {
  const binary = app.isPackaged
    ? path.join(process.resourcesPath, 'ollama')
    : path.resolve(__dirname, '..', '..', 'ollama')

  console.log(`Starting server`)
  const proc = spawn(binary, ['serve'])
  proc.stdout.on('data', data => {
    console.log(`server: ${data}`)
  })
  proc.stderr.on('data', data => {
    console.error(`server: ${data}`)
  })

  process.on('exit', () => {
    proc.kill()
  })
}

77
function installCLI() {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
78
79
80
81
82
83
  const symlinkPath = '/usr/local/bin/ollama'

  if (fs.existsSync(symlinkPath) && fs.readlinkSync(symlinkPath) === ollama) {
    return
  }

84
85
86
87
  dialog
    .showMessageBox({
      type: 'info',
      title: 'Ollama CLI installation',
Jeffrey Morgan's avatar
Jeffrey Morgan committed
88
      message: 'To make the Ollama command work in your terminal, it needs administrator privileges.',
89
90
91
92
      buttons: ['OK'],
    })
    .then(result => {
      if (result.response === 0) {
93
        const command = `
Jeffrey Morgan's avatar
Jeffrey Morgan committed
94
95
    do shell script "ln -F -s ${ollama} /usr/local/bin/ollama" with administrator privileges
    `
96
97
98
99
100
101
102
103
104
105
106
107
        exec(`osascript -e '${command}'`, (error: Error | null, stdout: string, stderr: string) => {
          if (error) {
            console.error(`exec error: ${error}`)
            return
          }
          console.log(`stdout: ${stdout}`)
          console.error(`stderr: ${stderr}`)
        })
      }
    })
}

108
109
110
111
app.on('ready', () => {
  if (process.platform === 'darwin') {
    app.dock.hide()

Eva Ho's avatar
Eva Ho committed
112
113
114
    if (!store.has('first-time-run')) {
      // This is the first run
      app.setLoginItemSettings({ openAtLogin: true })
Jeffrey Morgan's avatar
Jeffrey Morgan committed
115
      store.set('first-time-run', false)
Eva Ho's avatar
Eva Ho committed
116
117
118
119
120
    } else {
      // The app has been run before
      app.setLoginItemSettings({ openAtLogin: app.getLoginItemSettings().openAtLogin })
    }

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
146
147
148
    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) {
          console.error('Failed to move to applications folder')
          console.error(e)
149
150
151
        }
      }
    }
152
  }
Jeffrey Morgan's avatar
Jeffrey Morgan committed
153

154
  createSystemtray()
Jeffrey Morgan's avatar
Jeffrey Morgan committed
155

156
157
158
159
  if (app.isPackaged) {
    installCLI()
  }
})
Jeffrey Morgan's avatar
Jeffrey Morgan committed
160

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

Jeffrey Morgan's avatar
Jeffrey Morgan committed
176
177
178
179
async function heartbeat() {
  analytics.track({
    anonymousId: id(),
    event: 'heartbeat',
Jeffrey Morgan's avatar
Jeffrey Morgan committed
180
181
182
    properties: {
      version: app.getVersion(),
    },
Jeffrey Morgan's avatar
Jeffrey Morgan committed
183
184
185
  })
}

186
if (app.isPackaged) {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
187
  heartbeat()
188
  autoUpdater.checkForUpdates()
189
  setInterval(() => {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
190
    heartbeat()
191
    autoUpdater.checkForUpdates()
Jeffrey Morgan's avatar
Jeffrey Morgan committed
192
  }, 60 * 60 * 1000)
193
}
Jeffrey Morgan's avatar
Jeffrey Morgan committed
194

195
196
197
198
autoUpdater.on('error', e => {
  console.error('update check failed', e)
})

Jeffrey Morgan's avatar
Jeffrey Morgan committed
199
200
201
202
203
204
205
206
207
208
209
210
211
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()
    })
})