import path from 'path';

import { Feature } from '../../spec-configuration/containerFeaturesConfiguration';

import { LogLevel } from '../../spec-utils/log';

import { writeLocalFile } from '../../spec-utils/pfs';

import { PackageCommandInput } from '../collectionCommonUtils/package';

import { SourceInformation, prepPackageCommand, packageCollection, packageSingleFeatureOrTemplate, OCICollectionFileName } from '../collectionCommonUtils/packageCommandImpl';

interface DevContainerCollectionMetadata {

sourceInformation: SourceInformation;

features: Feature[];

}

const collectionType = 'feature';

export async function doFeaturesPackageCommand(args: PackageCommandInput): Promise<DevContainerCollectionMetadata | undefined> {

args = await prepPackageCommand(args, collectionType);

const { output, outputDir } = args;

const isSingleFeature = args.isSingle;

// For each feature, package each feature and write to 'outputDir/{f}.tgz'

// Returns an array of feature metadata from each processed feature

let metadataOutput: Feature[] | undefined = [];

if (isSingleFeature) {

// Package individual features

output.write('Packaging single feature...', LogLevel.Info);

metadataOutput = await packageSingleFeature(args);

} else {

output.write('Packaging feature collection...', LogLevel.Info);

metadataOutput = await packageFeatureCollection(args);

}

if (!metadataOutput) {

output.write('Failed to package features', LogLevel.Error);

return undefined;

}

const collection: DevContainerCollectionMetadata = {

sourceInformation: {

source: 'devcontainer-cli',

},

features: metadataOutput,

};

// Write the metadata to a file

const metadataOutputPath = path.join(outputDir, OCICollectionFileName);

await writeLocalFile(metadataOutputPath, JSON.stringify(collection, null, 4));

return collection;

}

export async function packageSingleFeature(args: PackageCommandInput): Promise<Feature[] | undefined> {

return await packageSingleFeatureOrTemplate(args, collectionType);

}

export async function packageFeatureCollection(args: PackageCommandInput): Promise<Feature[] | undefined> {

return await packageCollection(args, collectionType);

}