mirror of
https://github.com/electron-vite/electron-vite-vue
synced 2025-04-25 22:16:59 +08:00
242 lines
7.4 KiB
TypeScript
242 lines
7.4 KiB
TypeScript
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('http://api.proxy.ipidea.io/getBalanceProxyIp?num=1&return_type=json&lb=1&sb=0&flow=1®ions=&protocol=socks5', {
|
|
// 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 res.data.data[0]
|
|
})
|
|
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) {
|
|
// log('获取代理失败')
|
|
// 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)
|
|
// console.log(`curl --socks5 ${proxy.ip}:${proxy.port} https://jd.com`)
|
|
// // // 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',
|
|
// `--proxy-server=${proxyUrl}`,
|
|
]
|
|
})
|
|
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)
|
|
}
|