#include "pgsql.h"

#include "QorePGConnection.h"

#ifndef PG_TYPE_RELTYPE_OID

#define PG_TYPE_RELTYPE_OID 0

#endif

#ifndef PG_ATTRIBUTE_RELTYPE_OID

#define PG_ATTRIBUTE_RELTYPE_OID 0

#endif

#ifndef PG_PROC_RELTYPE_OID

#define PG_PROC_RELTYPE_OID 0

#endif

#ifndef PG_CLASS_RELTYPE_OID

#define PG_CLASS_RELTYPE_OID 0

#endif

#ifdef HAVE_PQLIBVERSION

#define PGSQL_HAVE_GETCLIENTVERSION 1

#else

#define PGSQL_HAVE_GETCLIENTVERSION 0

#endif

namespace Qore::Pgsql;

const PGSQL_HAVE_GETCLIENTVERSION = bool(PGSQL_HAVE_GETCLIENTVERSION);

namespace Qore::Pgsql;

const PG_TYPE_BOOL = BOOLOID;

const PG_TYPE_BYTEA = BYTEAOID;

const PG_TYPE_CHAR = CHAROID;

const PG_TYPE_NAME = NAMEOID;

const PG_TYPE_INT8 = INT8OID;

const PG_TYPE_INT2 = INT2OID;

const PG_TYPE_INT2VECTOR = INT2VECTOROID;

const PG_TYPE_INT4 = INT4OID;

const PG_TYPE_REGPROC = REGPROCOID;

const PG_TYPE_TEXT = TEXTOID;

const PG_TYPE_OID = OIDOID;

const PG_TYPE_TID = TIDOID;

const PG_TYPE_XID = XIDOID;

const PG_TYPE_CID = CIDOID;

const PG_TYPE_VECTOROID = OIDVECTOROID;

const PG_TYPE_TYPE_RELTYPE = PG_TYPE_RELTYPE_OID;

const PG_TYPE_ATTRIBUTE_RELTYPE = PG_ATTRIBUTE_RELTYPE_OID;

const PG_TYPE_PROC_RELTYPE = PG_PROC_RELTYPE_OID;

const PG_TYPE_CLASS_RELTYPE = PG_CLASS_RELTYPE_OID;

const PG_TYPE_POINT = POINTOID;

const PG_TYPE_LSEG = LSEGOID;

const PG_TYPE_PATH = PATHOID;

const PG_TYPE_BOX = BOXOID;

const PG_TYPE_POLYGON = POLYGONOID;

const PG_TYPE_LINE = LINEOID;

const PG_TYPE_FLOAT4 = FLOAT4OID;

const PG_TYPE_FLOAT8 = FLOAT8OID;

const PG_TYPE_ABSTIME = ABSTIMEOID;

const PG_TYPE_RELTIME = RELTIMEOID;

const PG_TYPE_TINTERVAL = TINTERVALOID;

const PG_TYPE_UNKNOWN = UNKNOWNOID;

const PG_TYPE_CIRCLE = CIRCLEOID;

const PG_TYPE_CASH = CASHOID;

const PG_TYPE_MACADDR = MACADDROID;

const PG_TYPE_INET = INETOID;

const PG_TYPE_CIDR = CIDROID;

const PG_TYPE_ACLITEM = ACLITEMOID;

const PG_TYPE_BPCHAR = BPCHAROID;

const PG_TYPE_VARCHAR = VARCHAROID;

const PG_TYPE_DATE = DATEOID;

const PG_TYPE_TIME = TIMEOID;

const PG_TYPE_TIMESTAMP = TIMESTAMPOID;

const PG_TYPE_TIMESTAMPTZ = TIMESTAMPTZOID;

const PG_TYPE_INTERVAL = INTERVALOID;

const PG_TYPE_TIMETZ = TIMETZOID;

const PG_TYPE_BIT = BITOID;

const PG_TYPE_VARBIT = VARBITOID;

const PG_TYPE_NUMERIC = NUMERICOID;

const PG_TYPE_REFCURSOR = REFCURSOROID;

const PG_TYPE_REGPROCEDURE = REGPROCEDUREOID;

const PG_TYPE_REGOPER = REGOPEROID;

const PG_TYPE_REGOPERATOR = REGOPERATOROID;

const PG_TYPE_REGCLASS = REGCLASSOID;

const PG_TYPE_REGTYPE = REGTYPEOID;

