import { emptyDirectory } from "../util.ts";
import _logger from "../logger.ts";
import { PkgSpec } from "../bindings/PkgSpec.ts";
import { SchemaVariantSpec } from "../bindings/SchemaVariantSpec.ts";
import { SchemaSpec } from "../bindings/SchemaSpec.ts";
import { bfsPropTree, ExpandedPropSpec } from "../spec/props.ts";
import {
ExpandedPkgSpec,
ExpandedSchemaSpec,
ExpandedSchemaVariantSpec,
} from "../spec/pkgs.ts";
import { PropSpec } from "../bindings/PropSpec.ts";
import { PipelineOptions, selectedProviders } from "../pipelines/types.ts";
// Import all providers to ensure they register themselves
import "../pipelines/aws/spec.ts";
import "../pipelines/hetzner/provider.ts";
import "../pipelines/dummy/spec.ts";
const logger = _logger.ns("siSpecs").seal();
const SI_SPEC_DIR = "si-specs";
export async function generateSiSpecs(options: PipelineOptions) {
const specs: ExpandedPkgSpec[] = [];
for (const config of selectedProviders(options)) {
logger.info(`Generating specs for provider: ${config.name}`);
const providerSpecs = await config.loadSchemas(options);
specs.push(...providerSpecs);
}
// WRITE OUT SPECS
await emptyDirectory(SI_SPEC_DIR);
let imported = 0;
for (const spec of specs) {
const specJson = JSON.stringify(unexpandPackageSpec(spec), null, 2);
const name = spec.name;
try {
logger.debug(`Writing ${name}.json`);
const blob = new Blob([specJson]);
if (blob.size > 4 * 1024 * 1024) {
logger.warn(`${spec.name} is bigger than 4MBs. Skipping.`);
continue;
}
await Deno.writeFile(`${SI_SPEC_DIR}/${name}.json`, blob.stream());
} catch (e) {
console.log(`Error writing to file: ${name}: ${e}`);
continue;
}
imported += 1;
}
console.log(`built ${imported} out of ${specs.length}`);
}
function unexpandPackageSpec(expandedSpec: ExpandedPkgSpec): PkgSpec {
// Take out cfSchema and other props we don't want in the final spec
return {
...expandedSpec,
schemas: expandedSpec.schemas.map(unexpandSchema),
};
}
function unexpandSchema(expanded: ExpandedSchemaSpec): SchemaSpec {
return {
...expanded,
variants: expanded.variants.map(unexpandVariant),
};
}
function unexpandVariant(
expanded: ExpandedSchemaVariantSpec,
): SchemaVariantSpec {
const { superSchema: _, ...variant } = expanded;
bfsPropTree(
[
variant.domain,
variant.resourceValue,
variant.secrets,
variant.secretDefinition,
],
unexpandProperty,
);
return { ...variant, sockets: [] };
}
function unexpandProperty(expanded: ExpandedPropSpec): PropSpec {
const deleteable = expanded as Partial<ExpandedPropSpec>;
delete deleteable.metadata;
delete deleteable.joiValidation;
delete deleteable.cfProp;
return expanded;
}
