// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for details.

import { IRunOptions, PlatformType } from "../extension/launchArgs";
import { Telemetry } from "./telemetry";
import { TelemetryGenerator, IHasErrorCode } from "./telemetryGenerators";
import { ProjectVersionHelper, RNPackageVersions } from "./projectVersionHelper";
import { ErrorHelper } from "./error/errorHelper";
import { InternalErrorCode } from "./error/internalErrorCode";

export interface ITelemetryPropertyInfo {
    value: any;
    isPii: boolean;
}

export interface ICommandTelemetryProperties {
    [propertyName: string]: ITelemetryPropertyInfo;
}

export interface IExternalTelemetryProvider {
    sendTelemetry: (event: string, props: Telemetry.ITelemetryProperties, error?: Error) => void;
}

export class TelemetryHelper {
    public static sendSimpleEvent(
        eventName: string,
        properties?: Telemetry.ITelemetryProperties,
    ): void {
        const event = TelemetryHelper.createTelemetryEvent(eventName, properties);
        Telemetry.send(event);
    }

    public static createTelemetryEvent(
        eventName: string,
        properties?: Telemetry.ITelemetryProperties,
    ): Telemetry.TelemetryEvent {
        return new Telemetry.TelemetryEvent(eventName, properties);
    }

    public static telemetryProperty(propertyValue: any, pii?: boolean): ITelemetryPropertyInfo {
        return { value: String(propertyValue), isPii: pii || false };
    }

    public static addPropertyToTelemetryProperties(
        propValue: any,
        propName: string,
        properties: ICommandTelemetryProperties = {},
    ): any {
        properties[propName] = {
            value: propValue,
            isPii: false,
        };

        return properties;
    }

    /**
     * Adds platform based packages versions and other properties to the telemetry properties object
     */
    public static addPlatformPropertiesToTelemetryProperties(
        args: IRunOptions,
        versions: RNPackageVersions,
        properties: ICommandTelemetryProperties,
    ): any {
        properties = TelemetryHelper.addPropertyToTelemetryProperties(
            versions.reactNativeVersion,
            "reactNativeVersion",
            properties,
        );
        if (args.platform === PlatformType.Windows) {
            if (ProjectVersionHelper.isVersionError(versions.reactNativeWindowsVersion)) {
                throw ErrorHelper.getInternalError(
                    InternalErrorCode.ReactNativeWindowsIsNotInstalled,
                );
            }
            properties = TelemetryHelper.addPropertyToTelemetryProperties(
                versions.reactNativeWindowsVersion,
                "reactNativeWindowsVersion",
                properties,
            );
        }

        if (args.platform === PlatformType.macOS) {
            if (ProjectVersionHelper.isVersionError(versions.reactNativeMacOSVersion)) {
                throw ErrorHelper.getInternalError(
                    InternalErrorCode.ReactNativemacOSIsNotInstalled,
                );
            }
            properties = TelemetryHelper.addPropertyToTelemetryProperties(
                versions.reactNativeMacOSVersion,
                "reactNativeMacOSVersion",
                properties,
            );
        }

        if (args.platform === PlatformType.Exponent) {
            properties = TelemetryHelper.addPropertyToTelemetryProperties(
                args.expoHostType,
                "expoHostType",
                properties,
            );
            properties = TelemetryHelper.addPropertyToTelemetryProperties(
                args.expoPlatformType,
                "expoPlatformType",
                properties,
            );
        }

        return properties;
    }

    public static addTelemetryEventProperties(
        event: Telemetry.TelemetryEvent,
        properties: ICommandTelemetryProperties,
    ): void {
        if (!properties) {
            return;
        }

        Object.keys(properties).forEach((propertyName: string): void => {
            TelemetryHelper.addTelemetryEventProperty(
                event,
                propertyName,
                properties[propertyName].value,
                properties[propertyName].isPii,
            );
        });
    }

    public static sendErrorEvent(eventName: string, error: Error, errorDescription?: string): void {
        const event = TelemetryHelper.createTelemetryEvent(eventName);
        TelemetryHelper.addTelemetryEventErrorProperty(event, error, errorDescription, "");
        Telemetry.send(event);
    }

    public static sendDirect(eventName: string, properties?: Telemetry.ITelemetryProperties): void {
        const event = TelemetryHelper.createTelemetryEvent(eventName, { isDirect: true });
        if (properties) {
            TelemetryHelper.addTelemetryEventProperties(event, properties);
        }
        Telemetry.send(event);
    }

