135 строки
3.2 KiB
Perl
135 строки
3.2 KiB
Perl
|
#!/usr/bin/env perl
|
||
|
|
|
||
|
|
use strict;
|
||
|
|
use warnings;
|
||
|
|
use Getopt::Long();
|
||
|
|
use Pod::Usage();
|
||
|
|
|
||
|
|
my $curl = 'curl';
|
||
|
|
my $shell = 'zsh';
|
||
|
|
my $help = 0;
|
||
|
|
Getopt::Long::GetOptions(
|
||
|
|
'curl=s' => \$curl,
|
||
|
|
'shell=s' => \$shell,
|
||
|
|
'help' => \$help,
|
||
|
|
) or Pod::Usage::pod2usage();
|
||
|
|
Pod::Usage::pod2usage() if $help;
|
||
|
|
|
||
|
|
my $regex = '\s+(?:(-[^\s]+),\s)?(--[^\s]+)\s*(\<.+?\>)?\s+(.*)';
|
||
|
|
my @opts = parse_main_opts('--help', $regex);
|
||
|
|
|
||
|
|
if ($shell eq 'fish') {
|
||
|
|
print "# curl fish completion\n\n";
|
||
|
|
print qq{$_ \n} foreach (@opts);
|
||
|
|
} elsif ($shell eq 'zsh') {
|
||
|
|
my $opts_str;
|
||
|
|
|
||
|
|
$opts_str .= qq{ $_ \\\n} foreach (@opts);
|
||
|
|
chomp $opts_str;
|
||
|
|
|
||
|
|
my $tmpl = <<"EOS";
|
||
|
|
#compdef curl
|
||
|
|
|
||
|
|
# curl zsh completion
|
||
|
|
|
||
|
|
local curcontext="\$curcontext" state state_descr line
|
||
|
|
typeset -A opt_args
|
||
|
|
|
||
|
|
local rc=1
|
||
|
|
|
||
|
|
_arguments -C -S \\
|
||
|
|
$opts_str
|
||
|
|
'*:URL:_urls' && rc=0
|
||
|
|
|
||
|
|
return rc
|
||
|
|
EOS
|
||
|
|
|
||
|
|
print $tmpl;
|
||
|
|
} else {
|
||
|
|
die("Unsupported shell: $shell");
|
||
|
|
}
|
||
|
|
|
||
|
|
sub parse_main_opts {
|
||
|
|
my ($cmd, $regex) = @_;
|
||
|
|
|
||
|
|
my @list;
|
||
|
|
my @lines = call_curl($cmd);
|
||
|
|
|
||
|
|
foreach my $line (@lines) {
|
||
|
|
my ($short, $long, $arg, $desc) = ($line =~ /^$regex/) or next;
|
||
|
|
|
||
|
|
my $option = '';
|
||
|
|
|
||
|
|
$arg =~ s/\:/\\\:/g if defined $arg;
|
||
|
|
|
||
|
|
$desc =~ s/'/'\\''/g if defined $desc;
|
||
|
|
$desc =~ s/\[/\\\[/g if defined $desc;
|
||
|
|
$desc =~ s/\]/\\\]/g if defined $desc;
|
||
|
|
$desc =~ s/\:/\\\:/g if defined $desc;
|
||
|
|
|
||
|
|
if ($shell eq 'fish') {
|
||
|
|
$option .= "complete --command curl";
|
||
|
|
$option .= " --short-option '" . strip_dash(trim($short)) . "'"
|
||
|
|
if defined $short;
|
||
|
|
$option .= " --long-option '" . strip_dash(trim($long)) . "'"
|
||
|
|
if defined $long;
|
||
|
|
$option .= " --description '" . strip_dash(trim($desc)) . "'"
|
||
|
|
if defined $desc;
|
||
|
|
} elsif ($shell eq 'zsh') {
|
||
|
|
$option .= '{' . trim($short) . ',' if defined $short;
|
||
|
|
$option .= trim($long) if defined $long;
|
||
|
|
$option .= '}' if defined $short;
|
||
|
|
$option .= '\'[' . trim($desc) . ']\'' if defined $desc;
|
||
|
|
|
||
|
|
$option .= ":'$arg'" if defined $arg;
|
||
|
|
|
||
|
|
$option .= ':_files'
|
||
|
|
if defined $arg and ($arg eq '<file>' || $arg eq '<filename>'
|
||
|
|
|| $arg eq '<dir>');
|
||
|
|
}
|
||
|
|
|
||
|
|
push @list, $option;
|
||
|
|
}
|
||
|
|
|
||
|
|
# Sort longest first, because zsh won't complete an option listed
|
||
|
|
# after one that's a prefix of it.
|
||
|
|
@list = sort {
|
||
|
|
$a =~ /([^=]*)/; my $ma = $1;
|
||
|
|
$b =~ /([^=]*)/; my $mb = $1;
|
||
|
|
|
||
|
|
length($mb) <=> length($ma)
|
||
|
|
} @list if $shell eq 'zsh';
|
||
|
|
|
||
|
|
return @list;
|
||
|
|
}
|
||
|
|
|
||
|
|
sub trim { my $s = shift; $s =~ s/^\s+|\s+$//g; return $s };
|
||
|
|
sub strip_dash { my $s = shift; $s =~ s/^-+//g; return $s };
|
||
|
|
|
||
|
|
sub call_curl {
|
||
|
|
my ($cmd) = @_;
|
||
|
|
my $output = `"$curl" $cmd`;
|
||
|
|
if ($? == -1) {
|
||
|
|
die "Could not run curl: $!";
|
||
|
|
} elsif ((my $exit_code = $? >> 8) != 0) {
|
||
|
|
die "curl returned $exit_code with output:\n$output";
|
||
|
|
}
|
||
|
|
return split /\n/, $output;
|
||
|
|
}
|
||
|
|
|
||
|
|
__END__
|
||
|
|
|
||
|
|
=head1 NAME
|
||
|
|
|
||
|
|
completion.pl - Generates tab-completion files for various shells
|
||
|
|
|
||
|
|
=head1 SYNOPSIS
|
||
|
|
|
||
|
|
completion.pl [options...]
|
||
|
|
|
||
|
|
--curl path to curl executable
|
||
|
|
--shell zsh/fish
|
||
|
|
--help prints this help
|
||
|
|
|
||
|
|
=cut
|