revbank/plugins/nomoney

#!perl

use List::Util qw(none uniqstr);

my @deny_plugins = (
	"give",
	#"market",
	#"products",
	"take",
	#"unlisted",
	"withdraw",
);

my $allow_multi_user = 1;

sub _derive_plugin($symbol) {
	return $1 if $symbol =~ /^RevBank::Plugin::(\w+)::/;
	return;
}

sub _inform($unresolved, $username, $skip_print = 0) {
	call_hooks("beep");

	say "Not possible:";

	for my $entry ($unresolved->entries) {
		unless ($skip_print) {
			my $line = ($entry->as_printable)[0];
			say $line;

			$line =~ s/^\s+//;
			call_hooks("log_info", "nomoney: $line");
		}

		my $broke_users = $entry->attribute('nomoney_users');

		for my $account (sort keys %$broke_users) {
			my $balance = RevBank::Accounts::balance($account);
			
			my $m = sprintf(
				"%s have %s",
				($account eq $username ? "You don't" : "$account doesn't"),
				abs($broke_users->{$account}),
			);

			call_hooks("log_info", "nomoney: $m (balance: $balance)");
			my $b = ($balance < 0 ? "\e[31;1m$balance\e[m" : $balance);
			say "\e[31;1m$m\e[m (balance: $b)";
		}
	}
}

my %unresolved;  # to share state between hooks, keyed per real cart

sub hook_checkout_prepare($class, $cart, $username, $transaction_id, @) {
	my $unresolved = $unresolved{$cart} = RevBank::Cart->new;

	my $deltas = $cart->deltas($username);
	my %balances;

	for my $account (keys %$deltas) {
		next if $deltas->{$account} > 0;
		next if RevBank::Accounts::is_special($account);

		my $old = $balances{$account} = RevBank::Accounts::balance($account);
		my $new = $old + $deltas->{$account};

		next if $new >= 0 or $new > $old;

		for my $entry ($cart->entries) {
			my $plugin = $entry->attribute('plugin') // _derive_plugin($entry->{caller});
			next if not $plugin;
			next if none { $plugin eq $_ } @deny_plugins;

			my @contra_users = uniqstr sort grep {
				not RevBank::Accounts::is_special($_)
				and $_ ne $username
			} map {
				$_->{account}
			} $entry->contras;

			next if $allow_multi_user and @contra_users > 1;
			next if none { $account eq $_ } $entry->account // $username, @contra_users;

			$unresolved->add_entry($entry);
		}
	}

	return if not $unresolved->size;  # allow transaction as is

	my $newline = 0;
	if ($cart->changed) {
		# Show original cart before changing it, if it hasn't been shown before
		say "Pending:";
		$cart->display;
		$newline = 1;
	}
	$cart->delete($_) for @{ $unresolved->{entries} };

	# Find entries that can be done, by brute force, and add them back.
	RESOLVE: {
		my $resolved_deltas = $cart->deltas($username);
		my %resolved_balances = %balances;
		$resolved_balances{$_} += $resolved_deltas->{$_} for keys %$resolved_deltas;

		for my $entry ($unresolved->entries) {
			my $single = RevBank::Cart->new;
			$single->add_entry($entry);
			my $trial_deltas = $single->deltas($username);

			my %broke_users;
			$entry->attribute('nomoney_users', \%broke_users);

			for my $account (keys %$trial_deltas) {
				next if RevBank::Accounts::is_special($account);
				next if $trial_deltas->{$account} > 0;

				my $trial_balance = $resolved_balances{$account} + $trial_deltas->{$account};
				
				if ($trial_balance < 0) {
					$broke_users{$account} += $trial_deltas->{$account};
				}
			}

			if (not %broke_users) {
				$cart->add_entry($entry);
				$unresolved->delete($entry);
				redo RESOLVE;
			}
		}
	}

	if (not $cart->size) {
		print "\n" if $newline;
		_inform($unresolved, $username, 1);
		return ABORT;
	}

	return;
}

sub hook_abort($class, $cart, @) {
	delete $unresolved{$cart};
	return;
}

sub hook_checkout_done($class, $cart, $account, $transaction_id, @) {
	my $n = $unresolved{$cart}->size or return;
	print "\n";

	_inform($unresolved{$cart}, $account);
	delete $unresolved{$cart};

	my $message = $n == 1 ? "THIS ENTRY WAS IGNORED" : "THESE ENTRIES WERE IGNORED";
	say "\e[1;4m$message.\e[0m" if $n;
	return;
}