const PG_TYPE_RECORD = RECORDOID;

const PG_TYPE_CSTRING = CSTRINGOID;

const PG_TYPE_ANY = ANYOID;

const PG_TYPE_VOID = VOIDOID;

const PG_TYPE_TRIGGER = TRIGGEROID;

const PG_TYPE_LANGUAGE_HANDLER = LANGUAGE_HANDLEROID;

const PG_TYPE_INTERNAL = INTERNALOID;

const PG_TYPE_OPAQUE = OPAQUEOID;

const PG_TYPE_ANYELEMENT = ANYELEMENTOID;

const PG_TYPE_XML = XMLOID;

const PG_TYPE_JSON = JSONOID;

const PG_TYPE_JSONB = JSONBOID;

const PG_TYPE_INT4ARRAY = QPGT_INT4ARRAYOID;

const PG_TYPE_CIRCLEARRAY = QPGT_CIRCLEARRAYOID;

const PG_TYPE_MONEYARRAY = QPGT_MONEYARRAYOID;

const PG_TYPE_BOOLARRAY = QPGT_BOOLARRAYOID;

const PG_TYPE_BYTEAARRAY = QPGT_BYTEAARRAYOID;

const PG_TYPE_NAMEARRAY = QPGT_NAMEARRAYOID;

const PG_TYPE_INT2ARRAY = QPGT_INT2ARRAYOID;

const PG_TYPE_TEXTARRAY = QPGT_TEXTARRAYOID;

const PG_TYPE_OIDARRAY = QPGT_OIDARRAYOID;

const PG_TYPE_TIDARRAY = QPGT_TIDARRAYOID;

const PG_TYPE_XIDARRAY = QPGT_XIDARRAYOID;

const PG_TYPE_CIDARRAY = QPGT_CIDARRAYOID;

const PG_TYPE_BPCHARARRAY = QPGT_BPCHARARRAYOID;

const PG_TYPE_VARCHARARRAY = QPGT_VARCHARARRAYOID;

const PG_TYPE_INT8ARRAY = QPGT_INT8ARRAYOID;

const PG_TYPE_POINTARRAY = QPGT_POINTARRAYOID;

const PG_TYPE_LSEGARRAY = QPGT_LSEGARRAYOID;

const PG_TYPE_PATHARRAY = QPGT_PATHARRAYOID;

const PG_TYPE_BOXARRAY = QPGT_BOXARRAYOID;

const PG_TYPE_FLOAT4ARRAY = QPGT_FLOAT4ARRAYOID;

const PG_TYPE_FLOAT8ARRAY = QPGT_FLOAT8ARRAYOID;

const PG_TYPE_ABSTIMEARRAY = QPGT_ABSTIMEARRAYOID;

const PG_TYPE_RELTIMEARRAY = QPGT_RELTIMEARRAYOID;

const PG_TYPE_TINTERVALARRAY = QPGT_TINTERVALARRAYOID;

const PG_TYPE_POLYGONARRAY = QPGT_POLYGONARRAYOID;

const PG_TYPE_MACADDRARRAY = QPGT_MACADDRARRAYOID;

const PG_TYPE_INETARRAY = QPGT_INETARRAYOID;

const PG_TYPE_CIDRARRAY = QPGT_CIDRARRAYOID;

const PG_TYPE_TIMESTAMPARRAY = QPGT_TIMESTAMPARRAYOID;

const PG_TYPE_DATEARRAY = QPGT_DATEARRAYOID;

const PG_TYPE_TIMEARRAY = QPGT_TIMEARRAYOID;

const PG_TYPE_TIMESTAMPTZARRAY = QPGT_TIMESTAMPTZARRAYOID;

const PG_TYPE_INTERVALARRAY = QPGT_INTERVALARRAYOID;

const PG_TYPE_NUMERICARRAY = QPGT_NUMERICARRAYOID;

const PG_TYPE_TIMETZARRAY = QPGT_TIMETZARRAYOID;

const PG_TYPE_BITARRAY = QPGT_BITARRAYOID;

const PG_TYPE_VARBITARRAY = QPGT_VARBITARRAYOID;

const PG_TYPE_ANYARRAY = ANYARRAYOID;

const PG_TYPE_XMLARRAY = XMLARRAYOID;

const PG_TYPE_JSONARRAY = JSONOID;

const PG_TYPE_JSONBARRAY = JSONBARRAYOID;

