#! /usr/bin/env perl
#
# Copyright (C) 2009-2025 Alexis Bienvenüe <paamc@passoire.fr>
#
# This file is part of Auto-Multiple-Choice
#
# Auto-Multiple-Choice 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.
#
# Auto-Multiple-Choice 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 Auto-Multiple-Choice.  If not, see
# <http://www.gnu.org/licenses/>.

use warnings;
use 5.012;

use Getopt::Long;
use AMC::Basic;
use AMC::NamesFile;
use AMC::Data;

my $notes_id        = '';
my $liste_file      = '';
my $liste_key       = '';
my $liste_enc       = 'utf-8';
my $csv_build_name  = '';
my $data_dir        = '';
my $preassoc        = '';
my $hard_fail       = '';
my $multiple_copies = '';

GetProjectOptions(
    ":assoc_code|notes-id=s"                 => \$notes_id,
    ":pre_association|pre-association!"      => \$preassoc,
    ":listeetudiants:dir|liste=s"            => \$liste_file,
    ":liste_key|liste-key=s"                 => \$liste_key,
    ":csv_build_name|csv-build-name=s"       => \$csv_build_name,
    ":data:dir|data=s"                       => \$data_dir,
    ":liste:enc|encodage-liste=s"            => \$liste_enc,
    ":multiple_copies:bool|multiple-copies!" => \$multiple_copies,
    "hard-fail!"                             => \$hard_fail,
);

$preassoc = 1 if ( $notes_id eq '<preassoc>' );

die "Needs notes-id"   if ( !$notes_id && !$preassoc );
die "Needs liste-key"  if ( !$liste_key );
die "Needs liste_file" if ( !-s $liste_file );
die "Needs data_dir"   if ( !-d $data_dir );

my $data    = AMC::Data->new($data_dir);
my $scoring = $data->module('scoring');
my $assoc   = $data->module('association');
my $capture = $data->module('capture');
my $layout;

$layout = $data->module('layout')
  if ($preassoc);

debug "Automatic association $liste_file [$liste_enc] / $liste_key";

# function that "cleans" IDs, removing leading zeros (so that 0001234
# will be the same as 1234)

sub clean_id {
    my ($i) = @_;
    $i =~ s/^0+//;
    return ($i);
}

# First read from the students list the possible values for the
# primary key to be found there (from column named $liste_key).

my $liste_e = AMC::NamesFile::new(
    $liste_file,
    encodage    => $liste_enc,
    identifiant => $csv_build_name
);

my %bon_code;
for my $ii ( 0 .. ( $liste_e->taille() - 1 ) ) {
    my $id = $liste_e->data_n( $ii, $liste_key );
    $bon_code{ clean_id($id) } = $id if ( defined($id) );
}

debug "Cleaned student list keys: " . join( ',', keys %bon_code );

# Open association database and clear old automatic association

$assoc->begin_transaction('ASSA');
annotate_source_change($capture);

$assoc->check_keys( $liste_key, $notes_id );
$assoc->clear_auto;

# Loop on all codes that can be read on the scans.

my $return_code = 0;

my $sth =
  $scoring->statement( $preassoc ? 'preAssocCounts'
    : $multiple_copies ? 'codesValues'
    :                    'codesCounts' );
if ($preassoc) {
    $sth->execute();
} else {
    $sth->execute($notes_id);
}
while ( my $v = $sth->fetchrow_hashref ) {
    if ( $v->{nb} == 1 ) {

        # nb is the number of scans on which the same code value has
        # been read. If nb=1, this is OK: we can process
        # association...  We also process association if
        # --multiple-copies is used, in such case multiple copies for
        # the same student are allowed.

        my $id_in_list = $bon_code{ clean_id( $v->{value} ) };
        if ( defined($id_in_list) ) {

            # Association OK
            debug "Association OK for code value $v->{value} ($id_in_list)";
            $assoc->set_auto( ( map { $v->{$_} } (qw/student copy/) ),
                $id_in_list );
        } else {

            # ... unless this value is NOT in the students list!
            debug_and_stderr( "Code value $v->{value}"
                  . " not found in students list: ignoring" );
            $return_code = 2;
        }
    } else {

        # Code value found on several sheets: do nothing, wait for the
        # user to make a manual association for these sheets.
        debug_and_stderr "Incorrect association for code value \""
          . $v->{value}
          . "\": $v->{nb} instances";
        $return_code = 2;
    }
}

$assoc->end_transaction('ASSA');

exit($return_code) if ($hard_fail);
