Files
rclone-ui/lib/tray.ts
T

120 lines
3.3 KiB
TypeScript

import * as Sentry from '@sentry/browser'
import { Menu } from '@tauri-apps/api/menu'
import { MenuItem } from '@tauri-apps/api/menu'
import { resolveResource } from '@tauri-apps/api/path'
import type { TrayIconEvent } from '@tauri-apps/api/tray'
import { TrayIcon } from '@tauri-apps/api/tray'
import { getAllWindows, getCurrentWindow } from '@tauri-apps/api/window'
import { ask } from '@tauri-apps/plugin-dialog'
import { platform } from '@tauri-apps/plugin-os'
import { buildMenu } from './menu'
import { resetMainWindow } from './window'
export async function triggerTrayRebuild() {
return getAllWindows().then((windows) => {
windows.find((w) => w.label === 'main')?.emit('rebuild-tray')
})
}
let interval: NodeJS.Timeout | null = null
async function getTray() {
return await TrayIcon.getById('main-tray')
}
export async function showLoadingTray() {
console.log('[showLoadingTray]')
if (platform() === 'linux') {
console.log('[showLoadingTray] platform is linux, skipping')
return
}
const tray = await getTray()
if (!tray) {
console.error('[showLoadingTray] tray not found')
return
}
const globeIconPath = await resolveResource('icons/favicon/frame_00_delay-0.1s.png')
const quitItem = await MenuItem.new({
id: 'quit-loading',
text: 'Quit',
action: async () => {
const answer = await ask('An operation is in progress, are you sure you want to exit?')
if (answer) {
await getCurrentWindow().emit('close-app')
}
},
})
const loadingMenu = await Menu.new({
id: 'loading-menu',
items: [quitItem],
})
await tray.setMenu(loadingMenu)
await tray.setIcon(globeIconPath)
await tray.setTooltip('Loading...')
let currentIcon = 1
interval = setInterval(async () => {
if (currentIcon > 17) {
currentIcon = 1
}
const globeIconPath = await resolveResource(
`icons/favicon/frame_${currentIcon < 10 ? '0' : ''}${currentIcon}_delay-0.1s.png`
)
await tray?.setIcon(globeIconPath)
currentIcon += 1
}, 200)
}
export async function showDefaultTray() {
console.log('[showDefaultTray]')
const tray = await getTray()
if (!tray) {
console.error('[showDefaultTray] tray not found')
return
}
if (interval) {
clearInterval(interval)
interval = null
}
const newMenu = await buildMenu()
await tray.setMenu(newMenu)
await tray.setIcon(await resolveResource('icons/favicon/icon.png'))
await tray.setTooltip('Rclone')
console.log('[showDefaultTray] tray menu rebuilt')
}
export async function initTray() {
try {
console.log('[initTray]')
await TrayIcon.new({
id: 'main-tray',
icon: (await resolveResource('icons/favicon/icon.png'))!,
tooltip: 'Rclone',
menuOnLeftClick: true,
action: async (event: TrayIconEvent) => {
if (event.type === 'Click') {
console.log('[onTrayAction] tray clicked:', event)
await resetMainWindow()
}
},
})
} catch (error) {
Sentry.captureException(error)
console.error('[initTray] failed to create tray')
console.error(error)
}
}