const PG_TYPE_UUID = UUIDOID;

const PG_TYPE_UUIDARRAY = QPGT_UUIDARRAYOID;

namespace Qore::Pgsql;

hash pgsql_bind(softint type, auto value) [flags=RET_VALUE_ONLY] {

if (!type) {

xsink->raiseException("PGSQL-BIND-ERROR", "expecting OID (type number) as first parameter to pgsql_bind()");

return QoreValue();

}

QoreHashNode *h = new QoreHashNode;

h->setKeyValue("^pgtype^", type, xsink);

h->setKeyValue("^value^", value.refSelf(), xsink);

return h;

}

hash pgsql_bind_array(*softlist<auto> value) [flags=CONSTANT] {

QoreHashNode *h = new QoreHashNode;

h->setKeyValue("^pgarray^", true, xsink);

h->setKeyValue("^value^", value ? value->refSelf() : nullptr, xsink);

return h;

}

hash pgsql_bind_uuid(string value) [flags=RET_VALUE_ONLY] {

QoreHashNode *h = new QoreHashNode;

h->setKeyValue("^pgtype^", (int64)UUIDOID, xsink);

h->setKeyValue("^value^", value->refSelf(), xsink);

return h;

}

hash<auto> pgsql_bind_vector(*softlist<softfloat> value) [flags=RET_VALUE_ONLY] {

QoreHashNode* h = new QoreHashNode(autoTypeInfo);

h->setKeyValue("^pgtype^", new QoreStringNode("vector"), xsink);

h->setKeyValue("^value^", value && value->size() ? value->refSelf() : nullptr, xsink);

return h;

}

namespace Qore::Pgsql;

const PGSQL_SQL_LIST_TABLES = "SELECT table_name, table_type FROM information_schema.tables WHERE table_schema = current_schema() AND table_type = 'BASE TABLE' ORDER BY table_name";

const PGSQL_SQL_LIST_SCHEMAS = "SELECT schema_name, schema_owner FROM information_schema.schemata WHERE schema_name NOT LIKE 'pg_%' AND schema_name != 'information_schema' ORDER BY schema_name";

const PGSQL_SQL_TABLE_COLUMNS = "SELECT column_name, data_type, is_nullable, column_default, character_maximum_length, numeric_precision, numeric_scale FROM information_schema.columns WHERE table_schema = current_schema() AND table_name = %v ORDER BY ordinal_position";

const PGSQL_SQL_TABLE_INDEXES = "SELECT i.relname as index_name, ix.indisunique as is_unique, ix.indisprimary as is_primary, array_agg(a.attname ORDER BY array_position(ix.indkey, a.attnum)) as column_names FROM pg_class t JOIN pg_index ix ON t.oid = ix.indrelid JOIN pg_class i ON i.oid = ix.indexrelid JOIN pg_attribute a ON a.attrelid = t.oid AND a.attnum = ANY(ix.indkey) WHERE t.relname = %v AND t.relkind = 'r' GROUP BY i.relname, ix.indisunique, ix.indisprimary ORDER BY i.relname";

const PGSQL_SQL_TABLE_FOREIGN_KEYS = "SELECT tc.constraint_name, kcu.column_name, ccu.table_name AS foreign_table, ccu.column_name AS foreign_column FROM information_schema.table_constraints AS tc JOIN information_schema.key_column_usage AS kcu ON tc.constraint_name = kcu.constraint_name AND tc.table_schema = kcu.table_schema JOIN information_schema.constraint_column_usage AS ccu ON ccu.constraint_name = tc.constraint_name AND ccu.table_schema = tc.table_schema WHERE tc.constraint_type = 'FOREIGN KEY' AND tc.table_name = %v";

const PGSQL_SQL_LIST_FUNCTIONS = "SELECT p.proname as function_name, pg_get_function_result(p.oid) as result_type, pg_get_function_arguments(p.oid) as argument_types FROM pg_proc p JOIN pg_namespace n ON p.pronamespace = n.oid WHERE n.nspname = current_schema() ORDER BY p.proname";

const PGSQL_SQL_LIST_SEQUENCES = "SELECT sequence_name, data_type, start_value, increment FROM information_schema.sequences WHERE sequence_schema = current_schema() ORDER BY sequence_name";

const PGSQL_SQL_LIST_VIEWS = "SELECT table_name as view_name, view_definition FROM information_schema.views WHERE table_schema = current_schema() ORDER BY table_name";