elk/composables/users.ts

162 lines
4.7 KiB
TypeScript
Raw Normal View History

2022-11-22 23:08:36 +00:00
import { login as loginMasto } from 'masto'
import type { AccountCredentials, Instance, WsEvents } from 'masto'
2022-11-26 19:33:36 +00:00
import { clearUserDrafts } from './statusDrafts'
2022-11-22 23:08:36 +00:00
import type { UserLogin } from '~/types'
2022-11-25 14:21:07 +00:00
import { DEFAULT_POST_CHARS_LIMIT, DEFAULT_SERVER, STORAGE_KEY_CURRENT_USER, STORAGE_KEY_SERVERS, STORAGE_KEY_USERS } from '~/constants'
2022-11-22 23:08:36 +00:00
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 : '')
2022-11-22 23:08:36 +00:00
export const currentUser = computed<UserLogin | undefined>(() => {
let user: UserLogin | undefined
2022-11-23 04:25:48 +00:00
if (currentUserId.value) {
user = users.value.find(user => user.account?.id === currentUserId.value)
2022-11-22 23:08:36 +00:00
if (user)
return user
}
// Fallback to the first account
2022-11-23 04:25:48 +00:00
return users.value[0]
2022-11-22 23:08:36 +00:00
})
2022-11-29 20:51:52 +00:00
export const publicServer = ref(DEFAULT_SERVER)
const publicInstance = ref<Instance | null>(null)
2022-11-29 20:51:52 +00:00
export const currentServer = computed<string>(() => currentUser.value?.server || publicServer.value)
2022-11-22 23:08:36 +00:00
2022-11-23 04:25:48 +00:00
export const useUsers = () => users
2022-11-23 03:48:01 +00:00
export const currentInstance = computed<null | Instance>(() => currentUserId.value ? servers.value[currentUserId.value] ?? null : publicInstance.value)
2022-11-25 14:21:07 +00:00
export const characterLimit = computed(() => currentInstance.value?.configuration.statuses.maxCharacters ?? DEFAULT_POST_CHARS_LIMIT)
export async function loginTo(user?: Omit<UserLogin, 'account'> & { account?: AccountCredentials }) {
if (user) {
const existing = users.value.find(u => u.server === user.server && u.token === user.token)
if (existing && currentUserId.value !== user.account?.id)
currentUserId.value = user.account?.id
2022-11-22 23:08:36 +00:00
}
2022-11-30 21:28:55 +00:00
const config = useRuntimeConfig()
const route = useRoute()
const router = useRouter()
const server = user?.server || route.params.server as string || publicServer.value
2022-11-22 23:08:36 +00:00
const masto = await loginMasto({
url: `https://${server}`,
accessToken: user?.token,
2022-11-30 21:28:55 +00:00
disableVersionCheck: !!config.public.disableVersionCheck,
2022-11-22 23:08:36 +00:00
})
2022-11-29 20:51:52 +00:00
if (!user?.token) {
publicServer.value = server
publicInstance.value = await masto.instances.fetch()
2022-11-29 20:51:52 +00:00
}
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) {
await router.push({
...route,
force: true,
})
}
return masto
2022-11-22 23:08:36 +00:00
}
2022-11-23 04:20:59 +00:00
export async function signout() {
// TODO: confirm
if (!currentUser.value)
return
2022-11-26 19:33:36 +00:00
const _currentUserId = currentUser.value.account.id
const index = users.value.findIndex(u => u.account?.id === _currentUserId)
if (index !== -1) {
// Clear stale data
delete servers.value[_currentUserId]
clearUserDrafts()
2022-11-28 22:57:27 +00:00
clearUserFeatureFlags()
2022-11-23 04:20:59 +00:00
2022-11-29 05:43:06 +00:00
currentUserId.value = ''
2022-11-26 19:33:36 +00:00
// Remove the current user from the users
users.value.splice(index, 1)
}
// Set currentUserId to next user if available
2022-11-23 04:25:48 +00:00
currentUserId.value = users.value[0]?.account?.id
2022-11-26 19:33:36 +00:00
if (!currentUserId.value)
await useRouter().push(`/${currentServer.value}/public`)
2022-11-23 04:20:59 +00:00
await loginTo(currentUser.value)
2022-11-23 04:20:59 +00:00
}
2022-12-02 02:18:57 +00:00
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])
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 }
}
2022-12-02 02:18:57 +00:00
export function checkLogin() {
if (!currentUser.value) {
openSigninDialog()
return false
}
return true
}