import {
AttachInternetGatewayCommandInput,
CreateInternetGatewayCommandInput,
DeleteInternetGatewayCommandInput,
DescribeInternetGatewaysCommandInput,
DetachInternetGatewayCommandInput,
} from "@aws-sdk/client-ec2";
import z from "zod";
import {
dryRunSchema,
filterSchema,
paginationSchema,
regionSchema,
tagSpecificationSchema,
} from "./common";
// List Internet Gateways
const listInternetGatewaysBaseSchema = z.object({
InternetGatewayIds: z
.array(z.string())
.optional()
.describe("The IDs of the internet gateways"),
Filters: filterSchema,
DryRun: dryRunSchema,
...paginationSchema,
}) satisfies z.ZodType<DescribeInternetGatewaysCommandInput>;
// Create Internet Gateway
const createInternetGatewayBaseSchema = z.object({
DryRun: dryRunSchema,
TagSpecifications: tagSpecificationSchema,
}) satisfies z.ZodType<CreateInternetGatewayCommandInput>;
// Delete Internet Gateway
const deleteInternetGatewayBaseSchema = z.object({
InternetGatewayId: z.string().describe("The ID of the internet gateway"),
DryRun: dryRunSchema,
}) satisfies z.ZodType<DeleteInternetGatewayCommandInput>;
// Attach Internet Gateway
const attachInternetGatewayBaseSchema = z.object({
InternetGatewayId: z.string().describe("The ID of the internet gateway"),
VpcId: z.string().describe("The ID of the VPC"),
DryRun: dryRunSchema,
}) satisfies z.ZodType<AttachInternetGatewayCommandInput>;
// Detach Internet Gateway
const detachInternetGatewayBaseSchema = z.object({
InternetGatewayId: z.string().describe("The ID of the internet gateway"),
VpcId: z.string().describe("The ID of the VPC"),
DryRun: dryRunSchema,
}) satisfies z.ZodType<DetachInternetGatewayCommandInput>;
// Export the schemas
export const listInternetGatewaysSchema = z.object({
region: regionSchema,
InternetGatewayArgs: listInternetGatewaysBaseSchema,
});
export const createInternetGatewaySchema = z.object({
region: regionSchema,
InternetGatewayArgs: createInternetGatewayBaseSchema,
});
export const deleteInternetGatewaySchema = z.object({
region: regionSchema,
InternetGatewayArgs: deleteInternetGatewayBaseSchema,
});
export const attachInternetGatewaySchema = z.object({
region: regionSchema,
InternetGatewayArgs: attachInternetGatewayBaseSchema,
});
export const detachInternetGatewaySchema = z.object({
region: regionSchema,
InternetGatewayArgs: detachInternetGatewayBaseSchema,
});
// Export the types
export type ListInternetGatewaysArgs = z.infer<
typeof listInternetGatewaysSchema
>;
export type CreateInternetGatewayArgs = z.infer<
typeof createInternetGatewaySchema
>;
export type DeleteInternetGatewayArgs = z.infer<
typeof deleteInternetGatewaySchema
>;
export type AttachInternetGatewayArgs = z.infer<
typeof attachInternetGatewaySchema
>;
export type DetachInternetGatewayArgs = z.infer<
typeof detachInternetGatewaySchema
>;
