From 4f664dbfc3a86154827b75ec08b79375e73b461a Mon Sep 17 00:00:00 2001
From: MaysWind <i@mayswind.net>
Date: Mon, 6 Oct 2025 16:05:04 +0800
Subject: [PATCH] code refactor

---
 src/components/desktop/TwoColumnSelect.vue | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/components/desktop/TwoColumnSelect.vue b/src/components/desktop/TwoColumnSelect.vue
index 8c7bf4ca..3c1c8f49 100644
--- a/src/components/desktop/TwoColumnSelect.vue
+++ b/src/components/desktop/TwoColumnSelect.vue
@@ -1,6 +1,6 @@
 <template>
     <v-select
-        class="two-column-main-select"
+        ref="twoColumnMainSelect"
         persistent-placeholder
         :density="density"
         :variant="variant"
@@ -89,6 +89,7 @@
 </template>
 
 <script setup lang="ts">
+import { VSelect } from 'vuetify/components/VSelect';
 import { VTextField } from 'vuetify/components/VTextField';
 
 import { ref, computed, useTemplateRef, nextTick } from 'vue';
@@ -142,6 +143,7 @@ const {
     updateCurrentSecondaryValue
 } = useTwoColumnListItemSelectionBase(props);
 
+const twoColumnMainSelect = useTemplateRef<VSelect>('twoColumnMainSelect');
 const filterInput = useTemplateRef<VTextField>('filterInput');
 const dropdownMenu = useTemplateRef<HTMLElement>('dropdownMenu');
 
@@ -216,8 +218,8 @@ function isSecondarySelected(subItem: unknown): boolean {
 function updateMenuPosition(): void {
     if (props.autoUpdateMenuPosition) {
         nextTick(() => {
-            const mainSelectRect = document.querySelector('.two-column-main-select')?.getBoundingClientRect();
-            const selectMenu = document.querySelector('.two-column-select-menu') as (HTMLElement | null);
+            const mainSelectRect = twoColumnMainSelect.value?.$el.getBoundingClientRect();
+            const selectMenu = dropdownMenu.value?.parentElement?.parentElement;
             const selectMenuRect = selectMenu?.getBoundingClientRect();
 
             if (mainSelectRect && selectMenu && selectMenuRect) {