import type { Browser, Page } from 'puppeteer'
import puppeteer from 'puppeteer-extra'
import StealthPlugin from 'puppeteer-extra-plugin-stealth'
import { clog } from "../poe/login";
import { awaitWrap, randomNum } from '../tools';
import Mock from 'mockjs'
import axios from 'axios'
import { SocksProxyAgent } from 'socks-proxy-agent'
import proxyChain from 'proxy-chain'
// const proxyChain = require('proxy-chain');

puppeteer.use(StealthPlugin())


function getProxy(options = {} as any) {
  const log = clog(options)
  log('开始获取代理ip')
  // return axios.get('https://tq.lunaproxy.com/getflowip?neek=1037914&num=1&type=1&sep=1&regions=us&ip_si=1&level=1&sb=', {
  // // return axios.get('https://www.miyaip.com/api/ProxyGenerage/PublicGenerateProxy?country=us&city=jaffrey®ion=nh&num=1&apiSwitch=0&mealType=2&genType=2&username=nmfk549724@163.com&secret=E3BCBmiyaipC23358C250F5', {
  //   // return axios.get('http://api.tianqiip.com/getip?secret=loy0r7fpmnlshm8l&num=1&type=json&port=3&time=3&mr=1&sign=5f73ab58ad7ab40346311014bef59b79', {
  //   timeout: 10 * 1000,
  // }).then(res => {
  //   // const [user, pass] = res.data.split(':')
  //   // return { user, pass, ip, port }
  //   // // return axios.get('https://www.miyaip.com/api/ProxyGenerage/PublicGenerateProxy?country=us&city=cairo®ion=ny&num=1&apiSwitch=0&mealType=2&genType=2&username=nmfk549724@163.com&secret=E3BCBmiyaipC23358C250F5').then(res => {
  //   const [ip, port] = res.data.replace(/(\n|\r\s)/g, '').split(':')
  //   console.log({ ip, port, })
  //   // const [ip, pory, user, pass] = res.data.split(':')
  //   return { ip, port, }
  // })
  return Promise.resolve({
    ip: '43.130.10.70',
    port: '22993',
    // user: '1E783B07miyaip489C251B3FA7',
    // pass: '1g7E3M4U5w03vO'
  })
}

// getProxy()//.then(res => console.log(res.data))


export const browsers = new Map<string, Browser>()
// 登录
async function login(options = {} as any): Promise<Page> {
  const { user, pass } = options

  const log = clog(options)

  // const [pErr, proxy] = await awaitWrap(getProxy(options))
  // if (pErr) {
  //   console.log('获取代理失败', pErr)
  //   return
  // }
  // console.log('获取代理成功', proxy)
  // // const proxyUrl = await proxyChain.anonymizeProxy(`socks5://${proxy.ip}:${proxy.port}`);
  // const proxyUrl = `socks5://${proxy.ip}:${proxy.port}`
  // console.log('proxyUrl', proxyUrl)
  // // const agent = new SocksProxyAgent(`socks5://${proxy.user}:${proxy.pass}@${proxy.ip}:${proxy.port}`);

  // // console.log('proxy', proxy)

  log('启动浏览器')

  const browser = await puppeteer.launch({
    headless: false,
    args: [
      '--no-sandbox',
      '--disable-setuid-sandbox',
    ]
  })
  log('创建新页面')
  const page = await browser.newPage()
  // await page.authenticate({
  //   username: proxy.user,
  //   password: proxy.pass
  // })
  browsers.set(user, browser)

  // await page.setRequestInterception(true);
  // page.on('request', (request) => {
  //   request.continue({
  //     agent
  //   });
  // });

  await page.setExtraHTTPHeaders({
    'accept-language': 'en-US,en;q=0.9,hy;q=0.8'
  })

  // page.goto('https://ip.900cha.com/')

  // return

  // 进入
  log('准备进入 gpt 登录')
  await page.goto('https://platform.openai.com')

  // const [err, res] = await awaitWrap(page.waitForNavigation({ timeout: 10000 }))
  // if (err) throw err

  log('等待出现输入框')
  await page.waitForSelector('#username', { visible: true, timeout: 10000 })

  // 输入账号
  log('输入账号')
  await page.type('#username', user)
  await Promise.all([
    page.waitForNavigation(),
    page.keyboard.press('Enter')
  ])

  log('等待出现密码输入框')
  await page.waitForSelector('#password', { visible: true })
  log('输入密码')
  await page.type('#password', pass)


  log('准备登录')
  await Promise.all([
    page.waitForNavigation({ timeout: 10000 }),
    page.keyboard.press('Enter')
  ])

  log('登录成功')

  return page
}

