blob: c2eb774f786244c06f007e8ba5945ec4ea1ac98c [file] [log] [blame]
xf.li6c8fc1e2023-08-12 00:11:09 -07001#!/usr/bin/env perl
2#***************************************************************************
3# _ _ ____ _
4# Project ___| | | | _ \| |
5# / __| | | | |_) | |
6# | (__| |_| | _ <| |___
7# \___|\___/|_| \_\_____|
8#
9# Copyright (C) 1998 - 2022, Daniel Stenberg, <daniel@haxx.se>, et al.
10#
11# This software is licensed as described in the file COPYING, which
12# you should have received as part of this distribution. The terms
13# are also available at https://curl.se/docs/copyright.html.
14#
15# You may opt to use, copy, modify, merge, publish, distribute and/or sell
16# copies of the Software, and permit persons to whom the Software is
17# furnished to do so, under the terms of the COPYING file.
18#
19# This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
20# KIND, either express or implied.
21#
22# SPDX-License-Identifier: curl
23#
24###########################################################################
25
26use strict;
27use warnings;
28use Getopt::Long();
29use Pod::Usage();
30
31my $curl = 'curl';
32my $shell = 'zsh';
33my $help = 0;
34Getopt::Long::GetOptions(
35 'curl=s' => \$curl,
36 'shell=s' => \$shell,
37 'help' => \$help,
38) or Pod::Usage::pod2usage();
39Pod::Usage::pod2usage() if $help;
40
41my $regex = '\s+(?:(-[^\s]+),\s)?(--[^\s]+)\s*(\<.+?\>)?\s+(.*)';
42my @opts = parse_main_opts('--help all', $regex);
43
44if ($shell eq 'fish') {
45 print "# curl fish completion\n\n";
46 print qq{$_ \n} foreach (@opts);
47} elsif ($shell eq 'zsh') {
48 my $opts_str;
49
50 $opts_str .= qq{ $_ \\\n} foreach (@opts);
51 chomp $opts_str;
52
53my $tmpl = <<"EOS";
54#compdef curl
55
56# curl zsh completion
57
58local curcontext="\$curcontext" state state_descr line
59typeset -A opt_args
60
61local rc=1
62
63_arguments -C -S \\
64$opts_str
65 '*:URL:_urls' && rc=0
66
67return rc
68EOS
69
70 print $tmpl;
71} else {
72 die("Unsupported shell: $shell");
73}
74
75sub parse_main_opts {
76 my ($cmd, $regex) = @_;
77
78 my @list;
79 my @lines = call_curl($cmd);
80
81 foreach my $line (@lines) {
82 my ($short, $long, $arg, $desc) = ($line =~ /^$regex/) or next;
83
84 my $option = '';
85
86 $arg =~ s/\:/\\\:/g if defined $arg;
87
88 $desc =~ s/'/'\\''/g if defined $desc;
89 $desc =~ s/\[/\\\[/g if defined $desc;
90 $desc =~ s/\]/\\\]/g if defined $desc;
91 $desc =~ s/\:/\\\:/g if defined $desc;
92
93 if ($shell eq 'fish') {
94 $option .= "complete --command curl";
95 $option .= " --short-option '" . strip_dash(trim($short)) . "'"
96 if defined $short;
97 $option .= " --long-option '" . strip_dash(trim($long)) . "'"
98 if defined $long;
99 $option .= " --description '" . strip_dash(trim($desc)) . "'"
100 if defined $desc;
101 } elsif ($shell eq 'zsh') {
102 $option .= '{' . trim($short) . ',' if defined $short;
103 $option .= trim($long) if defined $long;
104 $option .= '}' if defined $short;
105 $option .= '\'[' . trim($desc) . ']\'' if defined $desc;
106
107 if (defined $arg) {
108 $option .= ":'$arg'";
109 if ($arg =~ /<file ?(name)?>|<path>/) {
110 $option .= ':_files';
111 } elsif ($arg =~ /<dir>/) {
112 $option .= ":'_path_files -/'";
113 } elsif ($arg =~ /<url>/i) {
114 $option .= ':_urls';
115 } elsif ($long =~ /ftp/ && $arg =~ /<method>/) {
116 $option .= ":'(multicwd nocwd singlecwd)'";
117 } elsif ($arg =~ /<method>/) {
118 $option .= ":'(DELETE GET HEAD POST PUT)'";
119 }
120 }
121 }
122
123 push @list, $option;
124 }
125
126 # Sort longest first, because zsh won't complete an option listed
127 # after one that's a prefix of it.
128 @list = sort {
129 $a =~ /([^=]*)/; my $ma = $1;
130 $b =~ /([^=]*)/; my $mb = $1;
131
132 length($mb) <=> length($ma)
133 } @list if $shell eq 'zsh';
134
135 return @list;
136}
137
138sub trim { my $s = shift; $s =~ s/^\s+|\s+$//g; return $s };
139sub strip_dash { my $s = shift; $s =~ s/^-+//g; return $s };
140
141sub call_curl {
142 my ($cmd) = @_;
143 my $output = `"$curl" $cmd`;
144 if ($? == -1) {
145 die "Could not run curl: $!";
146 } elsif ((my $exit_code = $? >> 8) != 0) {
147 die "curl returned $exit_code with output:\n$output";
148 }
149 return split /\n/, $output;
150}
151
152__END__
153
154=head1 NAME
155
156completion.pl - Generates tab-completion files for various shells
157
158=head1 SYNOPSIS
159
160completion.pl [options...]
161
162 --curl path to curl executable
163 --shell zsh/fish
164 --help prints this help
165
166=cut