#!/usr/bin/perl -w
use strict;
use XML::Simple;
use Data::Dumper;

# Date
use POSIX qw(strftime);
my $date = strftime "%d/%m/%Y %H:%M", localtime;

# Functions
sub uniq {
  my %seen;
  grep !$seen{$_}++, @_;
}

print "\033[36m \nDEFACEMENT CHECK [" . $date . "]\n\nSearching for suspects ...\n";
print "FOUND \t\t=> URLs\n";

# Import config
my @search;
my $keywords = 'conf/keywords.cfg';
if (open(my $f, '<:encoding(UTF-8)', $keywords)) {
  while (my $row = <$f>) {
    chomp $row;
    push @search, $row;
  }
} else {
  warn "Could not open file '$keywords' $!";
}

# Open URLS files
my @files = glob( "urls/*.urls" );
my $file;
my $somme_total_found = 0;
my $grand_total_found = 0;

foreach $file (@files)
{
  my @urls;
  my $url;
  my $somme_total_found = 0;
  open(FH, '<', $file) or die $!;
  while(<FH>){
    my $url = $_;
    my $survey = $url;
    # Create a user agent object
    use LWP::UserAgent;
    my $ua = LWP::UserAgent->new;
    # Create a request
    my $req = HTTP::Request->new(GET => $survey);
    # Pass request to the user agent and get a response back
    my $res = $ua->request($req);
    my $site = $res->content;

    my @found;
    my $total_found = 0;

    foreach my $s (@search) {
      foreach (grep(/$s/i, split(/\n/, $site))) {
        push @found, $_;
      }
    }

    # Results
    foreach (uniq(@found)) {
      $total_found++;
      print "\033[33m[" . $total_found . "]\t\t" . $_ . "\n";
    }
    if ($total_found == 0) {
      print "\033[32m[" . $total_found . "]\t\t=> " . $survey. "\n";
    } else {
      print "\033[31mTREATH [" . $total_found . "]\t=> " . $survey. "\n";
      $somme_total_found = $somme_total_found + $total_found;
    }
  }
  close(FH);
  print "\033[36mTotal TREATH detected on this website \t\t=> [" . $somme_total_found . "]\n\n";
  $grand_total_found = $somme_total_found + $grand_total_found;
}

print "\033[36mGrand Total TREATH detected on all website \t\t=> [" . $grand_total_found . "]\n\n";