// 获取组织id
async function getOrgId(page: Page, options: any) {
  const log = clog(options)
  log('准备进入组织页面')
  await page.goto('https://platform.openai.com/account/org-settings')
  await page.waitForSelector('input')
  log('开始获取组织id')
  const orgId = await page.$$eval('input', (inputs: HTMLInputElement[]) => inputs?.[1].value || '')
  log('获取组织id成功', { orgId })
  return orgId
}

// 申请
export async function application(page: Page, options = {} as any) {
  const log = clog(options)

  log('准备进入申请页面')
  await page.goto('https://openai.com/waitlist/gpt-4-api')

  log('等待出现输入框')
  await page.waitForSelector('input')

  log('开始申请')

  const f = Mock.mock('@first')
  const l = Mock.mock('@last')
  console.log(f, l);

  await page.type('#firstname', f)
  await page.type('#lastname', l)
  await page.type('#email', options.user)
  await page.type('#organizationId', options.orgId)

  const i = randomNum(1, 4)

  log('选择申请类型', { i })
  page.evaluate((i) => {
    document.querySelector('#primaryUse').nextSibling.childNodes[i].childNodes[1].click()
  }, i)

  log('等待 gpt 生成描述')
  const desc = await generateDescription(i)
  await page.type('#ideas', desc)

  log('点击提交申请')
  const isSuccess = await page.evaluate((selector, searchText) => {
    const elements = Array.from(document.querySelectorAll(selector));
    const target = elements.find(el => el.textContent.trim() === searchText);

    if (target) target.click()
    return !!target
  }, 'button', 'Join waitlist')

  if (isSuccess) {
    log('提交成功，等待 gpt 返回结果')
    console.log('isSuccess', isSuccess)
    await page.waitForSelector('button[type="submit"] + .ui-richtext')
    const text = await page.evaluate(() => document.querySelector('button[type="submit"] + .ui-richtext').textContent)
    log('gpt 返回已结果', { result: text })
  }


  // await page.click('form[data-gtm-form-interact-id] button[type="submit"]')
}


// 生成描述
export async function generateDescription(desIndex = 0): Promise<string> {
  const types = {
    1: 'Build a new product',
    2: 'Integrate into an existing product',
    3: 'General exploration of capabilities',
    4: 'Academic research',
  }

  return axios({
    url: 'https://api.openai-proxy.com/v1/chat/completions',
    method: 'post',
    headers: {
      Authorization: `Bearer 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`
    },
    data: {
      max_tokens: 1024,
      model: 'gpt-3.5-turbo',
      messages: [
        { "role": "user", "content": `我在申请gpt4.0，我的类型是 ${types[desIndex]}，不要解释，给我一段申请的理由，以我的身份，以英语形式发给我，结尾不要带任何名字或位置，三四句话就行` },
      ]
    },
    timeout: 1000 * 30,
  }).then(res => {
    console.log('gpt 生成描述', res.data.choices[0].message.content)
    return res.data.choices[0].message.content
  })
}

export async function batchApplication(options) {
  const log = clog(options)
  log('开始', { ident: 'gpt-batch-4.0' })

  const [error, page] = await awaitWrap(login(options))
  if (error) return log('登录失败', { error, ident: 'gpt-batch-4.0' })

  await page.waitForSelector('.ovr-section')

  const orgId = await getOrgId(page, options)
  options.orgId = orgId
  console.log('orgId', orgId)

  await application(page, options)
}