const cloud = require('wx-server-sdk')
cloud.init({ env: cloud.DYNAMIC_CURRENT_ENV })
const db = cloud.database()

// 根据生日计算年龄展示文案
function calcAge(birthday) {
  if (!birthday) return ''
  const dob = new Date(birthday).getTime()
  if (isNaN(dob)) return ''
  const diffMs = Date.now() - dob
  if (diffMs < 0) return ''
  const years = Math.floor(diffMs / (365.25 * 24 * 3600 * 1000))
  const months = Math.floor((diffMs % (365.25 * 24 * 3600 * 1000)) / (30.44 * 24 * 3600 * 1000))
  return years > 0 ? `${years}岁${months > 0 ? months + '个月' : ''}` : `${months}个月`
}

exports.main = async (event, context) => {
  const { OPENID } = cloud.getWXContext()
  const { pet } = event || {}

  if (!pet || !pet.name || !String(pet.name).trim()) {
    return { code: -1, message: '宠物信息不完整' }
  }

  try {
    const userRes = await db.collection('users').where({ openid: OPENID }).get()
    if (userRes.data.length === 0) {
      return { code: -1, message: '用户不存在' }
    }
    const user = userRes.data[0]
    const pets = Array.isArray(user.pets) ? [...user.pets] : []

    const petData = {
      name: String(pet.name).trim(),
      species: pet.species || 'dog',
      breed: pet.breed || '',
      gender: pet.gender || 'female',
      birthday: pet.birthday || '',
      age: calcAge(pet.birthday),
      bio: (pet.bio || '').trim(),
      tags: Array.isArray(pet.tags) ? pet.tags.slice(0, 8) : [],
      emoji: pet.emoji || '🐾',
      photos: pet.avatarUrl ? [pet.avatarUrl] : (Array.isArray(pet.photos) ? pet.photos : []),
      ownerId: OPENID,
    }

    let resultPet
    if (pet._id) {
      const idx = pets.findIndex(p => p._id === pet._id)
      if (idx === -1) {
        return { code: -1, message: '宠物档案不存在' }
      }
      resultPet = { ...pets[idx], ...petData, _id: pet._id }
      pets[idx] = resultPet
    } else {
      const newId = `pet_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`
      resultPet = { _id: newId, ...petData }
      pets.push(resultPet)
    }

    await db.collection('users').doc(user._id).update({
      data: {
        pets,
        updatedAt: db.serverDate(),
      },
    })

    return { code: 0, data: resultPet }
  } catch (e) {
    return { code: -1, message: e.message }
  }
}