From 313cafa23ce58625e76518a9077d8075a6631ce9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E5=92=B2=E6=99=BA=E5=AD=90?= <sxzz@sxzz.moe>
Date: Tue, 3 Jan 2023 17:39:00 +0800
Subject: [PATCH] feat: wip

---
 app.vue                                       |   4 +-
 components/account/AccountGuest.vue           |  18 +++
 components/account/AccountMoreButton.vue      |   2 +-
 components/nav/NavUser.vue                    |   4 +-
 components/timeline/TimelinePinned.vue        |   2 +-
 components/user/UserPicker.vue                |   4 +-
 components/user/UserSignIn.vue                |   2 +-
 components/user/UserSwitcher.vue              |   4 +-
 composables/command.ts                        |  10 +-
 .../createPushSubscription.ts                 |   4 +-
 .../push-notifications/usePushManager.ts      |   6 +
 composables/statusDrafts.ts                   |   2 +-
 composables/users.ts                          | 151 +++++++++++-------
 error.vue                                     |   2 +-
 layouts/default.vue                           |   5 +-
 middleware/permalink.global.ts                |  48 +-----
 pages/settings/users/index.vue                |   4 +-
 types/index.ts                                |   3 +-
 18 files changed, 146 insertions(+), 129 deletions(-)
 create mode 100644 components/account/AccountGuest.vue

diff --git a/app.vue b/app.vue
index ae3387eb..1465984c 100644
--- a/app.vue
+++ b/app.vue
@@ -3,9 +3,7 @@ setupPageHeader()
 provideGlobalCommands()
 
 // We want to trigger rerendering the page when account changes
-const key = computed(() =>
-  `${currentServer.value}:${checkUser(currentUser.value) ? currentUser.value.account.id : GUEST_ID}`,
-)
+const key = computed(() => currentUser.value ? getUniqueUserId(currentUser.value) : '')
 </script>
 
 <template>
diff --git a/components/account/AccountGuest.vue b/components/account/AccountGuest.vue
new file mode 100644
index 00000000..f29dc4b2
--- /dev/null
+++ b/components/account/AccountGuest.vue
@@ -0,0 +1,18 @@
+<script setup lang="ts">
+import type { UserLogin } from '~/types'
+
+defineProps<{
+  user: UserLogin
+}>()
+</script>
+
+<template>
+  <div flex="~ gap3" items-center>
+    <div bg="gray/40" rounded-full w-54px h-54px flex shrink-0 items-center justify-center text-5>
+      G
+    </div>
+    <div>
+      Guest from <span font-bold>{{ user.server }}</span>
+    </div>
+  </div>
+</template>
diff --git a/components/account/AccountMoreButton.vue b/components/account/AccountMoreButton.vue
index bd85aa16..5d158cce 100644
--- a/components/account/AccountMoreButton.vue
+++ b/components/account/AccountMoreButton.vue
@@ -7,7 +7,7 @@ const { account } = defineProps<{
 }>()
 let relationship = $(useRelationship(account))
 
