mirror of
https://github.com/elk-zone/elk.git
synced 2024-09-06 20:29:08 +01:00
201 lines
5.8 KiB
TypeScript
201 lines
5.8 KiB
TypeScript
import { login as loginMasto } from 'masto'
|
|
import type { Account, AccountCredentials, Instance, WsEvents } from 'masto'
|
|
import type { Ref } from 'vue'
|
|
import type { UserLogin } from '~/types'
|
|
import { DEFAULT_POST_CHARS_LIMIT, DEFAULT_SERVER, STORAGE_KEY_CURRENT_USER, STORAGE_KEY_SERVERS, STORAGE_KEY_USERS } from '~/constants'
|
|
|
|
const mock = process.mock
|
|
const users = useLocalStorage<UserLogin[]>(STORAGE_KEY_USERS, mock ? [mock.user] : [], { deep: true })
|
|
const servers = useLocalStorage<Record<string, Instance>>(STORAGE_KEY_SERVERS, mock ? mock.server : {}, { deep: true })
|
|
const currentUserId = useLocalStorage<string>(STORAGE_KEY_CURRENT_USER, mock ? mock.user.account.id : '')
|
|
|
|
export const currentUser = computed<UserLogin | undefined>(() => {
|
|
let user: UserLogin | undefined
|
|
if (currentUserId.value) {
|
|
user = users.value.find(user => user.account?.id === currentUserId.value)
|
|
if (user)
|
|
return user
|
|
}
|
|
// Fallback to the first account
|
|
return users.value[0]
|
|
})
|
|
|
|
export const currentUserHandle = computed(() => currentUser.value?.account.id
|
|
? `${currentUser.value.account.acct}@${currentUser.value.server}`
|
|
: '[anonymous]',
|
|
)
|
|
|
|
export const publicServer = ref(DEFAULT_SERVER)
|
|
const publicInstance = ref<Instance | null>(null)
|
|
export const currentServer = computed<string>(() => currentUser.value?.server || publicServer.value)
|
|
|
|
export const useUsers = () => users
|
|
|
|
export const currentInstance = computed<null | Instance>(() => currentUserId.value ? servers.value[currentUserId.value] ?? null : publicInstance.value)
|
|
|
|
export const characterLimit = computed(() => currentInstance.value?.configuration.statuses.maxCharacters ?? DEFAULT_POST_CHARS_LIMIT)
|
|
|
|
export async function loginTo(user?: Omit<UserLogin, 'account'> & { account?: AccountCredentials }) {
|
|
const config = useRuntimeConfig()
|
|
const route = useRoute()
|
|
const router = useRouter()
|
|
const server = user?.server || route.params.server as string || publicServer.value
|
|
const masto = await loginMasto({
|
|
url: `https://${server}`,
|
|
accessToken: user?.token,
|
|
disableVersionCheck: !!config.public.disableVersionCheck,
|
|
})
|
|
|
|
if (!user?.token) {
|
|
publicServer.value = server
|
|
publicInstance.value = await masto.instances.fetch()
|
|
}
|
|
|
|
else {
|
|
try {
|
|
const [me, server] = await Promise.all([
|
|
masto.accounts.verifyCredentials(),
|
|
masto.instances.fetch(),
|
|
])
|
|
|
|
user.account = me
|
|
currentUserId.value = me.id
|
|
servers.value[me.id] = server
|
|
|
|
if (!user.account.acct.includes('@'))
|
|
user.account.acct = `${user.account.acct}@${server.uri}`
|
|
|
|
if (!users.value.some(u => u.server === user.server && u.token === user.token))
|
|
users.value.push(user as UserLogin)
|
|
}
|
|
catch {
|
|
await signout()
|
|
}
|
|
}
|
|
|
|
setMasto(masto)
|
|
|
|
if ('server' in route.params && user?.token) {
|
|
await router.push({
|
|
...route,
|
|
force: true,
|
|
})
|
|
}
|
|
|
|
return masto
|
|
}
|
|
|
|
export async function signout() {
|
|
// TODO: confirm
|
|
if (!currentUser.value)
|
|
return
|
|
|
|
const _currentUserId = currentUser.value.account.id
|
|
|
|
const index = users.value.findIndex(u => u.account?.id === _currentUserId)
|
|
|
|
if (index !== -1) {
|
|
// Clear stale data
|
|
clearUserLocalStorage()
|
|
delete servers.value[_currentUserId]
|
|
|
|
currentUserId.value = ''
|
|
// Remove the current user from the users
|
|
users.value.splice(index, 1)
|
|
}
|
|
|
|
// Set currentUserId to next user if available
|
|
currentUserId.value = users.value[0]?.account?.id
|
|
|
|
if (!currentUserId.value)
|
|
await useRouter().push('/')
|
|
|
|
await loginTo(currentUser.value)
|
|
}
|
|
|
|
const notifications = reactive<Record<string, undefined | [Promise<WsEvents>, number]>>({})
|
|
|
|
export const useNotifications = () => {
|
|
const id = currentUser.value?.account.id
|
|
|
|
const clearNotifications = () => {
|
|
if (!id || !notifications[id])
|
|
return
|
|
notifications[id]![1] = 0
|
|
}
|
|
|
|
async function connect(): Promise<void> {
|
|
if (!id || notifications[id] || !currentUser.value?.token)
|
|
return
|
|
|
|
const masto = useMasto()
|
|
const stream = masto.stream.streamUser()
|
|
notifications[id] = [stream, 0]
|
|
;(await stream).on('notification', () => {
|
|
if (notifications[id])
|
|
notifications[id]![1]++
|
|
})
|
|
}
|
|
|
|
function disconnect(): void {
|
|
if (!id || !notifications[id])
|
|
return
|
|
notifications[id]![0].then(stream => stream.disconnect())
|
|
notifications[id] = undefined
|
|
}
|
|
|
|
watch(currentUser, disconnect)
|
|
connect()
|
|
|
|
return {
|
|
notifications: computed(() => id ? notifications[id]?.[1] ?? 0 : 0),
|
|
disconnect,
|
|
clearNotifications,
|
|
}
|
|
}
|
|
|
|
export function checkLogin() {
|
|
if (!currentUser.value) {
|
|
openSigninDialog()
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
/**
|
|
* Create reactive storage for the current user
|
|
*/
|
|
export function useUserLocalStorage<T extends object>(key: string, initial: () => T) {
|
|
// @ts-expect-error bind value to the function
|
|
const storages = useUserLocalStorage._ = useUserLocalStorage._ || new Map<string, Ref<Record<string, any>>>()
|
|
|
|
if (!storages.has(key))
|
|
storages.set(key, useLocalStorage(key, {}, { deep: true }))
|
|
const all = storages.get(key) as Ref<Record<string, T>>
|
|
|
|
return computed(() => {
|
|
const id = currentUser.value?.account.id
|
|
? `${currentUser.value.account.acct}@${currentUser.value.server}`
|
|
: '[anonymous]'
|
|
all.value[id] = Object.assign(initial(), all.value[id] || {})
|
|
return all.value[id]
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Clear all storages for the given account
|
|
*/
|
|
export function clearUserLocalStorage(account?: Account) {
|
|
if (!account)
|
|
account = currentUser.value?.account
|
|
if (!account)
|
|
return
|
|
|
|
const id = `${account.acct}@${currentUser.value?.server}`
|
|
// @ts-expect-error bind value to the function
|
|
;(useUserLocalStorage._ as Map<string, Ref<Record<string, any>>>).forEach((storage) => {
|
|
if (storage.value[id])
|
|
delete storage.value[id]
|
|
})
|
|
}
|