Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Optionally, create a db when starting a local container #525

Merged
merged 9 commits into from
Dec 14, 2024
Merged
Show file tree
Hide file tree
Changes from 8 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
105 changes: 104 additions & 1 deletion src/commands/local.mjs
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
import chalk from "chalk";
import { AbortError } from "fauna";

import { container } from "../cli.mjs";
import { ensureContainerRunning } from "../lib/docker-containers.mjs";
import { CommandError } from "../lib/errors.mjs";
import { colorize, Format } from "../lib/formatting/colorize.mjs";

/**
* Starts the local Fauna container
Expand All @@ -8,6 +13,7 @@ import { CommandError } from "../lib/errors.mjs";
* It will reject if the container is not ready after the maximum number of attempts.
*/
async function startLocal(argv) {
const color = argv.color;
await ensureContainerRunning({
imageName: argv.image,
containerName: argv.name,
Expand All @@ -17,8 +23,67 @@ async function startLocal(argv) {
pull: argv.pull,
interval: argv.interval,
maxAttempts: argv.maxAttempts,
color: argv.color,
color,
});
if (argv.database) {
await createDatabase(argv);
}
}

async function createDatabase(argv) {
const { fql } = container.resolve("fauna");
const { runQuery } = container.resolve("faunaClientV10");
const logger = container.resolve("logger");
const color = argv.color;
logger.stderr(
colorize(`[CreateDatabase] Creating database '${argv.database}'...`, {
format: Format.LOG,
color,
}),
);
try {
const db = await runQuery({
secret: "secret",
url: `http://${argv.hostIp}:${argv.hostPort}`,
query: fql`
let name = ${argv.database}
let database = Database.byName(name)
let protected = ${argv.protected ?? null}
let typechecked = ${argv.typechecked ?? null}
let priority = ${argv.priority ?? null}
if (database == null) {
Database.create({
name: name,
protected: protected,
typechecked: typechecked,
priority: priority,
})
} else if (protected == database.protected && typechecked == database.typechecked && priority == database.priority) {
database
} else {
abort(database)
}`,
options: { format: "decorated" },
});
logger.stderr(
colorize(`[CreateDatabase] Database '${argv.database}' created.`, {
format: Format.LOG,
color,
}),
);
logger.stderr(colorize(db.data, { format: Format.FQL, color }));
} catch (e) {
if (e instanceof AbortError) {
throw new CommandError(
`${chalk.red(`[CreateDatabase] Database '${argv.database}' already exists but with differrent properties than requested:\n`)}
-----------------
${colorize(e.abort, { format: Format.FQL, color })}
-----------------
${chalk.red("Please use choose a different name using --name or align the --typechecked, --priority, and --protected with what is currently present.")}`,
);
}
throw e;
}
}

/**
Expand Down Expand Up @@ -67,6 +132,26 @@ function buildLocalCommand(yargs) {
type: "boolean",
default: true,
},
database: {
describe:
"The name of a database to create in the container. Omit to create no database.",
type: "string",
},
typechecked: {
describe:
"Enable typechecking for the database. Valid only if --database is set.",
type: "boolean",
},
protected: {
describe:
"Enable protected mode for the database. Protected mode disallows destructive schema changes. Valid only if --database is set.",
type: "boolean",
},
priority: {
type: "number",
description:
"User-defined priority for the database. Valid only if --database is set.",
},
})
.check((argv) => {
if (argv.maxAttempts < 1) {
Expand All @@ -80,6 +165,24 @@ function buildLocalCommand(yargs) {
{ hideHelp: false },
);
}
if (argv.typechecked && !argv.database) {
throw new CommandError(
"--typechecked can only be set if --database is set.",
{ hideHelp: false },
);
}
if (argv.protected && !argv.database) {
throw new CommandError(
"--protected can only be set if --database is set.",
{ hideHelp: false },
);
}
if (argv.priority && !argv.database) {
throw new CommandError(
cleve-fauna marked this conversation as resolved.
Show resolved Hide resolved
"--priority can only be set if --database is set.",
{ hideHelp: false },
);
}
return true;
});
}
Expand Down
2 changes: 0 additions & 2 deletions src/lib/docker-containers.mjs
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,6 @@ async function findContainer({ containerName, hostPort }) {
`[FindContainer] Container '${containerName}' is already \
in use on hostPort '${diffPort.PublicPort}'. Please use a new name via \
arguments --name <newName> --hostPort ${hostPort} to start the container.`,
{ hideHelp: false },
);
}
return result;
Expand Down Expand Up @@ -212,7 +211,6 @@ async function createContainer({
throw new CommandError(
`[StartContainer] The hostPort '${hostPort}' on IP '${hostIp}' is already occupied. \
Please pass a --hostPort other than '${hostPort}'.`,
{ hideHelp: false },
);
}
const dockerContainer = await docker.createContainer({
Expand Down
Loading
Loading