#!/usr/bin/perl
use utf8;
use strict;
# On précise l'encodage des données de sortie :
binmode(STDOUT, ":encoding(UTF-8)");
if ($#ARGV != 1) {
    print "Il manque un argument à votre programme....\n";
    exit;
}

my $rep="$ARGV[0]";
my $rubrique="$ARGV[1]";
# Pour assurer que le nom du répertoire ne se termine pas par un "/" :
$rep=~ s/[\/]$//;

open my $txt_output, ">:encoding(UTF-8)", "titre-description-$rubrique.txt";
open my $xml_output, ">:encoding(UTF-8)", "titre-description-$rubrique.xml";
print $xml_output "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n<corpus>\n";
&parcoursarborescencefichiers($rep);
print $xml_output "</corpus>";
close $txt_output;
close $xml_output;

# Étiquetage du fichier .txt avec udpipe :
&UPTagging;

# Étiquetage du fichier .xml avec treetagger :
&TTTagging;

# Suppression des fichiers qui ont été utilisé dans le traitement mais qui ne servent plus: 
system("rm -f description.txt description_bis.txt title.txt title_bis.txt titre-description.xml titre-description.txt");

exit;

# Sous-programmes :
sub parcoursarborescencefichiers {
    my $path = shift(@_);
    opendir(DIR, $path) or die "can't open $path: $!\n";
    my @files = readdir(DIR);
    closedir(DIR);
    foreach my $file (@files) {
	   next if $file =~ /^\.\.?$/;
	   $file = $path."/".$file;


	   if (-d $file) {
	       &parcoursarborescencefichiers($file);
	   }

	   if (-f $file) {
            if ($file =~ /$rubrique.+\.xml$/) {
                print "Traitement du fichier $file \n";
                open my $input, "<:encoding(UTF-8)", $file;
                $/=undef;
                my $line=<$input>;
                close($input);

                while ($line=~/<item><title>(.+?)<\/title>.+?<description>(.+?)<\/description>/gs) {
                    my $title=&clean($1);
                    my $description=&clean($2);
                    print $txt_output "$title \n";
                    print $txt_output "$description \n";
                    my ($segmentized_title, $segmentized_description)=&segmentationTD($title,$description);

                    print $xml_output "<item><titre>\n$segmentized_title\n</titre><description>\n$segmentized_description\n</description></item>\n";
                }
            }
	   }
    }
}

sub clean {
    # Sous-programme de nettoyage des titres et descriptions
    my $text=shift @_;
    $text=~s/(^<!\[CDATA\[)|(\]\]>$)//g;
    $text.=".";
    $text=~s/\.+$/\./;
    return $text;
}

sub UPTagging {
    # Sous-programme d'étiquetage du fichier texte avec udpipe
    system("../udpipe/bin-osx/udpipe --tokenize --tag --parse --tokenizer=presegmented ../udpipe/modeles/french-sequoia-ud-2.5-191206.udpipe titre-description.txt > titre-description.udpipe");

}

sub TTTagging {
    # Sous-programme d'étiquetage du fichier XML avec TreeTagger
    system("../tree-tagger-MacOSX-3.2.3/tree-tagger -lemma -token -no-unknown -sgml ../tree-tagger-MacOSX-3.2.3/french-utf8.par titre-description.xml > titre-description");
    system("perl ../tree-tagger-MacOSX-3.2.3/treetagger2xml-utf8.pl titre-description UTF8")
}

sub segmentationTD {
    # Sous-programme pour la segmentation des phrases
    my ($arg1, $arg2)=@_;

    # Traitement du titre :
    open my $tmp, ">:encoding(UTF-8)","title.txt";
    print $tmp $arg1;
    close $tmp;
    system("perl ../tree-tagger-MacOSX-3.2.3/tokenise-utf8.pl title.txt > title_bis.txt");
    undef $/;
    open my $tmp2, "<:encoding(UTF-8)","title_bis.txt";
    my $segmentized_title=<$tmp2>;
    close $tmp2;

    # Traitement de description :
    open my $tmp, ">:encoding(UTF-8)","description.txt";
    print $tmp $arg2;
    close $tmp;
    system("perl ../tree-tagger-MacOSX-3.2.3/tokenise-utf8.pl description.txt > description_bis.txt");
    open my $tmp2, "<:encoding(UTF-8)","description_bis.txt";
    my $segmentized_description=<$tmp2>;
    close $tmp2;
    $/="\n";

    return $segmentized_title, $segmentized_description;
}
