trackingUtils.ts

import * as vscode from "vscode";
import axios from "axios";
import { getLeetCodeEndpoint } from "../commands/plugin";
import { Endpoint } from "../shared";
import { leetCodeManager } from "../leetCodeManager";

const getTimeZone = (): string => {
    const endPoint: string = getLeetCodeEndpoint();
    if (endPoint === Endpoint.LeetCodeCN) {
        return "Asia/Shanghai";
    } else {
        return "UTC";
    }
};

interface IReportData {
    event_key: string;
    type?: "click" | "expose" | string;
    anonymous_id?: string;
    tid?: number;
    ename?: string; // event name
    href?: string;
    referer?: string;
    extra?: string;
    target?: string;
}

interface ITrackData {
    reportCache: IReportData[];
    isSubmit: boolean;
    report: (reportItems: IReportData | IReportData[]) => void;
    submitReport: (useSendBeason: boolean) => Promise<void>;
    reportUrl: string;
}

const testReportUrl = "https://analysis.lingkou.xyz/i/event";
const prodReportUrl = "https://analysis.leetcode.cn/i/event";

function getReportUrl() {
    if (process.env.NODE_ENV === "production") {
        return prodReportUrl;
    } else {
        return testReportUrl;
    }
}

const _charStr = "abacdefghjklmnopqrstuvwxyzABCDEFGHJKLMNOPQRSTUVWXYZ0123456789!@#$%^&*()_+=";

function RandomIndex(min: number, max: number, i: number) {
    let index = Math.floor(Math.random() * (max - min + 1) + min);
    const numStart = _charStr.length - 10;
    if (i === 0 && index >= numStart) {
        index = RandomIndex(min, max, i);
    }
    return index;
}

function getRandomString(len: number) {
    const min = 0;
    const max = _charStr.length - 1;
    let _str = "";
    len = len || 15;
    for (let i = 0, index; i < len; i++) {
        index = RandomIndex(min, max, i);
        _str += _charStr[index];
    }
    return _str;
}

function getAllowReportDataConfig() {
    const leetCodeConfig = vscode.workspace.getConfiguration("leetcode");
    const allowReportData = !!leetCodeConfig.get<boolean>("allowReportData");
    return allowReportData;
}

class TrackData implements ITrackData {
    public reportCache: IReportData[] = [];

    public isSubmit: boolean = false;

    public reportUrl: string = getReportUrl();

    private sendTimer: NodeJS.Timeout | undefined;

    public report = (reportItems: IReportData | IReportData[]) => {
        if (!getAllowReportDataConfig()) return;

        this.sendTimer && clearTimeout(this.sendTimer);

        if (!Array.isArray(reportItems)) {
            reportItems = [reportItems];
        }
        const randomId = getRandomString(60);
        reportItems.forEach((item) => {
            this.reportCache.push({
                ...item,
                referer: "vscode",
                target: leetCodeManager.getUser() ?? "",
                anonymous_id: item.anonymous_id ?? (randomId as string),
            });
        });
        this.sendTimer = setTimeout(this.submitReport, 800);
    };

    public submitReport = async () => {
        if (!getAllowReportDataConfig()) return;
        const dataList = JSON.stringify(this.reportCache);

        if (!this.reportCache.length || this.isSubmit) {
            return;
        }
        this.reportCache = [];
        try {
            this.isSubmit = true;
            axios.defaults.withCredentials = true;
            await axios.post(this.reportUrl, `dataList=${encodeURIComponent(dataList)}`, {
                headers: {
                    "Content-Type": "application/x-www-form-urlencoded",
                    "x-timezone": getTimeZone(),
                },
            });
        } catch (e) {
            this.reportCache = this.reportCache.concat(JSON.parse(dataList));
        } finally {
            this.isSubmit = false;
        }
    };
}

export default new TrackData();