animein ref 1

index.js

const axios = require('axios');
const sharp = require('sharp');
const md5 = require("md5");
const fs = require("fs");
const { SocksProxyAgent } = require("socks-proxy-agent")
const { convertTo64 } = require('../utils/base64');
let httpsAgent = new SocksProxyAgent("socks5://47.243.240.46:1080" )
let fileName = "AniMeXin - "+ Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 4);
httpsAgent.timeout = 30000;
const getAnime = (image) => {
return new Promise((resolve, reject) => {
const signV1 = (obj) => {
    const str = JSON.stringify(obj);
    return md5(
        'https://h5.tu.qq.com' +
        (str.length + (encodeURIComponent(str).match(/%[89ABab]/g)?.length || 0)) +
        'HQ31X02e',
    );
};
const obj = {
        busiId: 'ai_painting_anime_entry',
        extra: JSON.stringify({
            face_rects: [],
            version: 2,
            platform: 'web',
            data_report: {
                parent_trace_id:"350df969-1f0f-a369-c206-b39338b358ht",
                root_channel: '',
                level: 0,
            },
        }),
        images: [image],
    };
    const sign = signV1(obj);
        axios.request({
            httpsAgent,
            method:'POST',
            data: obj,
            url:'https://ai.tu.qq.com/trpc.shadow_cv.ai_processor_cgi.AIProcessorCgi/Process',
            headers: {
                        'Content-Type': 'application/json',
                        'Origin': 'https://h5.tu.qq.com',
                        'Referer': 'https://h5.tu.qq.com/',
                        'User-Agent':
                        'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36',
                        'x-sign-value': sign,
                        'x-sign-version': 'v1',
}
        })
        .then((data) => {
            if (data.data.code !== 0) return reject(data.data);
            resolve(data.data);
        })
        .catch(err => {
            reject(err);
        });
    });
}

const resizeImage = (image) => {
    return new Promise((resolve, reject) => {
        sharp(new Buffer.from(image, 'base64'))
        .resize(720, 1080)
        .toFormat('jpeg')
        .toBuffer()
        .then(img => resolve(img.toString('base64')))
        .catch(err => reject(err));
    });
}
const resizeImage2 = (image) => {
    return new Promise((resolve, reject) => {
        sharp(new Buffer.from(image, 'base64'))
        .toFormat('jpeg')
        .toBuffer()
        .then(img => resolve(img.toString('base64')))
        .catch(err => reject(err));
    });
}
const cropImage = (image) => {
    return new Promise((resolve, reject) => {
        sharp(new Buffer.from(image, 'base64'))
        .extract({left: 0, top: 0, width: 2000, height: 1200})
        .toBuffer()
        .then(img => resolve(img.toString('base64')))
        .catch(err => reject(err));
    });
}
/**
 * Allows you to transform an image to apply an anime / manga style
 * 
 * @param {objet} args
 * @param {string} args.photo - Image to transform, can be image path, image url or base64 image
 * @param {string} args.destinyFolder - Path to save the transformed image, if not provided the image will be delivered in base64
 * @return {Promise<string>} Transformed image
 */
 const transform = (args) => {
    return new Promise(async (resolve, reject) => {
        if (typeof args.photo === 'undefined' || args.photo === '') return reject('An image must be provided to transform...');

        const image = await convertTo64(args.photo);
        const base64Image = image.split(';base64,').pop();
        const resizedImage = await resizeImage(base64Image);
        getAnime(resizedImage)
        .then(async (data) => {
            if(data.code === 0 && data.extra !== undefined && data.extra !== ''){
                const extra = JSON.parse(data.extra);
                const image = extra.img_urls.find(el => el.includes('act-artifacts.shadowcv.qq.com'));
                const image64 = await convertTo64(image);
                const imageCrop = await cropImage(image64);
                resolve({
                    base64: imageCrop,
                    url: image,
                    all: extra
                });
                if (typeof args.destinyFolder !== 'undefined' && args.destinyFolder !== '') {
                    fs.writeFile(`${args.destinyFolder}/${fileName}.jpg`, imageCrop, {encoding: 'base64'}, function(err){
    //Finished
            });}
            } else {
                reject('An error occurred while trying to transform the image');
            }
        })
        .catch(err => {
            console.log('error', err);
            reject('An error occurred while trying to transform the image');
        });
    });
}

module.exports = {
    transform
};