sub
RTxInitDB {
my
(
$self
,
$action
) =
@_
;
unshift
@INC
,
substr
(
delete
(
$INC
{
'RT.pm'
}), 0, -5)
if
$INC
{
'RT.pm'
};
unshift
@INC
,
"$RT::LocalPath/lib"
if
$RT::LocalPath
;
$RT::SbinPath
||=
$RT::LocalPath
;
$RT::SbinPath
=~ s/
local
$/sbin/;
foreach
my
$file
(
$RT::CORE_CONFIG_FILE
,
$RT::SITE_CONFIG_FILE
) {
next
if
!-e
$file
or -r
$file
;
die
"No permission to read $file\n-- please re-run $0 with suitable privileges.\n"
;
}
RT::LoadConfig();
my
$lib_path
= File::Basename::dirname(
$INC
{
'RT.pm'
});
my
@args
= (
"-Ilib"
);
push
@args
,
"-I$RT::LocalPath/lib"
if
$RT::LocalPath
;
push
@args
, (
"-I$lib_path"
,
"$RT::SbinPath/rt-setup-database"
,
"--action"
=>
$action
,
"--datadir"
=>
"etc"
,
((
$action
eq
'insert'
) ? (
"--datafile"
=>
"etc/initialdata"
) : ()),
"--dba"
=>
$RT::DatabaseUser
,
"--prompt-for-dba-password"
=>
''
);
print
"$^X @args\n"
;
(
system
($^X,
@args
) == 0) or
die
"...returned with error: $?\n"
;
}
sub
RTxFactory {
my
(
$self
,
$RTx
,
$name
,
$drop
) =
@_
;
my
$namespace
=
"$RTx\::$name"
;
$self
->RTxInit;
my
$dbh
=
$RT::Handle
->dbh;
my
@tables
=
$dbh
->tables;
my
(
%tablemap
,
%typemap
,
%modulemap
);
my
$driver
=
$RT::DatabaseType
;
my
$CollectionBaseclass
=
'RT::SearchBuilder'
;
my
$RecordBaseclass
=
'RT::Record'
;
my
$LicenseBlock
= <<
'.'
;
.
my
$Attribution
= <<
'.'
;
.
my
$RecordInit
=
''
;
@tables
=
map
{
do
{ {
my
$table
=
$_
;
$table
=~ s/.*\.//g;
$table
=~ s/\W//g;
$table
=~ s/^\Q
$name
\E_//i or
next
;
$table
ne
'sessions'
or
next
;
$table
=
ucfirst
(
lc
(
$table
));
$table
=~ s/
$_
/\u
$_
/
for
qw(field group custom member value)
;
$table
=~ s/(?<=Scrip)
$_
/\u
$_
/
for
qw(action condition)
;
$table
=~ s/
$_
/\U
$_
/
for
qw(Acl)
;
$table
=
$name
.
'_'
.
$table
;
$tablemap
{
$table
} =
$table
;
$modulemap
{
$table
} =
$table
;
if
(
$table
=~ /^(.*)s$/ ) {
$tablemap
{$1} =
$table
;
$modulemap
{$1} = $1;
}
$table
;
} } }
@tables
;
$tablemap
{
'CreatedBy'
} =
'User'
;
$tablemap
{
'UpdatedBy'
} =
'User'
;
$typemap
{
'id'
} =
'ro'
;
$typemap
{
'Creator'
} =
'auto'
;
$typemap
{
'Created'
} =
'auto'
;
$typemap
{
'Updated'
} =
'auto'
;
$typemap
{
'UpdatedBy'
} =
'auto'
;
$typemap
{
'LastUpdated'
} =
'auto'
;
$typemap
{
'LastUpdatedBy'
} =
'auto'
;
$typemap
{
lc
(
$_
)} =
$typemap
{
$_
}
for
keys
%typemap
;
foreach
my
$table
(
@tables
) {
if
(
$drop
) {
$dbh
->
do
(
"DROP TABLE $table"
);
$dbh
->
do
(
"DROP sequence ${table}_id_seq"
)
if
$driver
eq
'Pg'
;
$dbh
->
do
(
"DROP sequence ${table}_seq"
)
if
$driver
eq
'Oracle'
;
next
;
}
my
$tablesingle
=
$table
;
$tablesingle
=~ s/^\Q
$name
\E_//i;
$tablesingle
=~ s/s$//;
my
$tableplural
=
$tablesingle
.
"s"
;
if
(
$tablesingle
eq
'ACL'
) {
$tablesingle
=
"ACE"
;
$tableplural
=
"ACL"
;
}
my
%requirements
;
my
$CollectionClassName
=
$namespace
.
"::"
.
$tableplural
;
my
$RecordClassName
=
$namespace
.
"::"
.
$tablesingle
;
my
$path
=
$namespace
;
$path
=~ s/::/\//g;
my
$RecordClassPath
=
$path
.
"/"
.
$tablesingle
.
".pm"
;
my
$CollectionClassPath
=
$path
.
"/"
.
$tableplural
.
".pm"
;
my
$CreateInParams
;
my
$CreateOutParams
;
my
$ClassAccessible
=
""
;
my
$FieldsPod
=
""
;
my
$CreatePod
=
""
;
my
$CreateSub
=
""
;
my
%fields
;
my
$sth
=
$dbh
->prepare(
"DESCRIBE $table"
);
if
(
$driver
eq
'Pg'
) {
$sth
=
$dbh
->prepare(
<<".");
SELECT a.attname, format_type(a.atttypid, a.atttypmod),
a.attnotnull, a.atthasdef, a.attnum
FROM pg_class c, pg_attribute a
WHERE c.relname ILIKE '$table'
AND a.attnum > 0
AND a.attrelid = c.oid
ORDER BY a.attnum
.
}
elsif
(
$driver
eq
'mysql'
) {
$sth
=
$dbh
->prepare(
"DESCRIBE $table"
);
}
else
{
die
"$driver is currently unsupported"
;
}
$sth
->execute;
while
(
my
$row
=
$sth
->fetchrow_hashref() ) {
my
(
$field
,
$type
,
$default
);
if
(
$driver
eq
'Pg'
) {
$field
=
$row
->{
'attname'
};
$type
=
$row
->{
'format_type'
};
$default
=
$row
->{
'atthasdef'
};
if
(
$default
!= 0 ) {
my
$tth
=
$dbh
->prepare(
<<".");
SELECT substring(d.adsrc for 128)
FROM pg_attrdef d, pg_class c
WHERE c.relname = 'acct'
AND c.oid = d.adrelid
AND d.adnum = $row->{'attnum'}
.
$tth
->execute();
my
@default
=
$tth
->fetchrow_array;
$default
=
$default
[0];
}
}
elsif
(
$driver
eq
'mysql'
) {
$field
=
$row
->{
'Field'
};
$type
=
$row
->{
'Type'
};
$default
=
$row
->{
'Default'
};
}
$fields
{
$field
} = 1;
if
(
$typemap
{
$field
} eq
'auto'
) {
$ClassAccessible
.= "
$field
=>
{
read
=> 1,
auto
=> 1,";
}
elsif
(
$typemap
{
$field
} eq
'ro'
) {
$ClassAccessible
.= "
$field
=>
{
read
=> 1,";
}
else
{
$ClassAccessible
.= "
$field
=>
{
read
=> 1,
write
=> 1,";
}
$ClassAccessible
.=
" type => '$type', default => '$default'},\n"
;
$FieldsPod
.=
$self
->_pod(
<<".");
^head2 $field
Returns the current value of $field.
(In the database, $field is stored as $type.)
.
unless
(
$typemap
{
$field
} eq
'auto'
||
$typemap
{
$field
} eq
'ro'
) {
$FieldsPod
.=
$self
->_pod(
<<".");
^head2 Set$field VALUE
Set $field to VALUE.
Returns (1, 'Status message') on success and (0, 'Error Message') on failure.
(In the database, $field will be stored as a $type.)
.
}
$FieldsPod
.=
$self
->_pod(
<<".");
^cut
.
if
(
$modulemap
{
$field
} ) {
$FieldsPod
.=
$self
->_pod(
<<".");
^head2 ${field}Obj
Returns the $modulemap{$field} Object which has the id returned by $field
^cut
sub ${field}Obj {
my \$self = shift;
my \$$field = ${namespace}::$modulemap{$field}->new(\$self->CurrentUser);
\$$field->Load(\$self->__Value('$field'));
return(\$$field);
}
.
$requirements
{
$tablemap
{
$field
} } =
"use ${namespace}::$modulemap{$field};"
;
}
unless
(
$typemap
{
$field
} eq
'auto'
||
$field
eq
'id'
) {
$CreateInParams
.=
" $field => '$default',\n"
;
$CreateOutParams
.=
" $field => \$args{'$field'},\n"
;
$CreatePod
.=
" $type '$field'"
;
$CreatePod
.=
" defaults to '$default'"
if
(
$default
);
$CreatePod
.=
".\n"
;
}
}
$CreateSub
=
<<".";
sub Create {
my \$self = shift;
my \%args = (
$CreateInParams
\@_);
\$self->SUPER::Create(
$CreateOutParams);
}
.
$CreatePod
.=
"\n=cut\n\n"
;
my
$CollectionClass
=
$LicenseBlock
.
$Attribution
.
$self
->_pod(
<<".") . $self->_magic_import($CollectionClassName);
^head1 NAME
$CollectionClassName -- Class Description
^head1 SYNOPSIS
use $CollectionClassName
^head1 DESCRIPTION
^head1 METHODS
^cut
package $CollectionClassName;
use $CollectionBaseclass;
use $RecordClassName;
use vars qw( \@ISA );
\@ISA= qw($CollectionBaseclass);
sub _Init {
my \$self = shift;
\$self->{'table'} = '$table';
\$self->{'primary_key'} = 'id';
.
if
(
$fields
{
'SortOrder'
} ) {
$CollectionClass
.=
$self
->_pod(
<<".");
# By default, order by name
\$self->OrderBy( ALIAS => 'main',
FIELD => 'SortOrder',
ORDER => 'ASC');
.
}
$CollectionClass
.=
$self
->_pod(
<<".");
return ( \$self->SUPER::_Init(\@_) );
}
^head2 NewItem
Returns an empty new $RecordClassName item
^cut
sub NewItem {
my \$self = shift;
return($RecordClassName->new(\$self->CurrentUser));
}
.
my
$RecordClassHeader
=
$Attribution
. "
^head1 NAME
$RecordClassName
^head1 SYNOPSIS
^head1 DESCRIPTION
^head1 METHODS
^cut
package
$RecordClassName
;
use
$RecordBaseclass
;
";
foreach
my
$key
(
keys
%requirements
) {
$RecordClassHeader
.=
$requirements
{
$key
} .
"\n"
;
}
$RecordClassHeader
.=
<<".";
use vars qw( \@ISA );
\@ISA= qw( $RecordBaseclass );
sub _Init {
my \$self = shift;
\$self->Table('$table');
\$self->SUPER::_Init(\@_);
}
.
my
$RecordClass
=
$LicenseBlock
.
$RecordClassHeader
.
$self
->_pod(
<<".") . $self->_magic_import($RecordClassName);
$RecordInit
^head2 Create PARAMHASH
Create takes a hash of values and creates a row in the database:
$CreatePod
$CreateSub
$FieldsPod
sub _CoreAccessible {
{
$ClassAccessible
}
};
.
print
"About to make $RecordClassPath, $CollectionClassPath\n"
;
`
mkdir
-p
$path
`;
open
( RECORD,
">$RecordClassPath"
);
print
RECORD
$RecordClass
;
close
(RECORD);
open
( COL,
">$CollectionClassPath"
);
print
COL
$CollectionClass
;
close
(COL);
}
}
sub
_magic_import {
my
$self
=
shift
;
my
$class
=
ref
(
$self
) ||
$self
;
my
$path
=
$class
;
$path
=~ s
my
$content
=
$self
->_pod(
<<".");
eval \"require ${class}_Overlay\";
if (\$@ && \$@ !~ qr{^Can't locate ${path}_Overlay.pm}) {
die \$@;
};
eval \"require ${class}_Vendor\";
if (\$@ && \$@ !~ qr{^Can't locate ${path}_Vendor.pm}) {
die \$@;
};
eval \"require ${class}_Local\";
if (\$@ && \$@ !~ qr{^Can't locate ${path}_Local.pm}) {
die \$@;
};
^head1 SEE ALSO
This class allows \"overlay\" methods to be placed
into the following files _Overlay is for a System overlay by the original author,
_Vendor is for 3rd-party vendor add-ons, while _Local is for site-local customizations.
These overlay files can contain new subs or subs to replace existing subs in this module.
If you'll be working with perl 5.6.0 or greater, each of these files should begin with the line
no warnings qw(redefine);
so that perl does not kick and scream when you redefine a subroutine or variable in your overlay.
${class}_Overlay, ${class}_Vendor, ${class}_Local
^cut
1;
.
return
$content
;
}
sub
_pod {
my
(
$self
,
$text
) =
@_
;
$text
=~ s/^\^/=/mg;
return
$text
;
}