-const isSelf = $computed(() => currentUser.value?.account.id === account.id)
+const isSelf = $computed(() => checkUser(currentUser.value) && currentUser.value.account.id === account.id)
 
 const masto = useMasto()
 const toggleMute = async () => {
diff --git a/components/nav/NavUser.vue b/components/nav/NavUser.vue
index 6e0d403e..76353153 100644
--- a/components/nav/NavUser.vue
+++ b/components/nav/NavUser.vue
@@ -9,8 +9,8 @@
         w-8
         :draggable="false"
       />
-      <div v-else>
-        TODO: Guest
+      <div v-else bg="gray/40" rounded-full w-8 h-8 flex items-center justify-center text-5>
+        G
       </div>
     </div>
 
diff --git a/components/timeline/TimelinePinned.vue b/components/timeline/TimelinePinned.vue
index e0862f01..705542a8 100644
--- a/components/timeline/TimelinePinned.vue
+++ b/components/timeline/TimelinePinned.vue
@@ -1,5 +1,5 @@
 <script setup lang="ts">
-const paginator = useMasto().accounts.iterateStatuses(currentUser.value!.account.id, { pinned: true })
+const paginator = useMasto().accounts.iterateStatuses(currentUser.value!.account!.id, { pinned: true })
 </script>
 
 <template>
diff --git a/components/user/UserPicker.vue b/components/user/UserPicker.vue
index 9d968e9b..40d200d4 100644
--- a/components/user/UserPicker.vue
+++ b/components/user/UserPicker.vue
@@ -16,7 +16,9 @@ const masto = useMasto()
           @click="switchUser(user, masto)"
         >
           <AccountAvatar v-if="!user.guest" w-13 h-13 :account="user.account" />
-          <span v-else>TODO: Guest from {{ user.server }}</span>
+          <div v-else bg="gray/40" rounded-full w-13 h-13 flex shrink-0 items-center justify-center text-5>
+            G
+          </div>
         </button>
       </template>
     </div>
diff --git a/components/user/UserSignIn.vue b/components/user/UserSignIn.vue
index c42b7594..5798c8a5 100644
--- a/components/user/UserSignIn.vue
+++ b/components/user/UserSignIn.vue
@@ -48,7 +48,7 @@ async function oauth() {
 }
 
 function explore() {
-  masto.loginTo({ server, guest: true })
+  masto.loginTo({ server, guestOnly: true })
 }
 
 async function handleInput() {
diff --git a/components/user/UserSwitcher.vue b/components/user/UserSwitcher.vue
index fccd5242..35c4fbf5 100644
--- a/components/user/UserSwitcher.vue
+++ b/components/user/UserSwitcher.vue
@@ -22,7 +22,7 @@ const masto = useMasto()
         @click="switchUser(user, masto)"
       >
         <AccountInfo v-if="!user.guest" :account="user.account" :hover-card="false" />
-        <span v-else>TODO: Guest from {{ user.server }}</span>
+        <AccountGuest v-else :user="user" />
         <div flex-auto />
         <div v-if="isSameUser(user, currentUser)" i-ri:check-line text-primary mya text-2xl />
       </button>
@@ -34,7 +34,7 @@ const masto = useMasto()
         @click="openSigninDialog"
       />
       <CommonDropdownItem
-        v-if="isMastoInitialised && currentUser"
+        v-if="isMastoInitialised && canSignOut"
         :text="$t('user.sign_out_account', [getFullHandle(currentUser)])"
         icon="i-ri:logout-box-line rtl-flip"
         @click="signout"
diff --git a/composables/command.ts b/composables/command.ts
index 4f03fbb2..fc71f325 100644
--- a/composables/command.ts
+++ b/composables/command.ts
@@ -248,7 +248,7 @@ export const provideGlobalCommands = () => {
   useCommand({
     scope: 'Actions',
 
-    visible: () => currentUser.value,
+    visible: () => !isGuest.value,
 
     name: () => t('action.compose'),
     icon: 'i-ri:quill-pen-line',
@@ -344,9 +344,9 @@ export const provideGlobalCommands = () => {
     parent: 'account-switch',
     scope: 'Switch account',
 
-    visible: () => !user.guest && user.account.id !== currentUser.value?.account?.id,
+    visible: () => !isSameUser(user, currentUser.value),
 
-    name: () => t('command.switch_account', [getFullHandle(user)]),
+    name: () => t('command.switch_account', [user.guest ? `guest from ${user.server}` : getFullHandle(user)]),
     icon: 'i-ri:user-shared-line',
 
     onActivate() {
@@ -356,8 +356,8 @@ export const provideGlobalCommands = () => {
   useCommand({
     scope: 'Account',
 
-    visible: () => currentUser.value,
-    name: () => currentUser.value ? t('user.sign_out_account', [getFullHandle(currentUser.value)]) : '',
+    visible: () => canSignOut.value,
+    name: () => t('user.sign_out_account', [getFullHandle(currentUser.value)]),
     icon: 'i-ri:logout-box-line',
 
     onActivate() {
diff --git a/composables/push-notifications/createPushSubscription.ts b/composables/push-notifications/createPushSubscription.ts
index 30599ab0..30ca1774 100644
--- a/composables/push-notifications/createPushSubscription.ts
+++ b/composables/push-notifications/createPushSubscription.ts
@@ -97,7 +97,7 @@ async function subscribe(
 
 async function unsubscribeFromBackend(fromSWPushManager: boolean, removePushNotification = true) {
   const cu = currentUser.value
-  if (cu) {
+  if (checkUser(cu)) {
     await removePushNotifications(cu)
     removePushNotification && await removePushNotificationData(cu, fromSWPushManager)
   }
@@ -105,7 +105,7 @@ async function unsubscribeFromBackend(fromSWPushManager: boolean, removePushNoti
 
 async function removePushNotificationDataOnError(e: Error) {
   const cu = currentUser.value
-  if (cu)
+  if (checkUser(cu))
     await removePushNotificationData(cu, true)
 
   throw e
diff --git a/composables/push-notifications/usePushManager.ts b/composables/push-notifications/usePushManager.ts
index 62f5b879..a7ac0134 100644
--- a/composables/push-notifications/usePushManager.ts
+++ b/composables/push-notifications/usePushManager.ts
@@ -117,6 +117,9 @@ export const usePushManager = () => {
   }
 
   const saveSettings = async (policy?: SubscriptionPolicy) => {
+    if (!checkUser(currentUser.value))
+      return
+
     if (policy)
       pushNotificationData.value.policy = policy
 
@@ -133,6 +136,9 @@ export const usePushManager = () => {
   }
 
   const undoChanges = () => {
+    if (!checkUser(currentUser.value))
+      return
+
     const current = pushNotificationData.value
     const previous = history.value[0].snapshot
     current.favourite = previous.favourite
diff --git a/composables/statusDrafts.ts b/composables/statusDrafts.ts
index 6c0e040f..f7e4508a 100644
--- a/composables/statusDrafts.ts
+++ b/composables/statusDrafts.ts
@@ -51,7 +51,7 @@ function mentionHTML(acct: string) {
 
 export function getReplyDraft(status: Status) {
   const accountsToMention: string[] = []
-  const userId = currentUser.value?.account.id
+  const userId = currentUser.value!.account!.id
   if (status.account.id !== userId)
     accountsToMention.push(status.account.acct)
   accountsToMention.push(...(status.mentions.filter(mention => mention.id !== userId).map(mention => mention.acct)))
diff --git a/composables/users.ts b/composables/users.ts
index 883f01c8..902a0c3d 100644
--- a/composables/users.ts
+++ b/composables/users.ts
@@ -45,107 +45,133 @@ const users = await initializeUsers()
 const instances = 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 : '')
 const isGuestId = computed(() => !currentUserId.value || currentUserId.value.startsWith(`${GUEST_ID}@`))
+const defaultUser: UserLogin<false> = {
+  server: DEFAULT_SERVER,
+  guest: true,
+}
 
-export const currentUser = computed<UserLogin | undefined>(() => {
-  if (!currentUserId.value)
-    // Fallback to the first account
-    return users.value[0]
-
-  if (isGuestId.value) {
-    const server = currentUserId.value.replace(`${GUEST_ID}@`, '')
-    return users.value.find(user => user.guest && user.server === server)
+export const currentUser = computed<UserLogin>(() => {
+  let user: UserLogin | undefined
+  if (!currentUserId.value) {
+  // Fallback to the first account
+    user = users.value[0]
   }
-
-  return users.value.find(user => user.account?.id === currentUserId.value)
+  else if (isGuestId.value) {
+    const server = currentUserId.value.replace(`${GUEST_ID}@`, '')
+    user = users.value.find(user => user.guest && user.server === server)
+  }
+  else {
+    user = users.value.find(user => user.account?.id === currentUserId.value)
+  }
+  return user || defaultUser
 })
 
-export const currentServer = computed<string>(() => currentUser.value?.server || DEFAULT_SERVER)
+export const currentServer = computed<string>(() => currentUser.value.server)
 export const currentInstance = computed<null | Instance>(() => {
   return instances.value[currentServer.value] ?? null
 })
 export const checkUser = (val: UserLogin | undefined): val is UserLogin<true> => !!(val && !val.guest)
 export const isGuest = computed(() => !checkUser(currentUser.value))
-export const getUniqueUserId = (user: UserLogin) => user.guest ? `${GUEST_ID}@${user.server}` : user.account.id
+export const getUniqueUserId = (user: UserLogin) =>
+  user.guest ? `${GUEST_ID}@${user.server}` : user.account.id
 export const isSameUser = (a: UserLogin | undefined, b: UserLogin | undefined) =>
   a && b && getUniqueUserId(a) === getUniqueUserId(b)
 
 export const currentUserHandle = computed(() =>
-  isGuestId.value ? GUEST_ID : currentUser.value!.account!.acct
-  ,
+  currentUser.value.guest ? GUEST_ID : currentUser.value.account!.acct,
 )
 
 export const useUsers = () => users
 
 export const characterLimit = computed(() => currentInstance.value?.configuration.statuses.maxCharacters ?? DEFAULT_POST_CHARS_LIMIT)
 
-async function loginTo(user?: UserLogin) {
+async function loginTo({ server, token, vapidKey, pushSubscription, guest = false }: { guest?: boolean } & Omit<UserLogin, 'guest'>) {
   const route = useRoute()
   const router = useRouter()
-  const server = user?.server || (route.params.server as string) || DEFAULT_SERVER
+
+  let user: UserLogin | undefined = token
+    ? users.value.find(u => u.server === server && u.token === token)
+    : ((guest
+        ? undefined
+        : users.value.find(u => u.server === server && u.token))
+      || users.value.find(u => u.server === server && u.guest))
+
+  const needPush = !user
+  if (!user) {
+    if (token) {
+      user = {
+        server,
+        guest: false,
+        token,
+        vapidKey,
+        pushSubscription,
+        account: undefined as any, // to be assigned later
+      }
+    }
+    else {
+      user = { server, guest: true }
+    }
+  }
+
   const masto = await loginMasto({
-    url: `https://${server}`,
-    accessToken: user?.token,
+    url: `https://${user.server}`,
+    accessToken: user.token,
     disableVersionCheck: true,
     // Suppress warning of `masto/fetch` usage
     disableExperimentalWarning: true,
   })
 
-  if (!user?.token) {
+  if (user.guest) {
     const instance = await masto.instances.fetch()
-
     instances.value[server] = instance
-    if (!users.value.some(u => u.server === server && u.guest))
-      users.value.push({ server, guest: true })
-
-    currentUserId.value = `${GUEST_ID}@${server}`
   }
-
   else {
-    try {
-      const [me, instance, pushSubscription] = await Promise.all([
-        masto.accounts.verifyCredentials(),
-        masto.instances.fetch(),
-        // if PWA is not enabled, don't get push subscription
-        useRuntimeConfig().public.pwaEnabled
-          // we get 404 response instead empty data
-          ? masto.pushSubscriptions.fetch().catch(() => Promise.resolve(undefined))
-          : Promise.resolve(undefined),
-      ])
+    const [me, instance, pushSubscription] = await Promise.all([
+      masto.accounts.verifyCredentials(),
+      masto.instances.fetch(),
+      // if PWA is not enabled, don't get push subscription
+      useRuntimeConfig().public.pwaEnabled
+      // we get 404 response instead empty data
+        ? masto.pushSubscriptions.fetch().catch(() => Promise.resolve(undefined))
+        : Promise.resolve(undefined),
+    ])
 
-      if (!me.acct.includes('@'))
-        me.acct = `${me.acct}@${instance.uri}`
+    if (!me.acct.includes('@'))
+      me.acct = `${me.acct}@${instance.uri}`
 
-      user.account = me
-      user.pushSubscription = pushSubscription
-      currentUserId.value = me.id
-      instances.value[server] = instance
-
-      if (!users.value.some(u => u.server === user.server && u.token === user.token))
-        users.value.push(user)
-    }
-    catch {
-      await signout()
-    }
+    user.account = me
+    user.pushSubscription = pushSubscription
+    instances.value[server] = instance
   }
 
+  if (needPush)
+    users.value.push(user)
+
+  currentUserId.value = getUniqueUserId(user)
+
   // This only cleans up the URL; page content should stay the same
   if (route.path === '/signin/callback') {
     await router.push('/home')
   }
 
-  else if ('server' in route.params && user?.token && !useNuxtApp()._processingMiddleware) {
+  else if ('server' in route.params && user.server !== route.params.server) {
     await router.push({
       ...route,
+      params: {
+        ...route.params,
+        server: user.server,
+      },
       force: true,
     })
   }
 
   return masto
 }
+export type LoginTo = typeof loginTo
 
 export const switchUser = (user: UserLogin, masto: ElkMasto) => {
   const router = useRouter()
-  if (!user.guest && !isGuest.value && user.account.id === currentUser.value!.account!.id)
+  if (!user.guest && !isGuest.value && user.account.id === currentUser.value.account!.id)
     router.push(getAccountRoute(user.account))
   else
     masto.loginTo(user)
@@ -173,7 +199,7 @@ export function getUsersIndexByUserId(userId: string) {
   return users.value.findIndex(u => u.account?.id === userId)
 }
 
-export async function removePushNotificationData(user: UserLogin, fromSWPushManager = true) {
+export async function removePushNotificationData(user: UserLogin<true>, fromSWPushManager = true) {
   // clear push subscription
   user.pushSubscription = undefined
   const { acct } = user.account!
@@ -212,13 +238,18 @@ export async function removePushNotifications(user: UserLogin<true>) {
   }
 }
 
+// do not sign out if there is only one guest user
+export const canSignOut = computed(() =>
+  users.value.length > 1 || !users.value[0].guest,
+)
+
 export async function signout() {
   // TODO: confirm
-  if (!currentUser.value)
+
+  if (!canSignOut.value)
     return
 
   const index = users.value.findIndex(u => isSameUser(u, currentUser.value))
-
   if (index !== -1) {
     // Clear stale data
     clearUserLocalStorage()
@@ -235,11 +266,8 @@ export async function signout() {
     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('/')
+  // Set currentUserId to next user
+  currentUserId.value = getUniqueUserId(users.value[0] ? users.value[0] : defaultUser)
 
   const masto = useMasto()
   await masto.loginTo(currentUser.value)
@@ -306,7 +334,7 @@ export function useUserLocalStorage<T extends object>(key: string, initial: () =
   const all = storages.get(key) as Ref<Record<string, T>>
 
   return computed(() => {
-    const id = isGuestId.value
+    const id = currentUser.value.guest
       ? GUEST_ID
       : currentUser.value!.account!.acct
     all.value[id] = Object.assign(initial(), all.value[id] || {})
@@ -343,10 +371,11 @@ export const createMasto = () => {
 
       if (key === 'loginTo') {
         return (...args: any[]): Promise<MastoClient> => {
-          return apiPromise.value = loginTo(...args).then((r) => {
+          return apiPromise.value = (loginTo as any)(...args).then((r: any) => {
             api.value = r
             return masto
-          }).catch(() => {
+          }).catch((err: any) => {
+            console.error(err)
             // Show error page when Mastodon server is down
             throw createError({
               fatal: true,
diff --git a/error.vue b/error.vue
index a7e3d23c..d01ff58a 100644
--- a/error.vue
+++ b/error.vue
@@ -22,7 +22,7 @@ const reload = async () => {
   try {
     if (!masto.loggedIn.value)
       await masto.loginTo(currentUser.value)
-    clearError({ redirect: currentUser.value ? '/home' : `/${currentServer.value}/public` })
+    clearError({ redirect: !isGuest.value ? '/home' : `/${currentServer.value}/public` })
   }
   catch {
     state.value = 'error'
diff --git a/layouts/default.vue b/layouts/default.vue
index 89d62f58..43d1d4b3 100644
--- a/layouts/default.vue
+++ b/layouts/default.vue
@@ -32,9 +32,8 @@ const wideLayout = computed(() => route.meta.wideLayout ?? false)
                       >
                         <AccountInfo :account="currentUser.account" md:break-words />
                       </NuxtLink>
-                      <div v-else>
-                        TODO: guest {{ currentUser.server }} @ default.vue
-                      </div>
+                      <AccountGuest v-else :user="currentUser" />
+
                       <UserDropdown />
                     </div>
                   </div>
diff --git a/middleware/permalink.global.ts b/middleware/permalink.global.ts
index f636156a..9e09a131 100644
--- a/middleware/permalink.global.ts
+++ b/middleware/permalink.global.ts
@@ -1,4 +1,4 @@
-export default defineNuxtRouteMiddleware(async (to, from) => {
+export default defineNuxtRouteMiddleware(async (to) => {
   if (process.server)
     return
 
@@ -13,23 +13,6 @@ export default defineNuxtRouteMiddleware(async (to, from) => {
 
   const user = currentUser.value
 
-  if (!user) {
-    if (from.params.server !== to.params.server) {
-      await masto.loginTo({
-        server: to.params.server as string,
-      })
-    }
-    return
-  }
-
-  // No need to additionally resolve an id if we're already logged in
-  if (user.server === to.params.server)
-    return
-
-  // Tags don't need to be redirected to a local id
-  if (to.params.tag)
-    return
-
   // Handle redirecting to new permalink structure for users with old links
   if (!to.params.server) {
     return {
@@ -41,28 +24,11 @@ export default defineNuxtRouteMiddleware(async (to, from) => {
     }
   }
 
-  try {
-    // If we're already on an account page, we can search for this on the new instance
-    if (to.params.account && to.name !== 'status' && to.params.account.includes('@')) {
-      const account = await fetchAccountByHandle(to.params.account as string)
-      if (account)
-        return getAccountRoute(account)
-    }
+  // No need to additionally resolve an id if we're already logged in
+  if (user.server === to.params.server)
+    return
 
-    if (!masto.loggedIn.value)
-      await masto.loginTo(currentUser.value)
-
-    // If we're logged in, search for the local id the account or status corresponds to
-    const { value } = await masto.search({ q: `https:/${to.fullPath}`, resolve: true, limit: 1 }).next()
-
-    const { accounts, statuses } = value
-    if (statuses[0])
-      return getStatusRoute(statuses[0])
-
-    if (accounts[0])
-      return getAccountRoute(accounts[0])
-  }
-  catch {}
-
-  return '/home'
+  masto.loginTo({
+    server: to.params.server as string,
+  })
 })
diff --git a/pages/settings/users/index.vue b/pages/settings/users/index.vue
index c896bf7c..1d06f3a6 100644
--- a/pages/settings/users/index.vue
+++ b/pages/settings/users/index.vue
@@ -71,9 +71,7 @@ async function importTokens() {
         <div flex="~ col gap2">
           <div v-for="user of loggedInUsers" :key="getUniqueUserId(user)">
             <AccountInfo v-if="!user.guest" :account="user.account" :hover-card="false" />
-            <div v-else>
-              TODO: Guest @ settings/users/index.vue
-            </div>
+            <AccountGuest v-else :user="user" />
           </div>
         </div>
         <div my4 border="t base" />
diff --git a/types/index.ts b/types/index.ts
index 650db96b..6a68e2a8 100644
--- a/types/index.ts
+++ b/types/index.ts
@@ -1,6 +1,7 @@
 import type { Account, AccountCredentials, Attachment, CreateStatusParams, Emoji, Instance, MastoClient, Notification, PushSubscription, Status } from 'masto'
 import type { Ref } from 'vue'
 import type { MarkNonNullable, Mutable } from './utils'
+import type { LoginTo } from '~/composables/users'
 
 export interface AppInfo {
   id: string
@@ -30,7 +31,7 @@ export type UserLogin<WithToken extends boolean = boolean> = {
 } & ((WithToken extends false ? UserLoginGuest : never) | (WithToken extends true ? UserLoginWithToken : never))
 
 export interface ElkMasto extends MastoClient {
-  loginTo(user?: UserLogin): Promise<MastoClient>
+  loginTo: LoginTo
   loggedIn: Ref<boolean>
 }