import fs from 'fs'
import mysql from 'mysql2/promise'

export async function update(field, list) {
  const { mysql: config } = JSON.parse(fs.readFileSync('config.json', 'utf8'))
  const conn = await mysql.createConnection(config)

  console.log('创建临时表')
  await conn.query(`CREATE TEMPORARY TABLE temp_updates (id INT PRIMARY KEY, ${mysql.escapeId(field)} INT UNSIGNED)`)

  console.log(`数据总量 ${list.length} 条, 每次 10000 条写入 ${Math.ceil(list.length / 10000)} 次`)
  for (let i = 0; i < list.length; i += 10000) {
    console.log(`第 ${i / 10000 + 1} 次写入`)
    const values = list.slice(i, i + 10000)
    await conn.query(`INSERT INTO temp_updates (id, ${field}) VALUES ?`, [values])
  }

  console.log('更新数据从临时表转入生产库')
  await conn.query(`UPDATE web_images t JOIN temp_updates tu ON t.id = tu.id SET t.${field} = tu.${field}`)
  await conn.end()
  console.log('更新完成')
}

export async function likes(day = 1) {
  const { mysql: config } = JSON.parse(fs.readFileSync('config.json', 'utf8'))
  const conn = await mysql.createConnection(config)
  const [rows] = await conn.query(`SELECT * FROM web_praise WHERE create_time >= CURDATE() AND create_time < CURDATE() + INTERVAL ${day} DAY`)
  await conn.end()
  return rows
}

export async function collects(day = 1) {
  const { mysql: config } = JSON.parse(fs.readFileSync('config.json', 'utf8'))
  const conn = await mysql.createConnection(config)
  const [rows] = await conn.query(`SELECT * FROM web_collect WHERE create_time >= CURDATE() AND create_time < CURDATE() + INTERVAL ${day} DAY`)
  await conn.end()
  return rows
}

export async function views(day = 1) {
  const startDate = new Date(Date.now() - day * 24 * 60 * 60 * 1000)
  const currentDate = new Date()

  const files = fs.readdirSync('./data').filter(name => /\d{4}-\d{2}-\d{2}\.json$/.test(name)).filter(name => {
    const date = new Date(name.match(/\d{4}-\d{2}-\d{2}/)[0])
    return date >= startDate && date <= currentDate
  })

  const result = { articles: new Map(), games: new Map(), screenshots: new Map() }

  files.forEach(name => {
    const data = JSON.parse(fs.readFileSync(`./data/${name}`, 'utf8'))
    for (const type of ['articles', 'games', 'screenshots']) {
      Object.entries(data[type] || {}).forEach(([key, value]) => {
        result[type].set(key, (result[type].get(key) || 0) + value)
      })
    }
  })

  return Object.fromEntries(
    Object.entries(result).map(([key, map]) => [
      key,
      [...map.entries()].sort((a, b) => b[1] - a[1]).map(([k]) => k)
    ])
  )
}

// 计算总权重
export function totalWeight(data) {
  const total = Object.values(data).reduce((acc, cur) => acc + cur, 0)
  return Object.fromEntries(Object.entries(data).map(([key, value]) => [key, Math.round(value / total * 100)]))
}