import { CSSProperties, useContext, useEffect, useRef, useState } from 'react' import { FaGear, FaPlus } from 'react-icons/fa6' import { Account, db } from '@/db' import NewAccount from '@/components/accounts/New' import Settings from '@/components/Settings' import { Avatar, Dropdown } from 'flowbite-react' import { useRouter } from 'next/router' import { FormattedMessage, useIntl } from 'react-intl' import generateNotification from '@/utils/notification' import generator, { Entity, WebSocketInterface } from 'megalodon' import { Context } from '@/utils/i18n' type LayoutProps = { children: React.ReactNode } export default function Layout({ children }: LayoutProps) { const [accounts, setAccounts] = useState>([]) const [openNewModal, setOpenNewModal] = useState(false) const [openSettings, setOpenSettings] = useState(false) const [style, setStyle] = useState({}) const { switchLang } = useContext(Context) const router = useRouter() const { formatMessage } = useIntl() const streamings = useRef>([]) useEffect(() => { loadSettings() const fn = async () => { const acct = await db.accounts.toArray() setAccounts(acct) if (acct.length === 0) { setOpenNewModal(true) } acct.forEach(async account => { // Start user streaming for notification const client = generator(account.sns, account.url, account.access_token, 'Whalebird') const instance = await client.getInstance() const ws = generator(account.sns, instance.data.urls.streaming_api, account.access_token, 'Whalebird') const socket = ws.userSocket() streamings.current = [...streamings.current, socket] socket.on('connect', () => { console.log(`connect to user streaming for ${account.domain}`) }) socket.on('notification', (notification: Entity.Notification) => { const [title, body] = generateNotification(notification, formatMessage) if (title.length > 0) { new window.Notification(title, { body: body }) } }) }) } fn() return () => { streamings.current.forEach(streaming => { streaming.removeAllListeners() streaming.stop() }) streamings.current = [] console.log('close user streamings') } }, []) const closeNewModal = async () => { const acct = await db.accounts.toArray() setAccounts(acct) setOpenNewModal(false) if (acct.length === 0) { setOpenNewModal(true) } else if (!router.query.id) { router.push(`/accounts/${acct[0].id}`) } } const openAccount = (id: number) => { router.push(`/accounts/${id}`) } const openContextMenu = (id: number) => { document.getElementById(`${id}`).click() } const dropdownTrigger = (accountId: number) => const removeAccount = async (id: number) => { await db.accounts.delete(id) const acct = await db.accounts.toArray() setAccounts(acct) if (acct.length === 0) { router.push('/') setOpenNewModal(true) } } const selectedClassName = (id: number) => { if (id === parseInt(router.query.id as string)) { return 'bg-blue-950 cursor-pointer' } else { return 'cursor-pointer' } } const loadSettings = () => { if (typeof localStorage !== 'undefined') { const lang = localStorage.getItem('language') switchLang(lang) const fontSize = localStorage.getItem('fontSize') if (parseInt(fontSize)) { setStyle({ fontSize: `${fontSize}px` }) } } } return (
{children} setOpenSettings(false)} reloadSettings={loadSettings} />
) }