summaryrefslogtreecommitdiff
path: root/lib/Ninka/CommentExtractor.pm
blob: 8a2e66f98eb2acf3ba09829c546cf362eefeaa66 (plain)
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
102
103
104
105
106
107
108
109
110
111
package Ninka::CommentExtractor;

use strict;
use warnings;
use IPC::Open3 'open3';
use Symbol 'gensym';

sub new {
    my ($class, %args) = @_;

    my $self = bless({}, $class);

    die "parameter 'input_file' is mandatory" unless exists $args{input_file};

    $self->{verbose} = ($args{verbose} // 0) == 1;
    $self->{input_file} = $args{input_file};

    return $self;
}

sub execute {
    my ($self) = @_;

    my $command = $self->determine_comments_command();
    my $comments = execute_command($command);
    if ($command =~ /^comments/ && length($comments) == 0) {
        $command = create_head_cmd($self->{input_file}, 700);
        $comments = execute_command($command);
    }

    return $comments;
}

sub determine_comments_command {
    my ($self) = @_;

    my $input_file = $self->{input_file};

    if ($input_file =~ /\.([^\.]+)$/) {
        my $ext = $1;
        if ($ext =~ /^(pl|pm|py)$/) {
            return create_head_cmd($input_file, 400);
        } elsif ($ext =~ /^(jl|el)$/) {
            return create_head_cmd($input_file, 400);
        } elsif ($ext =~ /^(java|c|cpp|h|cxx|c\+\+|cc)$/) {
            my $comments_binary = 'comments';
            if (`which $comments_binary` ne '') {
                return "$comments_binary -c1 '$input_file' 2> /dev/null";
            } else {
                return create_head_cmd($input_file, 400);
            }
        } else {
            return create_head_cmd($input_file, 700);
        }
    } else {
        return create_head_cmd($input_file, 700);
    }
}

sub create_head_cmd {
    my ($input_file, $count_lines) = @_;

    return "head -$count_lines $input_file";
}

sub execute_command {
    my ($command) = @_;

    my ($child_in, $child_out, $child_err);
    $child_err = gensym();
    my $pid = open3($child_in, $child_out, $child_err, $command);
    my $comments = do { local $/; <$child_out> };
    chomp(my $error = join('; ', <$child_err>));
    waitpid $pid, 0;
    my $status = ($? >> 8);
    die "execution of program [$command] failed: status [$status], error [$error]" if ($status != 0);

    return $comments;
}

1;

__END__

=head1 NAME

Ninka::CommentExtractor

=head1 DESCRIPTION

Extracts comments from source code.
If no comment extractor is known for a language, then extracts top lines from source.

=head1 COPYRIGHT AND LICENSE

Copyright (C) 2009-2014  Yuki Manabe and Daniel M. German

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 2 of the
License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

=cut