use 5.38.0;

package Part2;

use Part1;

use List::AllUtils qw/ pairmap sum max /;
use List::UtilsBy  qw/ sort_by /;

sub rank_hand (@hand) {
    @hand = split '', shift @hand if @hand == 1;

    my %group;
    $group{$_}++ for @hand;

    my $jokers = delete $group{'`'} // 0;

    return 7 if 1 >= keys %group;

    return 6 if grep { $_ + $jokers >= 4 } values %group;

    my $max = max values %group;

    if ( $max + $jokers >= 3 ) {
        return 5 if keys %group <= 2;
        return 4;
    }

    return 3 if keys %group == 3;

    return 2 if keys %group == 4;

    return 1;
}

sub handify ($str) {
    state %map = (
        'T' => 10,
        'J' => 1,
        'Q' => 12,
        'K' => 13,
        A   => 14
    );

    my @cards =
      map { chr( ord('a') + $_ - 2 ) } map { $map{$_} // $_ } split '', $str;

    return \@cards;
}

sub parse_input ($input) {
    my @lines      = split "\n", $input;
    my @hand_score = pairmap {
        [ handify($a), $b ]
    }
    map { split " " } @lines;
    return @hand_score;
}

sub score (@hand) {
    join '', rank_hand(@hand), @hand;
}

sub solution_2 ($input) {
    my @hand_score = sort_by { score( $_->[0]->@* ) } parse_input($input);
    my $rank       = 0;
    return sum map { ++$rank * $_->[1] } @hand_score;
}

1;