"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.sendPost = exports.getPostOwner = exports.WebshotHelpers = exports.ScreenNameNormalizer = exports.SessionManager = exports.urlSegmentToId = exports.idToUrlSegment = exports.isValidUrlSegment = exports.parseLink = exports.linkBuilder = void 0;
const crypto = require("crypto");
const fs = require("fs");
const http = require("http");
const path = require("path");
const url_1 = require("url");
const util_1 = require("util");
const instagram_id_to_url_segment_1 = require("instagram-id-to-url-segment");
Object.defineProperty(exports, "idToUrlSegment", { enumerable: true, get: function () { return instagram_id_to_url_segment_1.instagramIdToUrlSegment; } });
const instagram_private_api_1 = require("instagram-private-api");
const socks_proxy_agent_1 = require("socks-proxy-agent");
const loggers_1 = require("./loggers");
const utils_1 = require("./utils");
const webshot_1 = require("./webshot");
const parseLink = (link) => {
    let match = /instagram\.com\/p\/([A-Za-z0-9\-_]+)/.exec(link);
    if (match)
        return { postUrlSegment: match[1] };
    match =
        /instagram\.com\/([^\/?#]+)/.exec(link) ||
            /^([^\/?#]+)$/.exec(link);
    if (match)
        return { userName: ScreenNameNormalizer.normalize(match[1]).split(':')[0] };
    return;
};
exports.parseLink = parseLink;
const isValidUrlSegment = (input) => /^[A-Za-z0-9\-_]+$/.test(input);
exports.isValidUrlSegment = isValidUrlSegment;
const linkBuilder = (config) => {
    if (config.userName)
        return `https://www.instagram.com/${config.userName}/`;
    if (config.postUrlSegment)
        return `https://www.instagram.com/p/${config.postUrlSegment}/`;
};
exports.linkBuilder = linkBuilder;
const urlSegmentToId = (urlSegment) => urlSegment.length <= 28 ?
    instagram_id_to_url_segment_1.urlSegmentToInstagramId(urlSegment) : instagram_id_to_url_segment_1.urlSegmentToInstagramId(urlSegment.slice(0, -28));
exports.urlSegmentToId = urlSegmentToId;
class SessionManager {
    constructor(client, file, credentials, codeServicePort) {
        this.init = () => {
            this.ig.state.generateDevice(this.username);
            this.ig.request.end$.subscribe(() => { this.save(); });
            const filePath = path.resolve(this.lockfile);
            if (fs.existsSync(filePath)) {
                try {
                    const serialized = JSON.parse(fs.readFileSync(filePath, 'utf8'));
                    return this.ig.state.deserialize(serialized).then(() => {
                        logger.info(`successfully loaded client session cookies for user ${this.username}`);
                    });
                }
                catch (err) {
                    logger.error(`failed to load client session cookies from file ${this.lockfile}: `, err);
                    return Promise.resolve();
                }
            }
            else {
                return this.login().catch((err) => {
                    logger.error(`error while trying to log in as user ${this.username}, error: ${err}`);
                    logger.warn('attempting to retry after 1 minute...');
                    if (fs.existsSync(filePath))
                        fs.unlinkSync(filePath);
                    util_1.promisify(setTimeout)(60000).then(this.init);
                });
            }
        };
        this.handle2FA = (submitter) => new Promise((resolve, reject) => {
            const token = crypto.randomBytes(20).toString('hex');
            logger.info('please submit the code with a one-time token from your browser with this path:');
            logger.info(`/confirm-2fa?code=<the code you received>&token=${token}`);
            let working;
            const server = http.createServer((req, res) => {
                const { pathname, query } = url_1.parse(req.url, true);
                if (!working && pathname === '/confirm-2fa' && query.token === token &&
                    typeof (query.code) === 'string' && /^\d{6}$/.test(query.code)) {
                    const code = query.code;
                    logger.debug(`received code: ${code}`);
                    working = true;
                    submitter(code)
                        .then(response => { res.write('OK'); res.end(); server.close(() => resolve(response)); })
                        .catch(err => { res.write('Error'); res.end(); reject(err); })
                        .finally(() => { working = false; });
                }
            });
            server.listen(this.codeServicePort);
        });
        this.login = () => this.ig.simulate.preLoginFlow()
            .then(() => this.ig.account.login(this.username, this.password))
            .catch((err) => {
            if (err instanceof instagram_private_api_1.IgLoginTwoFactorRequiredError) {
                const { two_factor_identifier, totp_two_factor_on } = err.response.body.two_factor_info;
                logger.debug(`2FA info: ${JSON.stringify(err.response.body.two_factor_info)}`);
                logger.info(`login is requesting two-factor authentication via ${totp_two_factor_on ? 'TOTP' : 'SMS'}`);
                return this.handle2FA(code => this.ig.account.twoFactorLogin({
                    username: this.username,
                    verificationCode: code,
                    twoFactorIdentifier: two_factor_identifier,
                    verificationMethod: totp_two_factor_on ? '0' : '1',
                }));
            }
            throw err;
        })
            .then(user => new Promise(resolve => {
            logger.info(`successfully logged in as ${this.username}`);
            process.nextTick(() => resolve(this.ig.simulate.postLoginFlow().then(() => user)));
        }));
        this.save = () => this.ig.state.serialize()
            .then((serialized) => {
            delete serialized.constants;
            return fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(serialized, null, 2), 'utf-8');
        });
        this.ig = client;
        this.lockfile = file;
        [this.username, this.password] = credentials;
        this.codeServicePort = codeServicePort;
    }
}
exports.SessionManager = SessionManager;
class ScreenNameNormalizer {
    static normalizeLive(username) {
        return __awaiter(this, void 0, void 0, function* () {
            if (this._queryUser) {
                return yield this._queryUser(username)
                    .catch((err) => {
                    if (!(err instanceof instagram_private_api_1.IgExactUserNotFoundError)) {
                        logger.warn(`error looking up user: ${err.message}`);
                        return `${username}:`;
                    }
                    return null;
                });
            }
            return this.normalize(username);
        });
    }
}
exports.ScreenNameNormalizer = ScreenNameNormalizer;
ScreenNameNormalizer.normalize = (username) => `${username.toLowerCase().replace(/^@/, '')}:`;
let browserLogin = (page) => Promise.resolve();
let browserSaveCookies = browserLogin;
let isWaitingForLogin = false;
const acceptCookieConsent = (page) => page.click('button:has-text("すべて許可")', { timeout: 5000 })
    .then(() => logger.info('accepted cookie consent'))
    .catch((err) => { if (err.name !== 'TimeoutError')
    throw err; });
exports.WebshotHelpers = {
    handleLogin: browserLogin,
    handleCookieConsent: acceptCookieConsent,
    get isWaitingForLogin() { return isWaitingForLogin; },
};
let getPostOwner = (segmentId) => Promise.reject();
exports.getPostOwner = getPostOwner;
let sendPost = (segmentId, receiver) => {
    throw Error();
};
exports.sendPost = sendPost;
const logger = loggers_1.getLogger('instagram');
const maxTrials = 3;
const retryInterval = 1500;
const ordinal = (n) => {
    switch ((Math.trunc(n / 10) % 10 === 1) ? 0 : n % 10) {
        case 1:
            return `${n}st`;
        case 2:
            return `${n}nd`;
        case 3:
            return `${n}rd`;
        default:
            return `${n}th`;
    }
};
const retryOnError = (doWork, onRetry) => new Promise(resolve => {
    const retry = (reason, count) => {
        setTimeout(() => {
            let terminate = false;
            onRetry(reason, count, defaultValue => { terminate = true; resolve(defaultValue); });
            if (!terminate)
                doWork().then(resolve).catch(error => retry(error, count + 1));
        }, retryInterval);
    };
    doWork().then(resolve).catch(error => retry(error, 1));
});
class default_1 {
    constructor(opt) {
        this.webshotCookies = [];
        this.launch = () => {
            this.webshot = new webshot_1.default(this.wsUrl, this.mode, () => this.webshotCookies, doOnNewPage => {
                this.queryUserMedia = ((userName, targetId) => {
                    let page;
                    const url = linkBuilder({ userName });
                    logger.debug(`pulling ${targetId !== '0' ? `feed ${url} up to ${targetId}` : `top of feed ${url}`}...`);
                    return doOnNewPage(newPage => {
                        page = newPage;
                        let timeout = this.webshotDelay;
                        const startTime = new Date().getTime();
                        const getTimerTime = () => new Date().getTime() - startTime;
                        const getTimeout = () => isWaitingForLogin ? 0 : Math.max(90000, timeout - getTimerTime());
                        return page.context().addCookies(this.webshotCookies)
                            .then(() => page.goto(url, { waitUntil: 'load', timeout: getTimeout() }))
                            .then(response => {
                            if (response.status() !== 200) {
                                const err = new Error(`error navigating to user page, error was: ${response.status()} ${response.statusText()}`);
                                throw Object.defineProperty(err, 'name', {
                                    value: 'ResponseError',
                                });
                            }
                        }).then(() => acceptCookieConsent(page))
                            .then(() => (next => Promise.race([
                            browserLogin(page)
                                .catch((err) => {
                                if (err.name === 'TimeoutError') {
                                    logger.warn('navigation timed out, assuming login has failed');
                                    isWaitingForLogin = false;
                                }
                                throw err;
                            })
                                .then(() => browserSaveCookies(page))
                                .then(() => page.goto(url)).then(next),
                            next(),
                        ]))(() => util_1.promisify(setTimeout)(2000).then(() => page.waitForSelector('article', { timeout: getTimeout() })))).then(handle => {
                            const postHandler = () => {
                                const toId = (href) => { var _a; return urlSegmentToId(((_a = /\/p\/(.*)\/$/.exec(href)) !== null && _a !== void 0 ? _a : [, ''])[1]); };
                                if (targetId === '0') {
                                    return handle.$$eval('a', as => as.filter(a => !a.querySelector('[aria-label="IGTV"]'))[0].href).then(href => Number(toId(href)) > 0 ? [toId(href)] : []);
                                }
                                return handle.$$eval('a', as => as.filter(a => !a.querySelector('[aria-label="IGTV"]')).map(a => a.href)).then(hrefs => {
                                    let id;
                                    const itemIds = [];
                                    for (const href of hrefs) {
                                        id = toId(href);
                                        if (id && utils_1.BigNumOps.compare(id, targetId) > 0)
                                            itemIds.push(id);
                                        else
                                            return itemIds;
                                    }
                                    logger.info('unable to find a smaller id than target, trying on next page...');
                                    return null;
                                });
                            };
                            return postHandler().then(itemIds => {
                                if (itemIds)
                                    return itemIds;
                                timeout += this.workInterval * 500;
                                return handle.$$('a')
                                    .then(as => { as.pop().scrollIntoViewIfNeeded(); return as.length + 1; })
                                    .then(loadedCount => page.waitForFunction(count => document.querySelectorAll('article a').length > count, loadedCount))
                                    .then(postHandler);
                            });
                        }).catch((err) => {
                            if (err.name !== 'TimeoutError' && err.name !== 'ResponseError')
                                throw err;
                            if (err.name === 'ResponseError') {
                                logger.warn(`error while fetching tweets for ${userName}: ${err.message}`);
                            }
                            else
                                logger.warn(`navigation timed out at ${getTimerTime()} ms`);
                            return [];
                        }).then(itemIds => util_1.promisify(setTimeout)(getTimeout()).then(() => itemIds.map(id => this.lazyGetMediaById(id))));
                    }).finally(() => { page.close(); });
                });
                setTimeout(this.work, this.workInterval * 1000 / this.lock.feed.length);
            });
        };
        this.queryUser = (username) => this.client.user.searchExact(username)
            .then(user => `${user.username}:${user.pk}`);
        this.workOnMedia = (lazyMediaItems, sendMedia) => this.webshot(lazyMediaItems, sendMedia, this.webshotDelay);
        this.urlSegmentToId = urlSegmentToId;
        this.lazyGetMediaById = (id) => ({
            pk: id,
            item: () => this.client.media.info(id).then(media => {
                const mediaItem = media.items[0];
                logger.debug(`api returned media post ${JSON.stringify(mediaItem)} for query id=${id}`);
                return mediaItem;
            }),
        });
        this.sendMedia = (source, ...to) => (msg, text, author) => {
            to.forEach(subscriber => {
                logger.info(`pushing data${source ? ` of ${source}` : ''} to ${JSON.stringify(subscriber)}`);
                retryOnError(() => this.bot.sendTo(subscriber, msg), (_, count, terminate) => {
                    if (count <= maxTrials) {
                        logger.warn(`retry sending to ${subscriber.chatID} for the ${ordinal(count)} time...`);
                    }
                    else {
                        logger.warn(`${count - 1} consecutive failures while sending message chain, trying plain text instead...`);
                        terminate(this.bot.sendTo(subscriber, author + text, true));
                    }
                });
            });
        };
        this.workOnFeed = (feed) => new Promise(resolve => {
            const match = /https:\/\/www\.instagram\.com\/([^\/]+)/.exec(feed);
            if (!match) {
                logger.error(`current feed "${feed}" is invalid, please remove this feed manually`);
                return resolve([]);
            }
            return resolve(this.queryUserMedia(match[1], this.lock.threads[feed].offset)
                .catch((error) => {
                logger.error(`error scraping media off profile page of ${match[1]}, error: ${error}`);
                return [];
            }));
        }).then(mediaItems => {
            const currentThread = this.lock.threads[feed];
            const updateDate = () => currentThread.updatedAt = new Date().toString();
            if (!mediaItems || mediaItems.length === 0) {
                updateDate();
                return;
            }
            const topOfFeed = mediaItems[0].pk;
            const updateOffset = () => { currentThread.offset = topOfFeed; };
            if (currentThread.offset === '-1') {
                updateOffset();
                return;
            }
            if (currentThread.offset === '0')
                mediaItems.splice(1);
            return this.workOnMedia(mediaItems, this.sendMedia(`thread ${feed}`, ...currentThread.subscribers))
                .then(updateDate).then(updateOffset);
        });
        this.work = () => {
            const lock = this.lock;
            if (this.workInterval < 1)
                this.workInterval = 1;
            if (this.isInactiveTime || lock.feed.length === 0) {
                setTimeout(this.work, this.workInterval * 1000 / lock.feed.length);
                return;
            }
            lock.feed.forEach((feed, index) => {
                if (!lock.threads[feed] ||
                    !lock.threads[feed].subscribers ||
                    lock.threads[feed].subscribers.length === 0) {
                    logger.warn(`nobody subscribes thread ${feed}, removing from feed`);
                    delete lock.threads[index];
                    lock.feed.splice(index, 1);
                    fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                }
            });
            const queuedFeeds = lock.feed.slice(0, lock.workon + 1).reverse();
            utils_1.chainPromises(utils_1.Arr.chunk(queuedFeeds, 5).map((arr, i) => () => Promise.all(arr.map((currentFeed, j) => {
                if (Date.now() - new Date(lock.threads[currentFeed].updatedAt).getTime() > 3600000)
                    return;
                const workon = (queuedFeeds.length - 1) - (i * 5 + j);
                fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                const promiseDelay = this.workInterval * (Math.random() + j) * 500 / lock.feed.length;
                const startTime = new Date().getTime();
                const getTimerTime = () => new Date().getTime() - startTime;
                const promise = util_1.promisify(setTimeout)(promiseDelay).then(() => {
                    logger.info(`about to pull from feed #${workon}: ${currentFeed}`);
                    if (j === arr.length - 1)
                        logger.info(`timeout for this batch job: ${Math.trunc(promiseDelay * 2)} ms`);
                    return this.workOnFeed(currentFeed);
                }).then(() => {
                    lock.workon = (workon || lock.feed.length) - 1;
                    if (j === arr.length - 1) {
                        logger.info(`batch job #${workon}-${workon + j} completed after ${getTimerTime()} ms`);
                    }
                    fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                });
                return Promise.race([promise, isWaitingForLogin ? utils_1.neverResolves() : util_1.promisify(setTimeout)(promiseDelay * 3)]);
            }))))
                .then(() => {
                let timeout = this.workInterval * 500;
                if (timeout < 1000)
                    timeout = 1000;
                fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                setTimeout(() => {
                    this.work();
                }, timeout);
            });
        };
        this.client = new instagram_private_api_1.IgApiClient();
        if (opt.proxyUrl) {
            try {
                const url = new URL(opt.proxyUrl);
                if (!/^socks(?:4a?|5h?)?:$/.test(url.protocol))
                    throw Error();
                if (!url.port)
                    url.port = '1080';
                this.client.request.defaults.agent = new socks_proxy_agent_1.SocksProxyAgent({
                    hostname: url.hostname,
                    port: url.port,
                    userId: url.username,
                    password: url.password,
                });
            }
            catch (e) {
                logger.warn(`invalid socks proxy url: ${opt.proxyUrl}, ignoring`);
            }
        }
        this.session = new SessionManager(this.client, opt.sessionLockfile, opt.credentials, opt.codeServicePort);
        this.lockfile = opt.lockfile;
        this.webshotCookiesLockfile = opt.webshotCookiesLockfile;
        this.lock = opt.lock;
        this.inactiveHours = opt.inactiveHours;
        this.workInterval = opt.workInterval;
        this.bot = opt.bot;
        this.webshotDelay = opt.webshotDelay;
        this.mode = opt.mode;
        this.wsUrl = opt.wsUrl;
        const cookiesFilePath = path.resolve(this.webshotCookiesLockfile);
        try {
            this.webshotCookies = JSON.parse(fs.readFileSync(cookiesFilePath, 'utf8'));
            logger.info(`loaded webshot cookies from file ${this.webshotCookiesLockfile}`);
        }
        catch (err) {
            logger.warn(`failed to load webshot cookies from file ${this.webshotCookiesLockfile}: `, err.message);
            logger.warn('cookies will be saved to this file when needed');
        }
        browserLogin = page => page.fill('input[name="username"]', opt.credentials[0], { timeout: 0 })
            .then(() => {
            if (isWaitingForLogin !== true)
                return;
            logger.warn('still waiting for login, pausing execution...');
            return utils_1.neverResolves();
        })
            .then(() => { isWaitingForLogin = true; logger.warn('blocked by login dialog, trying to log in manually...'); })
            .then(() => page.fill('input[name="password"]', opt.credentials[1], { timeout: 0 }))
            .then(() => page.click('button[type="submit"]', { timeout: 0 }))
            .then(() => (next => Promise.race([
            page.waitForSelector('#verificationCodeDescription', { timeout: 0 }).then(handle => handle.innerText()).then(text => {
                logger.info(`login is requesting two-factor authentication via ${/認証アプリ/.test(text) ? 'TOTP' : 'SMS'}`);
                return this.session.handle2FA(code => page.fill('input[name="verificationCode"]', code, { timeout: 0 }))
                    .then(() => page.click('button:has-text("実行")', { timeout: 0 }))
                    .then(next);
            }),
            next(),
        ]))(() => page.click('button:has-text("情報を保存")', { timeout: 0 }).then(() => { isWaitingForLogin = false; })));
        browserSaveCookies = page => page.context().cookies()
            .then(cookies => {
            this.webshotCookies = cookies;
            logger.info('successfully logged in, saving cookies to file...');
            fs.writeFileSync(path.resolve(this.webshotCookiesLockfile), JSON.stringify(cookies, null, 2), 'utf-8');
        });
        exports.WebshotHelpers.handleLogin = page => browserLogin(page)
            .then(() => page.waitForSelector('img[data-testid="user-avatar"]', { timeout: this.webshotDelay }))
            .then(() => browserSaveCookies(page))
            .catch((err) => {
            if (err.name === 'TimeoutError') {
                logger.warn('navigation timed out, assuming login has failed');
                isWaitingForLogin = false;
            }
            throw err;
        });
        ScreenNameNormalizer._queryUser = this.queryUser;
        const parseMediaError = (err) => {
            if (!(err instanceof instagram_private_api_1.IgResponseError && err.text === 'Media not found or unavailable')) {
                logger.warn(`error retrieving instagram media: ${err.message}`);
                return `获取媒体时出现错误：${err.message}`;
            }
            return '找不到请求的媒体，它可能已被删除。';
        };
        exports.getPostOwner = (segmentId) => this.client.media.info(urlSegmentToId(segmentId))
            .then(media => media.items[0].user)
            .then(user => `${user.username}:${user.pk}`)
            .catch((err) => { throw Error(parseMediaError(err)); });
        exports.sendPost = (segmentId, receiver) => {
            const lazyMedia = this.lazyGetMediaById(urlSegmentToId(segmentId));
            return lazyMedia.item().then(mediaItem => {
                const lock = this.lock;
                const feed = linkBuilder({ userName: mediaItem.user.username });
                if (lock.feed.includes(feed) && lock.threads[feed].offset < mediaItem.pk) {
                    logger.info(`post is newer than last offset of thread (${instagram_id_to_url_segment_1.instagramIdToUrlSegment(lock.threads[feed].offset)}), updating...`);
                    this.workOnFeed(feed);
                    if (lock.threads[feed].subscribers.some(subscriber => subscriber.chatID.toString() === receiver.chatID.toString() &&
                        subscriber.chatType === receiver.chatType))
                        return logger.info(`receiver has already subscribed to feed ${feed}, not sending again`);
                }
                lazyMedia.item = () => Promise.resolve(mediaItem);
                this.workOnMedia([lazyMedia], this.sendMedia(`instagram media ${segmentId}`, receiver));
            }).catch((err) => { this.bot.sendTo(receiver, parseMediaError(err)); });
        };
    }
    get isInactiveTime() {
        const timeToEpoch = (h = 0, m = 0) => new Date().setHours(h, m, 0, 0);
        return this.inactiveHours
            .map(rangeStr => ((start, end) => ({ start, end }))(...rangeStr.split('-', 2).map(timeStr => timeToEpoch(...timeStr.split(':', 2).map(Number)))))
            .some(range => (now => now >= range.start && now < range.end)(Date.now()));
    }
}
exports.default = default_1;