# Copyright (c) 2021-2025, PostgreSQL Global Development Group

use strict;
use warnings FATAL => 'all';

use PostgreSQL::Test::Cluster;
use PostgreSQL::Test::Utils;
use Test::More;

program_help_ok('createdb');
program_version_ok('createdb');
program_options_handling_ok('createdb');

my $node = PostgreSQL::Test::Cluster->new('main');
$node->init;
$node->start;

$node->issues_sql_like(
	[ 'createdb', 'foobar1' ],
	qr/statement: CREATE DATABASE foobar1/,
	'SQL CREATE DATABASE run');
$node->issues_sql_like(
	[
		'createdb',
		'--locale' => 'C',
		'--encoding' => 'LATIN1',
		'--template' => 'template0',
		'foobar2',
	],
	qr/statement: CREATE DATABASE foobar2 ENCODING 'LATIN1'/,
	'create database with encoding');

if ($ENV{with_icu} eq 'yes')
{
	# This fails because template0 uses libc provider and has no ICU
	# locale set.  It would succeed if template0 used the icu
	# provider.  XXX Maybe split into multiple tests?
	$node->command_fails(
		[
			'createdb',
			'--template' => 'template0',
			'--encoding' => 'UTF8',
			'--locale-provider' => 'icu',
			'foobar4',
		],
		'create database with ICU fails without ICU locale specified');

	$node->issues_sql_like(
		[
			'createdb',
			'--template' => 'template0',
			'--encoding' => 'UTF8',
			'--locale-provider' => 'icu',
			'--locale' => 'C',
			'--icu-locale' => 'en',
			'foobar5',
		],
		qr/statement: CREATE DATABASE foobar5 .* LOCALE_PROVIDER icu ICU_LOCALE 'en'/,
		'create database with ICU locale specified');

	$node->command_fails(
		[
			'createdb',
			'--template' => 'template0',
			'--encoding' => 'UTF8',
			'--locale-provider' => 'icu',
			'--icu-locale' => '@colNumeric=lower',
			'foobarX',
		],
		'fails for invalid ICU locale');

	$node->command_fails_like(
		[
			'createdb',
			'--template' => 'template0',
			'--locale-provider' => 'icu',
			'--encoding' => 'SQL_ASCII',
			'foobarX',
		],
		qr/ERROR:  encoding "SQL_ASCII" is not supported with ICU provider/,
		'fails for encoding not supported by ICU');

	# additional node, which uses the icu provider
	my $node2 = PostgreSQL::Test::Cluster->new('icu');
	$node2->init(extra => [ '--locale-provider=icu', '--icu-locale=en' ]);
	$node2->start;

	$node2->command_ok(
		[
			'createdb',
			'--template' => 'template0',
			'--locale-provider' => 'libc',
			'foobar55',
		],
		'create database with libc provider from template database with icu provider'
	);

	$node2->command_ok(
		[
			'createdb',
			'--template' => 'template0',
			'--icu-locale' => 'en-US',
			'foobar56',
		],
		'create database with icu locale from template database with icu provider'
	);

	$node2->command_ok(
		[
			'createdb',
			'--template' => 'template0',
			'--locale-provider' => 'icu',
			'--locale' => 'en',
			'--lc-collate' => 'C',
			'--lc-ctype' => 'C',
			'foobar57',
		],
		'create database with locale as ICU locale');
}
else
{
	$node->command_fails(
		[
			'createdb',
			'--template' => 'template0',
			'--locale-provider' => 'icu',
			'foobar4',
		],
		'create database with ICU fails since no ICU support');
}

$node->command_fails(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'tbuiltin1',
	],
	'create database with provider "builtin" fails without --locale');

$node->command_ok(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'--locale' => 'C',
		'tbuiltin2',
	],
	'create database with provider "builtin" and locale "C"');

$node->command_ok(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'--locale' => 'C',
		'--lc-collate' => 'C',
		'tbuiltin3',
	],
	'create database with provider "builtin" and LC_COLLATE=C');

$node->command_ok(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'--locale' => 'C',
		'--lc-ctype' => 'C',
		'tbuiltin4',
	],
	'create database with provider "builtin" and LC_CTYPE=C');

$node->command_ok(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'--lc-collate' => 'C',
		'--lc-ctype' => 'C',
		'--encoding' => 'UTF-8',
		'--builtin-locale' => 'C.UTF8',
		'tbuiltin5',
	],
	'create database with --builtin-locale C.UTF-8 and -E UTF-8');

$node->command_fails(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'--lc-collate' => 'C',
		'--lc-ctype' => 'C',
		'--encoding' => 'LATIN1',
		'--builtin-locale' => 'C.UTF-8',
		'tbuiltin6',
	],
	'create database with --builtin-locale C.UTF-8 and -E LATIN1');

$node->command_fails(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'--locale' => 'C',
		'--icu-locale' => 'en',
		'tbuiltin7',
	],
	'create database with provider "builtin" and ICU_LOCALE="en"');

$node->command_fails(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'builtin',
		'--locale' => 'C',
		'--icu-rules' => '""',
		'tbuiltin8',
	],
	'create database with provider "builtin" and ICU_RULES=""');

$node->command_fails(
	[
		'createdb',
		'--template' => 'template1',
		'--locale-provider' => 'builtin',
		'--locale' => 'C',
		'tbuiltin9',
	],
	'create database with provider "builtin" not matching template');

