添加无法签到逻辑

app.ts

@@ -13,7 +13,10 @@ import {initUserCache, sleep} from './lib/tool'
 
   // 启动浏览器
   loading.start('启动无头浏览器')
-  const browser = await puppeteer.launch({defaultViewport: null})
+  const browser = await puppeteer.launch({
+    headless: process.env.NODE_ENV !== 'development',
+    defaultViewport: null
+  })
   const page = await browser.newPage()
   loading.stop() && consola.success('成功启动无头浏览器')
 

constants/selectors.ts

@@ -2,6 +2,7 @@
 export const usernameInput = '#ls_username'
 export const passwordInput = '#ls_password'
 export const loginButton = '#lsform > div > div.y.pns > table > tbody > tr:nth-child(3) > td:nth-child(2) > button'
+export const usernameSpan = '#um > p:nth-child(2) > strong'
 
 // 签到
 export const startCheckinAnchor = '#um > p:nth-child(3) > a:nth-child(3)'

logic/auth.ts

@@ -2,7 +2,7 @@ import * as fs from 'fs'
 import * as inquirer from 'inquirer'
 import consola from 'consola'
 import {Page} from 'puppeteer'
-import {loginButton, passwordInput, startCheckinAnchor, usernameInput} from '../constants/selectors'
+import {loginButton, passwordInput, usernameInput, usernameSpan} from '../constants/selectors'
 import {userPath} from '../constants/paths'
 import {userQuestion} from '../constants/questions'
 import {readJSONSync, sleep} from '../lib/tool'
@@ -30,7 +30,7 @@ export const login = async (page: Page, username: string, password: string) => {
 
   await sleep(3000)
   // 登录失败
-  if (await page.$(startCheckinAnchor) === null) {
+  if (await page.$(usernameSpan) === null) {
     loading.stop() && consola.error('登录失败')
 
     const newUser = await askUser()

logic/checkin.ts

@@ -1,8 +1,14 @@
 import {Page} from 'puppeteer'
 import {autoModeRadioInput, checkinBtn, startCheckinAnchor, 开心} from '../constants/selectors'
 import {sleep} from '../lib/tool'
+import loading from '../lib/loading'
+import consola from 'consola'
 
 const checkin = async (page: Page) => {
+  if (await page.$(startCheckinAnchor) === null) {
+    return loading.stop() && consola.error('无法签到')
+  }
+
   // 开始签到
   await page.click(startCheckinAnchor)
   await sleep(1000)