    public static sendCommandSuccessTelemetry(
        commandName: string,
        commandProperties: ICommandTelemetryProperties,
        args: string[] = [],
    ): void {
        const successEvent: Telemetry.TelemetryEvent = TelemetryHelper.createBasicCommandTelemetry(
            commandName,
            args,
        );

        TelemetryHelper.addTelemetryEventProperties(successEvent, commandProperties);

        Telemetry.send(successEvent);
    }

    public static addTelemetryEventProperty(
        event: Telemetry.TelemetryEvent,
        propertyName: string,
        propertyValue: any,
        isPii: boolean,
    ): void {
        if (Array.isArray(propertyValue)) {
            TelemetryHelper.addMultiValuedTelemetryEventProperty(
                event,
                propertyName,
                propertyValue,
                isPii,
            );
        } else {
            TelemetryHelper.setTelemetryEventProperty(event, propertyName, propertyValue, isPii);
        }
    }

    public static addTelemetryEventErrorProperty(
        event: Telemetry.TelemetryEvent,
        error: Error,
        errorDescription?: string,
        errorPropPrefix: string = "",
    ): void {
        const errorWithErrorCode: IHasErrorCode = <IHasErrorCode>(<Record<string, any>>error);
        if (errorWithErrorCode.errorCode) {
            this.addTelemetryEventProperty(
                event,
                `${errorPropPrefix}error.code`,
                errorWithErrorCode.errorCode,
                false,
            );
            if (errorDescription) {
                this.addTelemetryEventProperty(
                    event,
                    `${errorPropPrefix}error.message`,
                    errorDescription,
                    false,
                );
            }
        } else {
            this.addTelemetryEventProperty(
                event,
                `${errorPropPrefix}error.code`,
                InternalErrorCode.UnknownError,
                false,
            );
        }
    }

    public static addPropertiesFromOptions(
        telemetryProperties: ICommandTelemetryProperties,
        knownOptions: any,
        commandOptions: { [flag: string]: any },
        nonPiiOptions: string[] = [],
    ): ICommandTelemetryProperties {
        // We parse only the known options, to avoid potential private information that may appear on the command line
        let unknownOptionIndex = 1;
        Object.keys(commandOptions).forEach((key: string) => {
            const value: any = commandOptions[key];
            if (Object.keys(knownOptions).includes(key)) {
                // This is a known option. We"ll check the list to decide if it"s pii or not
                if (typeof value !== "undefined") {
                    // We encrypt all options values unless they are specifically marked as nonPii
                    telemetryProperties[`options.${key}`] = this.telemetryProperty(
                        value,
                        !nonPiiOptions.includes(key),
                    );
                }
            } else {
                // This is a not known option. We"ll assume that both the option and the value are pii
                telemetryProperties[`unknownOption${unknownOptionIndex}.name`] =
                    this.telemetryProperty(key, /* isPii*/ true);
                telemetryProperties[`unknownOption${unknownOptionIndex}.value`] =
                    this.telemetryProperty(value, /* isPii*/ true);
                unknownOptionIndex++;
            }
        });
        return telemetryProperties;
    }

    public static generate<T>(
        name: string,
        extendedParamsToSend: ICommandTelemetryProperties = {},
        codeGeneratingTelemetry: { (telemetry: TelemetryGenerator): Promise<T> | T },
    ): Promise<T> {
        const generator: TelemetryGenerator = new TelemetryGenerator(name, extendedParamsToSend);
        return generator
            .time("", () => codeGeneratingTelemetry(generator))
            .finally(() => generator.send());
    }

    private static createBasicCommandTelemetry(
        commandName: string,
        args: string[] = [],
    ): Telemetry.TelemetryEvent {
        const commandEvent: Telemetry.TelemetryEvent = new Telemetry.TelemetryEvent(
            commandName || "command",
        );

        if (!commandName && args && args.length > 0) {
            commandEvent.setPiiProperty("command", args[0]);
        }

        if (args) {
            TelemetryHelper.addTelemetryEventProperty(commandEvent, "argument", args, true);
        }

        return commandEvent;
    }

    private static setTelemetryEventProperty(
        event: Telemetry.TelemetryEvent,
        propertyName: string,
        propertyValue: string,
        isPii: boolean,
    ): void {
        if (isPii) {
            event.setPiiProperty(propertyName, String(propertyValue));
        } else {
            event.properties[propertyName] = String(propertyValue);
        }
    }

    private static addMultiValuedTelemetryEventProperty(
        event: Telemetry.TelemetryEvent,
        propertyName: string,
        propertyValue: any,
        isPii: boolean,
    ): void {
        for (let i = 0; i < propertyValue.length; i++) {
            TelemetryHelper.setTelemetryEventProperty(
                event,
                propertyName + String(i),
                propertyValue[i],
                isPii,
            );
        }
    }
}