????
Your IP : 216.73.216.143
| Current Path : C:/opt/pgsql/utils/ |
|
|
| Current File : C:/opt/pgsql/utils/create_unpackaged.pl |
#!/usr/bin/env perl
#
# PostGIS - Spatial Types for PostgreSQL
# http://postgis.net
#
# Copyright (C) 2013-2022 Sandro Santilli <strk@kbt.io>
#
# This is free software; you can redistribute and/or modify it under
# the terms of the GNU General Public Licence. See the COPYING file.
#
use warnings;
use strict;
use POSIX 'strftime';
eval "exec perl -w $0 $@"
if (0);
die "Usage: perl $0 <extname> [<sql>]\n"
. " Prints SQL to add objects created by given SQL\n"
. " to extension with given name.\n"
unless @ARGV;
my $extname = shift(@ARGV);
my $scriptname = @ARGV ? $ARGV[0] : '-';
# drops are in the following order:
# 1. Indexing system stuff
# 2. Meta datatables <not done>
# 3. Aggregates
# 3. Casts
# 4. Operators
# 5. Functions
# 6. Types
# 7. Tables
my @aggs = ();
my @casts = ();
my @funcs = ();
my @types = ();
my %type_funcs = ();
my @ops = ();
my @opcs = ();
my @views = ();
my @tables = ();
my @sequences = ();
my @schemas = ();
sub strip_default {
my $line = shift;
# strip quotes first
$line =~ s/'[^']*'//ig;
# drop default then
$line =~ s/DEFAULT [^,)]*//ig;
return $line;
}
while( my $line = <>)
{
if ($line =~ /^create (or replace )?function/i) {
my $defn = $line;
while( not $defn =~ /\)/ ) {
$defn .= <>;
}
push (@funcs, $defn)
}
elsif ($line =~ /^create or replace view\s*(\w+)/i) {
push (@views, $1);
}
elsif ($line =~ /^create table \s*([\w\.]+)/i) {
#print STDERR "XXX table $1\n";
my $fqtn = $1;
push (@tables, $fqtn);
my $defn = $line;
while( not $defn =~ /\)/ ) {
#print STDERR "XXX defn $defn\n";
if ($defn =~ /([\w]+) serial\b/i) {
my $seq = "${fqtn}_$1_seq";
#print STDERR "XXX serial field [$seq]\n";
push (@sequences, $seq);
}
$defn = <>;
}
}
elsif ($line =~ /^create schema \s*([\w\.]+)/i) {
push (@schemas, $1);
}
elsif ( $line =~ /^create operator class (\w+)/i ) {
my $opcname = $1;
my $am = '';
while( not $line =~ /;\s*$/ ) {
if ( $line =~ /( USING (\w+))/ ) {
$am = $1;
last;
}
$line .= <>;
}
if ( $am eq '' ) {
die "Couldn't parse CREATE OPERATOR CLASS $opcname\n";
} else {
$opcname .= $am;
}
push (@opcs, $opcname)
}
elsif ($line =~ /^create operator.*\(/i) {
my $defn = $line;
while( not $defn =~ /;\s*$/ ) {
$defn .= <>;
}
push (@ops, $defn)
}
elsif ($line =~ /^create aggregate/i) {
my $defn = $line;
while( not $defn =~ /;\s*$/ ) {
$defn .= <>;
}
push (@aggs, $defn)
}
elsif ($line =~ /^create type ([\w\.]+)/i) {
push (@types, $1);
while( not $line =~ /;\s*$/ ) {
$line = <>;
if ( $line =~ /(input|output|send|receive|typmod_in|typmod_out|analyze)\s*=\s*(\w+)/ ) {
my $role = ${1};
my $fname = ${2};
$type_funcs{$fname} = $role;
}
}
}
elsif ($line =~ /^create domain ([\w\.]+)/i) {
push (@types, $1);
}
elsif ($line =~ /^create cast/i) {
push (@casts, $line)
}
}
#close( INPUT );
sub add_if_not_exists
{
my $obj = shift;
# Prevent troubles by refusing to accept single quotes
# in objects
die "Invalid characters in object definition: $obj" if $obj =~ /'/;
$obj =~ m/([^ ]*) (.*)/s; # can be multiline
my $type = $1;
my $sig = $2;
print "SELECT _postgis_package_object('$type', '$sig');\n";
}
my $time = POSIX::strftime("%F %T", gmtime(defined($ENV{SOURCE_DATE_EPOCH}) ? $ENV{SOURCE_DATE_EPOCH} : time));
print <<"EOF";
-- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- --
--
-- PostGIS - Spatial Types for PostgreSQL
-- http://postgis.net
--
-- This is free software; you can redistribute and/or modify it under
-- the terms of the GNU General Public Licence. See the COPYING file.
--
-- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- --
--
-- Generated on: $time
-- by: $0
-- for: $extname
-- from: $scriptname
--
-- Do not edit manually, your changes will be lost.
--
-- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- --
-- complain if script is sourced in psql
\\echo Use "CREATE EXTENSION ${extname} to load this file. \\quit
CREATE FUNCTION _postgis_package_object(type text, sig text)
RETURNS VOID
AS \$\$
DECLARE
sql text;
proc regproc;
obj text := format('%s %s', type, sig);
BEGIN
sql := format('ALTER EXTENSION ${extname} ADD %s', obj);
EXECUTE sql;
RAISE NOTICE 'newly registered %', obj;
EXCEPTION
WHEN object_not_in_prerequisite_state THEN
IF SQLERRM ~ '\\m${extname}\\M'
THEN
RAISE NOTICE '% already registered', obj;
ELSE
RAISE EXCEPTION '%', SQLERRM;
END IF;
WHEN
undefined_function OR
undefined_table OR
undefined_object
-- TODO: handle more exceptions ?
THEN
RAISE NOTICE '% % does not exist yet', type, sig;
WHEN OTHERS THEN
RAISE EXCEPTION 'Trying to add % to ${extname}, got % (%)', obj, SQLERRM, SQLSTATE;
END;
\$\$ LANGUAGE 'plpgsql';
EOF
print "-- Register all views.\n";
foreach my $view (@views)
{
add_if_not_exists("VIEW $view");
}
print "-- Register all tables.\n";
# we reverse table definitions so foreign key constraints
# are more likely not to get in our way
@tables = reverse(@tables);
foreach my $table (@tables)
{
add_if_not_exists("TABLE $table");
}
print "-- Register all sequences.\n";
foreach my $seq (@sequences)
{
add_if_not_exists("SEQUENCE $seq");
}
print "-- Register all aggregates.\n";
foreach my $agg (@aggs)
{
if ( $agg =~ /create aggregate\s*([\w\.]+)\s*\(\s*.*basetype = ([\w\.]+)/ism )
{
add_if_not_exists("AGGREGATE $1 ($2)");
}
elsif ( $agg =~ /create aggregate\s*([\w\.]+)\s*\(\s*([\w,\.\s\[\]]+)\s*\)/ism )
{
add_if_not_exists("AGGREGATE $1 ($2)");
}
else
{
die "Couldn't parse AGGREGATE line: $agg\n";
}
}
print "-- Register all operators classes and families.\n";
foreach my $opc (@opcs)
{
add_if_not_exists("OPERATOR CLASS $opc");
add_if_not_exists("OPERATOR FAMILY $opc");
}
print "-- Register all operators.\n";
foreach my $op (@ops)
{
if ($op =~ /create operator ([^(]+)\s*\(.*LEFTARG\s*=\s*(\w+),\s*RIGHTARG\s*=\s*(\w+).*/ism )
{
add_if_not_exists("OPERATOR $1 ($2,$3)");
}
else
{
die "Couldn't parse OPERATOR line: $op\n";
}
}
print "-- Register all casts.\n";
foreach my $cast (@casts)
{
if ($cast =~ /create cast\s*\((.+?)\)/i )
{
add_if_not_exists("CAST ($1)");
}
else
{
die "Couldn't parse CAST line: $cast\n";
}
}
print "-- Register all functions.\n";
foreach my $fn ( @funcs )
{
if ($fn =~ /.* function ([^(]+)\((.*)\)/is ) # can be multiline
{
my $fn_nm = $1;
my $fn_arg = $2;
$fn_arg = strip_default($fn_arg);
add_if_not_exists("FUNCTION $fn_nm ($fn_arg)");
}
else
{
die "Couldn't parse FUNCTION line: $fn\n";
}
}
print "-- Register all types.\n";
foreach my $type (@types)
{
add_if_not_exists("TYPE $type");
}
# NOTE: cannot add schema "topology" to extension "postgis_topology"
# because the schema contains the extension
#
#print "-- Register all schemas.\n";
#if (@schemas)
#{
# foreach my $schema (@schemas)
# {
# add_if_not_exists("SCHEMA \"$schema\"");
# }
#}
print <<"EOF";
DROP FUNCTION _postgis_package_object(text, text);
-- Security checks
DO LANGUAGE 'plpgsql' \$BODY\$
DECLARE
rec RECORD;
BEGIN
-- Check ownership of extension functions
-- matches ownership of extension itself
FOR rec IN
SELECT
p.oid,
p.proowner,
e.extowner
FROM pg_catalog.pg_depend AS d
INNER JOIN pg_catalog.pg_extension AS e ON (d.refobjid = e.oid)
INNER JOIN pg_catalog.pg_proc AS p ON (d.objid = p.oid)
WHERE d.refclassid = 'pg_catalog.pg_extension'::pg_catalog.regclass
AND deptype = 'e'
AND e.extname = '${extname}'
AND d.classid = 'pg_catalog.pg_proc'::pg_catalog.regclass
AND p.proowner != e.extowner
LOOP
RAISE EXCEPTION 'Function % is owned by % but extension is owned by %',
rec.oid::regprocedure, rec.proowner::regrole, rec.extowner::regrole;
END LOOP;
-- TODO: check ownership of more objects ?
END;
\$BODY\$;
EOF
print "\n";
1;