1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
package MySQL::Utils;
use strict;
use base qw(Exporter);
our @EXPORT_OK = qw(parse_arg auth_args_string read_file debug_level debug);
sub parse_arg {
my ($opts, $arg, $num) = @_;
my %opts = %$opts;
debug(1, "parsing arg $num: `$arg'\n");
my $authnum = $num + 1;
my %auth = ();
for my $auth (qw/host port user password socket/) {
$auth{$auth} = $opts{"$auth$authnum"} || $opts{$auth};
delete $auth{$auth} unless $auth{$auth};
}
if ($arg =~ /^db:(.*)/) {
return new MySQL::Database(db => $1, auth => \%auth);
}
if ($opts{"host$authnum"} ||
$opts{"port$authnum"} ||
$opts{"user$authnum"} ||
$opts{"password$authnum"} ||
$opts{"socket$authnum"})
{
return new MySQL::Database(db => $arg, auth => \%auth);
}
if (-f $arg) {
return new MySQL::Database(file => $arg, auth => \%auth);
}
my %dbs = available_dbs(%auth);
debug(2, " available databases: ", (join ', ', keys %dbs), "\n");
if ($dbs{$arg}) {
return new MySQL::Database(db => $arg, auth => \%auth);
}
return "`$arg' is not a valid file or database.\n";
}
sub available_dbs {
my %auth = @_;
my $args = auth_args_string(%auth);
# evil but we don't use DBI because I don't want to implement -p properly
# not that this works with -p anyway ...
open(MYSQLSHOW, "mysqlshow$args |")
or die "Couldn't execute `mysqlshow$args': $!\n";
my @dbs = ();
while (<MYSQLSHOW>) {
next unless /^\| (\w+)/;
push @dbs, $1;
}
close(MYSQLSHOW) or die "mysqlshow$args failed: $!";
return map { $_ => 1 } @dbs;
}
sub auth_args_string {
my %auth = @_;
my $args = '';
for my $arg (qw/host port user password socket/) {
$args .= " --$arg=$auth{$arg}" if $auth{$arg};
}
return $args;
}
sub read_file {
my ($file) = @_;
open(FILE, $file) or die "Couldn't open `$file': $!\n";
my @contents = <FILE>;
close(FILE);
return @contents;
}
{
my $debug_level = 0;
sub debug_level {
my ($new_debug_level) = @_;
$debug_level = $new_debug_level if defined $new_debug_level;
return $debug_level;
}
sub debug {
my $level = shift;
print STDERR @_ if ($debug_level >= $level) && @_;
}
}
1;
|
