aboutsummaryrefslogtreecommitdiffstats
path: root/src/main/index.ts
blob: 76c508a9cca0925fc6d3a72bafb40956973bf479 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
import { app, BrowserWindow, shell, ipcMain } from 'electron'
import { join } from 'path'
import { electronApp, optimizer, is, platform } from '@electron-toolkit/utils'
import { WebSocketServer, WebSocket } from 'ws'
import icon from '../../resources/icon.png?asset'

let mainWindow: BrowserWindow
let store: import('electron-store').default<{ channelNicknames: Record<string, string> }>

async function initStore(): Promise<void> {
  const Store = (await import('electron-store')).default
  store = new Store({
    defaults: {
      channelNicknames: {}
    }
  })
}

function createWindow(): void {
  mainWindow = new BrowserWindow({
    width: 900,
    height: 670,
    show: false,
    title: 'tiny-discord-feed',
    autoHideMenuBar: true,
    ...(platform.isLinux ? { icon } : {}),
    webPreferences: {
      preload: join(__dirname, '../preload/index.js'),
      sandbox: false,
      contextIsolation: true
    }
  })

  mainWindow.on('ready-to-show', () => {
    mainWindow.show()
  })

  mainWindow.webContents.setWindowOpenHandler((details) => {
    shell.openExternal(details.url)
    return { action: 'deny' }
  })

  if (is.dev && process.env['ELECTRON_RENDERER_URL']) {
    mainWindow.loadURL(process.env['ELECTRON_RENDERER_URL'])
  } else {
    mainWindow.loadFile(join(__dirname, '../renderer/index.html'))
  }
}

function setupWebSocket(): void {
  if (!mainWindow) {
    console.error('Cannot setup WebSocket: mainWindow not available')
    return
  }

  const wss = new WebSocketServer({ port: 8765, host: '0.0.0.0' })

  wss.on('connection', (ws: WebSocket) => {
    console.log('New WebSocket client connected')
    ws.on('message', (msg: Buffer) => {
      try {
        const parsed = JSON.parse(msg.toString())
        if (mainWindow && !mainWindow.isDestroyed()) {
          mainWindow.webContents.send('new-discord-message', parsed)
        }
      } catch (e) {
        console.error('Invalid JSON received:', e)
      }
    })

    ws.on('error', (error) => {
      console.error('WebSocket client error:', error)
    })
  })

  wss.on('error', (error) => {
    console.error('WebSocket server error:', error)
  })

  console.log('WebSocket server listening on 0.0.0.0:8765')
}

function setupIpcHandlers(): void {
  // Get channel nicknames
  ipcMain.handle('config:get-channel-nicknames', () => {
    return store.get('channelNicknames')
  })

  // Set channel nickname
  ipcMain.handle('config:set-channel-nickname', (_, channelId: string, nickname: string) => {
    const nicknames = { ...store.get('channelNicknames') }
    nicknames[channelId] = nickname
    store.set('channelNicknames', nicknames)
    return true
  })

  // Remove channel nickname
  ipcMain.handle('config:remove-channel-nickname', (_, channelId: string) => {
    const nicknames = { ...store.get('channelNicknames') }
    delete nicknames[channelId]
    store.set('channelNicknames', nicknames)
    return true
  })

  // Get all channel nicknames as array
  ipcMain.handle('config:get-channel-list', () => {
    const nicknames = store.get('channelNicknames')
    return Object.entries(nicknames).map(([id, nickname]) => ({ id, nickname }))
  })
}

app.whenReady().then(async () => {
  electronApp.setAppUserModelId('com.electron')
  app.on('browser-window-created', (_, window) => {
    optimizer.watchWindowShortcuts(window)
  })

  await initStore()
  createWindow()
  setupWebSocket()
  setupIpcHandlers()

  app.on('activate', function () {
    if (BrowserWindow.getAllWindows().length === 0) createWindow()
  })
})

app.on('window-all-closed', () => {
  if (process.platform !== 'darwin') {
    app.quit()
  }
})
send patches to the email below
yukais@pinapelz.com
include the subject [PATCH repo_name]
pinapelz.com
homepage