import codecs
import re

with open('layouts/admin/components/AdminHeader.uvue', 'r', encoding='utf-8') as f:
    text = f.read()

new_html = r"""      <view class="hbtn" @click="$emit('notify')">\n        <text>🔔</text>\n        <view class="dot" v-if="hasNotification"></view>\n      </view>\n      \n      <!-- 用户个人中心 / 下拉菜单 -->\n      <view class="user-profile-container" @click="toggleUserMenu">\n        <text class="user-name">crmeb demo</text>\n        <text class="user-arrow">▼</text>\n        \n        <view class="user-dropdown" v-if="showUserMenu">\n          <view class="dropdown-item" @click.stop="goToUserCenter">个人中心</view>\n          <view class="dropdown-item" @click.stop="handleLogout">退出登录</view>\n        </view>\n      </view>"""

new_html = new_html.replace(r'\n', '\n')

text = re.sub(r'<view class=\"hbtn\" @click=\"\$emit\(\'notify\'\)\">[\s\S]*?<\/view>\s*<\/view>', new_html + '\n    </view>', text)

new_script = r"""import { ref, computed } from 'vue'\nimport {\n  toggleSubSider,\n  showSubSider,\n  layoutMode,\n  isOverlayVisible,\n  isMobileMenuOpen,\n  openRoute\n} from '@/layouts/admin/store/adminNavStore.uts'\n\nconst showUserMenu = ref(false)\n\nfunction toggleUserMenu() {\n  showUserMenu.value = !showUserMenu.value\n}\n\nfunction goToUserCenter() {\n  showUserMenu.value = false\n  openRoute('demo_user_center')\n}\n\nfunction handleLogout() {\n  showUserMenu.value = false\n  uni.showModal({\n    title: '提示',\n    content: '确定要退出登录吗？',\n    success: (res) => {\n      if (res.confirm) {\n        uni.reLaunch({\n          url: '/pages/index/index'\n        })\n      }\n    }\n  })\n}"""
new_script = new_script.replace(r'\n', '\n')

text = re.sub(r'import \{ computed \} from \'vue\'[\s\S]*?\} from \'@/layouts/admin/store/adminNavStore\.uts\'', new_script, text)


new_style = r""".dot{\n  width: 8px;\n  height: 8px;\n  border-radius: 50%;\n  background:#ff4d4f;\n  position:absolute;\n  top: 6px;\n  right: 6px;\n}\n\n.user-profile-container {\n  position: relative;\n  display: flex;\n  flex-direction: row;\n  align-items: center;\n  margin-left: 10px;\n  cursor: pointer;\n}\n\n.user-name {\n  font-size: 14px;\n  color: #333;\n}\n\n.user-arrow {\n  font-size: 12px;\n  color: #666;\n  margin-left: 4px;\n}\n\n.user-dropdown {\n  position: absolute;\n  top: 40px;\n  right: 0;\n  width: 120px;\n  background-color: #fff;\n  border-radius: 4px;\n  box-shadow: 0 2px 12px rgba(0,0,0,0.1);\n  z-index: 1000;\n  display: flex;\n  flex-direction: column;\n}\n\n.dropdown-item {\n  height: 40px;\n  line-height: 40px;\n  text-align: center;\n  font-size: 14px;\n  color: #333;\n  cursor: pointer;\n}\n\n.dropdown-item:hover {\n  background-color: #f5f7fa;\n  color: #1890ff;\n}"""
new_style = new_style.replace(r'\n', '\n')
text = re.sub(r'\.dot\{[\s\S]*?\}', new_style, text)

with open('layouts/admin/components/AdminHeader.uvue', 'w', encoding='utf-8') as f:
    f.write(text)