drizzle-team--drizzle-orm/drizzle-seed/tests/pg/generatorsTest/pgSchema.ts

import {
	boolean,
	date,
	integer,
	interval,
	json,
	line,
	pgSchema,
	point,
	real,
	text,
	time,
	timestamp,
	uuid,
	varchar,
} from 'drizzle-orm/pg-core';

export const schema = pgSchema('seeder_lib_pg');

export const moodEnum = schema.enum('enum', ['sad', 'ok', 'happy']);

export const enumTable = schema.table('enum_table', {
	mood: moodEnum('mood_enum'),
});

export const defaultTable = schema.table('default_table', {
	defaultString: text('default_string'),
});

export const defaultArrayTable = schema.table('default_array_table', {
	defaultString: text('default_string').array(),
});

export const valuesFromArrayTable = schema.table('values_from_array_table', {
	valuesFromArrayNotNull: varchar('values_from_array_not_null', { length: 256 }).notNull(),
	valuesFromArrayWeightedNotNull: varchar('values_from_array_weighted_not_null', { length: 256 }).notNull(),
});

export const valuesFromArrayUniqueTable = schema.table('values_from_array_unique_table', {
	valuesFromArray: varchar('values_from_array', { length: 256 }).unique(),
	valuesFromArrayNotNull: varchar('values_from_array_not_null', { length: 256 }).unique().notNull(),
	valuesFromArrayWeighted: varchar('values_from_array_weighted', { length: 256 }).unique(),
	valuesFromArrayWeightedNotNull: varchar('values_from_array_weighted_not_null', { length: 256 }).unique().notNull(),
});

export const valuesFromArrayArrayTable = schema.table('values_from_array_array_table', {
	valuesFromArray: varchar('values_from_array', { length: 256 }).array(),
});

export const intPrimaryKeyTable = schema.table('int_primary_key_table', {
	intPrimaryKey: integer('int_primary_key').unique(),
});

export const numberTable = schema.table('number_table', {
	number: real('number'),
});

export const numberUniqueTable = schema.table('number_unique_table', {
	numberUnique: real('number_unique').unique(),
});

export const numberArrayTable = schema.table('number_array_table', {
	number: real('number').array(),
});

export const intTable = schema.table('int_table', {
	int: integer('int'),
});

export const intUniqueTable = schema.table('int_unique_table', {
	intUnique: integer('int_unique').unique(),
});

export const intArrayTable = schema.table('int_array_table', {
	int: integer('int').array(),
});

export const booleanTable = schema.table('boolean_table', {
	boolean: boolean('boolean'),
});

export const booleanArrayTable = schema.table('boolean_array_table', {
	boolean: boolean('boolean').array(),
});

export const dateTable = schema.table('date_table', {
	date: date('date'),
});

// TODO: add tests for data type with different modes
export const dateArrayTable = schema.table('date_array_table', {
	date: date('date', { mode: 'date' }).array(),
	dateString: date('date_string', { mode: 'string' }).array(),
});

export const timeTable = schema.table('time_table', {
	time: time('time'),
});

export const timeArrayTable = schema.table('time_array_table', {
	time: time('time').array(),
});

export const timestampTable = schema.table('timestamp_table', {
	timestamp: timestamp('timestamp'),
});

export const timestampArrayTable = schema.table('timestamp_array_table', {
	timestamp: timestamp('timestamp').array(),
});

export const jsonTable = schema.table('json_table', {
	json: json('json'),
});

export const jsonArrayTable = schema.table('json_array_table', {
	json: json('json').array(),
});

export const intervalTable = schema.table('interval_table', {
	interval: interval('interval'),
});

export const intervalUniqueTable = schema.table('interval_unique_table', {
	intervalUnique: interval('interval_unique').unique(),
});

export const intervalArrayTable = schema.table('interval_array_table', {
	interval: interval('interval').array(),
});

export const stringTable = schema.table('string_table', {
	string: text('string'),
});

export const stringUniqueTable = schema.table('string_unique_table', {
	stringUnique: varchar('string_unique', { length: 256 }).unique(),
});

export const stringArrayTable = schema.table('string_array_table', {
	string: text('string').array(),
});

export const emailTable = schema.table('email_table', {
	email: varchar('email', { length: 256 }).unique(),
});

export const emailArrayTable = schema.table('email_array_table', {
	email: varchar('email', { length: 256 }).array(),
});

export const firstNameTable = schema.table('first_name_table', {
	firstName: varchar('first_name', { length: 256 }),
});

export const firstNameUniqueTable = schema.table('first_name_unique_table', {
	firstNameUnique: varchar('first_name_unique', { length: 256 }).unique(),
});

export const firstNameArrayTable = schema.table('first_name_array_table', {
	firstName: varchar('first_name', { length: 256 }).array(),
});

export const lastNameTable = schema.table('last_name_table', {
	lastName: varchar('last_name', { length: 256 }),
});

export const lastNameUniqueTable = schema.table('last_name_unique_table', {
	lastNameUnique: varchar('last_name_unique', { length: 256 }).unique(),
});

