index.ts 4 KB
Newer Older
Jeffrey Morgan's avatar
Jeffrey Morgan committed
1
import { app, BrowserWindow, autoUpdater, dialog } from 'electron'
2
import { spawn, exec } from 'child_process'
Bruce MacDonald's avatar
Bruce MacDonald committed
3
import * as path from 'path'
Jeffrey Morgan's avatar
Jeffrey Morgan committed
4
5
6

require('@electron/remote/main').initialize()

Jeffrey Morgan's avatar
Jeffrey Morgan committed
7
8
9
10
11
// This allows TypeScript to pick up the magic constants that's auto-generated by Forge's Webpack
// plugin that tells the Electron app where to look for the Webpack-bundled app code (depending on
// whether you're running in development or production).
declare const MAIN_WINDOW_WEBPACK_ENTRY: string
declare const MAIN_WINDOW_PRELOAD_WEBPACK_ENTRY: string
Jeffrey Morgan's avatar
Jeffrey Morgan committed
12
13
14
15
16
17

// Handle creating/removing shortcuts on Windows when installing/uninstalling.
if (require('electron-squirrel-startup')) {
  app.quit()
}

Jeffrey Morgan's avatar
Jeffrey Morgan committed
18
19
const createWindow = (): void => {
  // Create the browser window.
Jeffrey Morgan's avatar
Jeffrey Morgan committed
20
21
22
23
  // Create the browser window.
  const mainWindow = new BrowserWindow({
    width: 800,
    height: 600,
Jeffrey Morgan's avatar
Jeffrey Morgan committed
24
25
26
    minWidth: 400,
    minHeight: 300,
    titleBarStyle: 'hiddenInset',
Jeffrey Morgan's avatar
Jeffrey Morgan committed
27
28
29
30
31
    webPreferences: {
      preload: MAIN_WINDOW_PRELOAD_WEBPACK_ENTRY,
      nodeIntegration: true,
      contextIsolation: false,
    },
Jeffrey Morgan's avatar
Jeffrey Morgan committed
32
33
  })

Jeffrey Morgan's avatar
Jeffrey Morgan committed
34
35
  require('@electron/remote/main').enable(mainWindow.webContents)

Jeffrey Morgan's avatar
Jeffrey Morgan committed
36
37
38
39
  // and load the index.html of the app.
  mainWindow.loadURL(MAIN_WINDOW_WEBPACK_ENTRY)
}

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

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

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

59
60
61
62
63
64
65
66
67
68
69
function installCLI() {
  dialog
    .showMessageBox({
      type: 'info',
      title: 'Ollama CLI installation',
      message: 'To install the ollama CLI, we need your permission. You will be prompted to confirm.',
      buttons: ['OK'],
    })
    .then(result => {
      if (result.response === 0) {
        let command = `
Jeffrey Morgan's avatar
Jeffrey Morgan committed
70
        do shell script "ln -F -s ${ollama} /usr/local/bin/ollama" with administrator privileges
71
72
73
74
75
76
77
78
79
80
81
82
83
84
        `

        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}`)
        })
      }
    })
}

Jeffrey Morgan's avatar
Jeffrey Morgan committed
85
86
87
// This method will be called when Electron has finished
// initialization and is ready to create browser windows.
// Some APIs can only be used after this event occurs.
88
89
90
91
app.on('ready', () => {
  createWindow()
  installCLI()
})
Jeffrey Morgan's avatar
Jeffrey Morgan committed
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

// 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()
  }
})

app.on('activate', () => {
  // On OS X it's common to re-create a window in the app when the
  // dock icon is clicked and there are no other windows open.
  if (BrowserWindow.getAllWindows().length === 0) {
    createWindow()
  }
})

// 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
112
113
114
autoUpdater.setFeedURL({
  url: `https://ollama.ai/api/update?os=${process.platform}&arch=${process.arch}&version=${app.getVersion()}`,
})
115

116
autoUpdater.checkForUpdates()
117
118
119
setInterval(() => {
  autoUpdater.checkForUpdates()
}, 60000)
Jeffrey Morgan's avatar
Jeffrey Morgan committed
120

121
122
123
124
autoUpdater.on('error', e => {
  console.error('update check failed', e)
})

Jeffrey Morgan's avatar
Jeffrey Morgan committed
125
126
127
128
129
130
131
132
133
134
135
136
137
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()
    })
})