$node->command_fails([ 'createdb', 'foobar1' ],
	'fails if database already exists');

$node->command_fails(
	[
		'createdb',
		'--template' => 'template0',
		'--locale-provider' => 'xyz',
		'foobarX',
	],
	'fails for invalid locale provider');

# Check use of templates with shared dependencies copied from the template.
my ($ret, $stdout, $stderr) = $node->psql(
	'foobar2',
	'CREATE ROLE role_foobar;
CREATE TABLE tab_foobar (id int);
ALTER TABLE tab_foobar owner to role_foobar;
CREATE POLICY pol_foobar ON tab_foobar FOR ALL TO role_foobar;');
$node->issues_sql_like(
	[ 'createdb', '--locale' => 'C', '--template' => 'foobar2', 'foobar3' ],
	qr/statement: CREATE DATABASE foobar3 TEMPLATE foobar2 LOCALE 'C'/,
	'create database with template');
($ret, $stdout, $stderr) = $node->psql(
	'foobar3',
	"SELECT pg_describe_object(classid, objid, objsubid) AS obj,
       pg_describe_object(refclassid, refobjid, 0) AS refobj
   FROM pg_shdepend s JOIN pg_database d ON (d.oid = s.dbid)
   WHERE d.datname = 'foobar3' ORDER BY obj;", on_error_die => 1);
chomp($stdout);
like(
	$stdout,
	qr/^policy pol_foobar on table tab_foobar\|role role_foobar
table tab_foobar\|role role_foobar$/,
	'shared dependencies copied over to target database');

# Check quote handling with incorrect option values.
$node->command_checks_all(
	[ 'createdb', '--encoding', "foo'; SELECT '1", 'foobar2' ],
	1,
	[qr/^$/],
	[qr/^createdb: error: "foo'; SELECT '1" is not a valid encoding name/s],
	'createdb with incorrect --encoding');
$node->command_checks_all(
	[ 'createdb', '--lc-collate', "foo'; SELECT '1", 'foobar2' ],
	1,
	[qr/^$/],
	[
		qr/^createdb: error: database creation failed: ERROR:  invalid LC_COLLATE locale name|^createdb: error: database creation failed: ERROR:  new collation \(foo'; SELECT '1\) is incompatible with the collation of the template database/s,
	],
	'createdb with incorrect --lc-collate');
$node->command_checks_all(
	[ 'createdb', '--lc-ctype', "foo'; SELECT '1", 'foobar2' ],
	1,
	[qr/^$/],
	[
		qr/^createdb: error: database creation failed: ERROR:  invalid LC_CTYPE locale name|^createdb: error: database creation failed: ERROR:  new LC_CTYPE \(foo'; SELECT '1\) is incompatible with the LC_CTYPE of the template database/s,
	],
	'createdb with incorrect --lc-ctype');

$node->command_checks_all(
	[ 'createdb', '--strategy', "foo", 'foobar2' ],
	1,
	[qr/^$/],
	[
		qr/^createdb: error: database creation failed: ERROR:  invalid create database strategy "foo"/s,
	],
	'createdb with incorrect --strategy');

# Check database creation strategy
$node->issues_sql_like(
	[
		'createdb',
		'--template' => 'foobar2',
		'--strategy' => 'wal_log',
		'foobar6',
	],
	qr/statement: CREATE DATABASE foobar6 STRATEGY wal_log TEMPLATE foobar2/,
	'create database with WAL_LOG strategy');

$node->issues_sql_like(
	[
		'createdb',
		'--template' => 'foobar2',
		'--strategy' => 'WAL_LOG',
		'foobar6s',
	],
	qr/statement: CREATE DATABASE foobar6s STRATEGY "WAL_LOG" TEMPLATE foobar2/,
	'create database with WAL_LOG strategy');

$node->issues_sql_like(
	[
		'createdb',
		'--template' => 'foobar2',
		'--strategy' => 'file_copy',
		'foobar7',
	],
	qr/statement: CREATE DATABASE foobar7 STRATEGY file_copy TEMPLATE foobar2/,
	'create database with FILE_COPY strategy');

$node->issues_sql_like(
	[
		'createdb',
		'--template' => 'foobar2',
		'--strategy' => 'FILE_COPY',
		'foobar7s',
	],
	qr/statement: CREATE DATABASE foobar7s STRATEGY "FILE_COPY" TEMPLATE foobar2/,
	'create database with FILE_COPY strategy');

# Create database owned by role_foobar.
$node->issues_sql_like(
	[
		'createdb',
		'--template' => 'foobar2',
		'--owner' => 'role_foobar',
		'foobar8',
	],
	qr/statement: CREATE DATABASE foobar8 OWNER role_foobar TEMPLATE foobar2/,
	'create database with owner role_foobar');
($ret, $stdout, $stderr) =
  $node->psql('foobar2', 'DROP OWNED BY role_foobar;', on_error_die => 1,);
ok($ret == 0, "DROP OWNED BY role_foobar");
($ret, $stdout, $stderr) =
  $node->psql('foobar2', 'DROP DATABASE foobar8;', on_error_die => 1,);
ok($ret == 0, "DROP DATABASE foobar8");

done_testing();