const mysql = require('mysql2/promise')
require('dotenv').config()

async function main() {
  const config = {
    host: process.env.DB_HOST || 'localhost',
    port: parseInt(process.env.DB_PORT || '3306'),
    user: process.env.DB_USER || 'root',
    password: process.env.DB_PASSWORD || '',
    database: process.env.DB_NAME || 'miniprogram'
  }

  const connection = await mysql.createConnection(config)
  
  try {
    const [columns] = await connection.execute("DESCRIBE goods")
    const columnNames = columns.map(col => col.Field)
    
    console.log('当前 goods 表字段:', columnNames)
    
    if (!columnNames.includes('is_on_sale')) {
      console.log('缺少 is_on_sale 字段，正在添加...')
      await connection.execute('ALTER TABLE goods ADD COLUMN is_on_sale tinyint(1) DEFAULT 1 COMMENT "是否上架 1-上架 0-下架"')
      console.log('成功添加 is_on_sale 字段')
    } else {
      console.log('is_on_sale 字段已存在')
    }
  } finally {
    await connection.end()
  }
}

main().catch(console.error)