microsoft/vscode-react-native

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

import * as pathModule from "path";

import { ErrorHelper } from "../error/errorHelper";
import { InternalErrorCode } from "../error/internalErrorCode";
import { FileSystem } from "./fileSystem";

interface IPackageDependencyDict {
    [packageName: string]: string;
}

export interface IPackageInformation {
    name: string;
    version: string;
    dependencies?: IPackageDependencyDict;
    main?: string;
    [key: string]: any;
}

export class Package {
    private INFORMATION_PACKAGE_FILENAME = "package.json";
    private DEPENDENCIES_SUBFOLDER = "node_modules";

    private fileSystem: FileSystem;

    private _path: string;

    constructor(path: string, { fileSystem = new FileSystem() } = {}) {
        this._path = path;
        this.fileSystem = fileSystem;
    }

    public getPackageVersionFromNodeModules(packageName: string): Promise<string> {
        return this.dependencyPackage(packageName).version();
    }

    public async parsePackageInformation(): Promise<IPackageInformation> {
        const data = await this.fileSystem.readFile(this.informationJsonFilePath(), "utf8");
        return <IPackageInformation>JSON.parse(data.toString());
    }

    public name(): Promise<string> {
        return this.parseProperty("name");
    }

    public dependencies(): Promise<IPackageDependencyDict> {
        return this.parseProperty("dependencies");
    }

    public devDependencies(): Promise<IPackageDependencyDict> {
        return this.parseProperty("devDependencies");
    }

    public async version(): Promise<string> {
        const version = await this.parseProperty("version");
        if (typeof version === "string") {
            return version;
        }
        throw ErrorHelper.getInternalError(
            InternalErrorCode.CouldNotParsePackageVersion,
            this.informationJsonFilePath(),
            version,
        );
    }

    public async setMainFile(value: string): Promise<void> {
        const packageInformation = await this.parsePackageInformation();
        packageInformation.main = value;
        return this.fileSystem.writeFile(
            this.informationJsonFilePath(),
            JSON.stringify(<Record<string, any>>packageInformation),
        );
    }

    public dependencyPath(dependencyName: string): string {
        return pathModule.resolve(this._path, this.DEPENDENCIES_SUBFOLDER, dependencyName);
    }

    public dependencyPackage(dependencyName: string): Package {
        return new Package(this.dependencyPath(dependencyName), { fileSystem: this.fileSystem });
    }

    public informationJsonFilePath(): string {
        return pathModule.resolve(this._path, this.INFORMATION_PACKAGE_FILENAME);
    }

    private async parseProperty(name: string): Promise<any> {
        const packageInformation = await this.parsePackageInformation();
        return packageInformation[name];
    }
}