export const lastNameArrayTable = schema.table('last_name_array_table', {
	lastName: varchar('last_name', { length: 256 }).array(),
});

export const fullNameTable = schema.table('full_name__table', {
	fullName: varchar('full_name_', { length: 256 }),
});

export const fullNameUniqueTable = schema.table('full_name_unique_table', {
	fullNameUnique: varchar('full_name_unique', { length: 256 }).unique(),
});

export const fullNameArrayTable = schema.table('full_name_array_table', {
	fullName: varchar('full_name', { length: 256 }).array(),
});

export const countryTable = schema.table('country_table', {
	country: varchar('country', { length: 256 }),
});

export const countryUniqueTable = schema.table('country_unique_table', {
	countryUnique: varchar('country_unique', { length: 256 }).unique(),
});

export const countryArrayTable = schema.table('country_array_table', {
	country: varchar('country', { length: 256 }).array(),
});

export const cityTable = schema.table('city_table', {
	city: varchar('city', { length: 256 }),
});

export const cityUniqueTable = schema.table('city_unique_table', {
	cityUnique: varchar('city_unique', { length: 256 }).unique(),
});

export const cityArrayTable = schema.table('city_array_table', {
	city: varchar('city', { length: 256 }).array(),
});

export const streetAddressTable = schema.table('street_address_table', {
	streetAddress: varchar('street_address', { length: 256 }),
});

export const streetAddressUniqueTable = schema.table('street_address_unique_table', {
	streetAddressUnique: varchar('street_address_unique', { length: 256 }).unique(),
});

export const streetAddressArrayTable = schema.table('street_address_array_table', {
	streetAddress: varchar('street_address', { length: 256 }).array(),
});

export const jobTitleTable = schema.table('job_Title_table', {
	jobTitle: text('job_title'),
});

export const jobTitleArrayTable = schema.table('job_title_array_table', {
	jobTitle: text('job_title').array(),
});

export const postcodeTable = schema.table('postcode_table', {
	postcode: varchar('postcode', { length: 256 }),
});

export const postcodeUniqueTable = schema.table('postcode_unique_table', {
	postcodeUnique: varchar('postcode_unique', { length: 256 }).unique(),
});

export const postcodeArrayTable = schema.table('postcode_array_table', {
	postcode: varchar('postcode', { length: 256 }).array(),
});

export const stateTable = schema.table('state_table', {
	state: text('state'),
});

export const stateArrayTable = schema.table('state_array_table', {
	state: text('state').array(),
});

export const companyNameTable = schema.table('company_name_table', {
	companyName: text('company_name'),
});

export const companyNameUniqueTable = schema.table('company_name_unique_table', {
	companyNameUnique: varchar('company_name_unique', { length: 256 }).unique(),
});

export const companyNameArrayTable = schema.table('company_name_array_table', {
	companyName: text('company_name').array(),
});

export const loremIpsumTable = schema.table('lorem_ipsum_table', {
	loremIpsum: text('lorem_ipsum'),
});

export const loremIpsumArrayTable = schema.table('lorem_ipsum_array_table', {
	loremIpsum: text('lorem_ipsum').array(),
});

export const pointTable = schema.table('point_table', {
	point: point('point'),
});

export const pointArrayTable = schema.table('point_array_table', {
	point: point('point').array(),
});

export const lineTable = schema.table('line_table', {
	line: line('line'),
});

export const lineArrayTable = schema.table('line_array_table', {
	line: line('line').array(),
});

// export const pointUniqueTable = schema.table("point_unique_table", {
//     pointUnique: point("point_unique").unique(),
// });

// export const lineUniqueTable = schema.table("line_unique_table", {
//     lineUnique: line("line_unique").unique(),
// });

export const phoneNumberTable = schema.table('phone_number_table', {
	phoneNumber: varchar('phoneNumber', { length: 256 }).unique(),
	phoneNumberTemplate: varchar('phone_number_template', { length: 256 }).unique(),
	phoneNumberPrefixes: varchar('phone_number_prefixes', { length: 256 }).unique(),
});

export const phoneNumberArrayTable = schema.table('phone_number_array_table', {
	phoneNumber: varchar('phoneNumber', { length: 256 }).array(),
	phoneNumberTemplate: varchar('phone_number_template', { length: 256 }).array(),
	phoneNumberPrefixes: varchar('phone_number_prefixes', { length: 256 }).array(),
});

export const weightedRandomTable = schema.table('weighted_random_table', {
	weightedRandom: varchar('weighted_random', { length: 256 }),
});

export const weightedRandomWithUniqueGensTable = schema.table('weighted_random_with_unique_gens_table', {
	weightedRandomWithUniqueGens: varchar('weighted_random_with_unique_gens', { length: 256 }).unique(),
});

export const uuidTable = schema.table('uuid_table', {
	uuid: uuid('uuid'),
});

export const uuidArrayTable = schema.table('uuid_array_table', {
	uuid: uuid('uuid').array(),
});