1#!/usr/bin/perl
2#
3#  Licensed to the Apache Software Foundation (ASF) under one or more
4#  contributor license agreements.  See the NOTICE file distributed with
5#  this work for additional information regarding copyright ownership.
6#  The ASF licenses this file to You under the Apache License, Version 2.0
7#  (the "License"); you may not use this file except in compliance with
8#  the License.  You may obtain a copy of the License at
9#
10#      http://www.apache.org/licenses/LICENSE-2.0
11#
12#  Unless required by applicable law or agreed to in writing, software
13#  distributed under the License is distributed on an "AS IS" BASIS,
14#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15#  See the License for the specific language governing permissions and
16#  limitations under the License.
17#
18# A script to allow Bash or Z-Shell to complete an Ant command-line.
19#
20# To install for Bash 2.0 or better, add the following to ~/.bashrc:
21#
22#     $ complete -C complete-ant-cmd ant build.sh
23#
24# To install for Z-Shell 2.5 or better, add the following to ~/.zshrc:
25#
26#     function ant_complete () {
27#         local args_line args
28#         read -l args_line
29#         set -A args $args_line
30#         set -A reply $(COMP_LINE=$args_line complete-ant-cmd ${args[1]} $1)
31#     }
32#     compctl -K ant_complete ant build.sh
33#
34# @author Mike Williams <mikew@cortexebusiness.com.au>
35
36my $cmdLine = $ENV{'COMP_LINE'};
37my $antCmd = $ARGV[0];
38my $word = $ARGV[1];
39
40my @completions;
41if ($word =~ /^-/) {
42    list( restrict( $word, getArguments() ));
43} elsif ($cmdLine =~ /-(f|buildfile)\s+\S*$/) {
44    list( getBuildFiles($word) );
45} else {
46    list( restrict( $word, getTargets() ));
47}
48
49exit(0);
50
51sub list {
52    for (@_) {
53        print "$_\n";
54    }
55}
56
57sub restrict {
58    my ($word, @completions) = @_;
59    grep( /^\Q$word\E/, @completions );
60}
61
62sub getArguments {
63    qw(-buildfile -debug -emacs -f -find -help -listener -logfile
64       -logger -projecthelp -quiet -verbose -version);
65}
66
67
68sub getBuildFiles {
69    my ($word) = @_;
70    grep( /\.xml$/, glob( "$word*" ));
71}
72
73sub getTargets {
74
75    # Look for build-file
76    my $buildFile = 'build.xml';
77    if ($cmdLine =~ /-(f|buildfile)\s+(\S+)/) {
78        $buildFile = $2;
79    }
80    return () unless (-f $buildFile);
81
82    # Run "ant -projecthelp" to list targets.  Keep a cache of results in a
83    # cache-file.
84    my $cacheFile = $buildFile;
85    $cacheFile =~ s|(.*/)?(.*)|${1}.ant-targets-${2}|;
86    if ((!-e $cacheFile) || (-M $buildFile) < (-M $cacheFile)) {
87        open( CACHE, '>'.$cacheFile ) || die "can\'t write $cacheFile: $!\n";
88        open( HELP, "$antCmd -projecthelp -f '$buildFile'|" ) || return();
89        my %targets;
90        while( <HELP> ) {
91            if (/^\s+(\S+)/) {
92                $targets{$1}++;
93            }
94        }
95        my @targets = sort keys %targets;
96        for (@targets) { print CACHE "$_\n"; }
97        return @targets;
98    }
99
100    # Read the target-cache
101    open( CACHE, $cacheFile ) || die "can\'t read $cacheFile: $!\n";
102    my @targets;
103    while (<CACHE>) {
104        chop;
105        s/\r$//;  # for Cygwin
106        push( @targets, $_ );
107    }
108    close( CACHE );
109    @targets;
110
111}
112
113
114
115