index.ts 6 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'
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
import * as fs from 'fs'
Jeffrey Morgan's avatar
Jeffrey Morgan committed
8

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

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

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

Eva Ho's avatar
Eva Ho committed
16
17
18
19
const logFile = new winston.transports.File({
  filename: path.join(app.getPath('home'), '.ollama', 'logs', 'server.log'),
  maxsize: 1024 * 1024 * 20,
  maxFiles: 5,
Eva Ho's avatar
Eva Ho committed
20
21
22
23
24
25
26
27
28
29
});

const logger = winston.createLogger({ 
  transports: [logFile],
  format: winston.format.combine(
    winston.format.timestamp(),
    winston.format.printf(info => `${info.timestamp} ${info.level}: ${info.message}`)
  )
})

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
const createSystemtray = () => {
36
  let iconPath = path.join(__dirname, '..', '..', 'assets', 'ollama_icon_16x16Template.png')
Eva Ho's avatar
Eva Ho committed
37
38

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

42
  tray = new Tray(iconPath)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
43

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

46
47
48
49
50
51
52
  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
53
54
}

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

Jeffrey Morgan's avatar
Jeffrey Morgan committed
57
58
function server() {
  const binary = app.isPackaged
59
60
  ? path.join(process.resourcesPath, 'ollama')
  : path.resolve(process.cwd(), '..', 'ollama')
Jeffrey Morgan's avatar
Jeffrey Morgan committed
61
62
63

  const proc = spawn(binary, ['serve'])
  proc.stdout.on('data', data => {
Eva Ho's avatar
Eva Ho committed
64
    logger.info(`server: ${data.toString()}`)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
65
66
  })
  proc.stderr.on('data', data => {
Eva Ho's avatar
Eva Ho committed
67
    logger.error(`server: ${data.toString()}`)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
68
69
  })

70
  proc.on('exit', () => {
Eva Ho's avatar
Eva Ho committed
71
    logger.info('Restarting the server...');
72
73
74
75
    server();
  })

  proc.on('disconnect', () => {
Eva Ho's avatar
Eva Ho committed
76
    logger.info('Server disconnected. Reconnecting...');
77
78
79
    server();
  })

Jeffrey Morgan's avatar
Jeffrey Morgan committed
80
81
82
83
84
  process.on('exit', () => {
    proc.kill()
  })
}

85

86
function installCLI() {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
87
88
89
90
91
92
  const symlinkPath = '/usr/local/bin/ollama'

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

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

Eva Ho's avatar
Eva Ho committed
111
112
          logger.info(`CLI: ${stdout}}`)
          logger.error(`CLI: ${stderr}`)
113
114
115
116
117
        })
      }
    })
}

118
119
120
121
app.on('ready', () => {
  if (process.platform === 'darwin') {
    app.dock.hide()

Eva Ho's avatar
Eva Ho committed
122
123
124
    if (!store.has('first-time-run')) {
      // This is the first run
      app.setLoginItemSettings({ openAtLogin: true })
Jeffrey Morgan's avatar
Jeffrey Morgan committed
125
      store.set('first-time-run', false)
Eva Ho's avatar
Eva Ho committed
126
127
128
129
130
    } else {
      // The app has been run before
      app.setLoginItemSettings({ openAtLogin: app.getLoginItemSettings().openAtLogin })
    }

131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
    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
158
            logger.error(`[Move to Applications] Failed to move to applications folder - ${e.message}}`)
159
          }
160
161
        }
      }
162
163

      installCLI()
164
    }
165
  }
Jeffrey Morgan's avatar
Jeffrey Morgan committed
166

167
  createSystemtray()
168
  server()
169
})
Jeffrey Morgan's avatar
Jeffrey Morgan committed
170

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

Jeffrey Morgan's avatar
Jeffrey Morgan committed
186
187
188
189
async function heartbeat() {
  analytics.track({
    anonymousId: id(),
    event: 'heartbeat',
Jeffrey Morgan's avatar
Jeffrey Morgan committed
190
191
192
    properties: {
      version: app.getVersion(),
    },
Jeffrey Morgan's avatar
Jeffrey Morgan committed
193
194
195
  })
}

196
if (app.isPackaged) {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
197
  heartbeat()
198
  autoUpdater.checkForUpdates()
199
  setInterval(() => {
Jeffrey Morgan's avatar
Jeffrey Morgan committed
200
    heartbeat()
201
    autoUpdater.checkForUpdates()
Jeffrey Morgan's avatar
Jeffrey Morgan committed
202
  }, 60 * 60 * 1000)
203
}
Jeffrey Morgan's avatar
Jeffrey Morgan committed
204

205
autoUpdater.on('error', e => {
Eva Ho's avatar
Eva Ho committed
206
  logger.error(`auto updater: update check failed - ${e.message}`)
207
208
})

Jeffrey Morgan's avatar
Jeffrey Morgan committed
209
210
211
212
213
214
215
216
217
218
219
220
221
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()
    })
})