Include section for supported AUR helpers.

Signed-off-by: Nicolás Ortega Froysa <nicolas@ortegas.org>

README.md

@@ -45,6 +45,10 @@ There are two modes for undoing pacman transactions:
 
 Look at the man-page (`man pacundo`) for more information.
 
+### Supported AUR Helpers
+
+- [yay](https://github.com/Jguer/yay)
+
 ## License
 
 This project is licensed under the terms & conditions of the Zlib license.

pacundo.1

@@ -21,19 +21,22 @@ is especially helpful if the last update broke your system for some reason.
 .SH "OPTIONS"
 .TP
 \fB\-i\fR
-Interactively select the transactions to undo (default behavior)
+Enter interactive mode to select package operations to undo (default behavior)
 .TP
 \fB\-r\fR
-Non-interactively undo transactions
+Non-interactively undo entire transactions
 .TP
 \fB\-t\fR <\fInum\fR>
-Select number of transactions to include (default: 1)
+Specify the number of transactions to include (default: 1)
+.TP
+\fB\-d\fR
+Dry run, i.e. don't actually do anything
 .TP
 \fB\-h\fR
-Show help information
+Show this help information
 .TP
 \fB\-v\fR
-Show program version
+Print program version
 
 .SH "AUTHOR"
 Written by Nicolás A. Ortega Froysa.

pacundo.pl

@@ -47,9 +47,9 @@ USAGE:
 	$PROG_NAME -v
 
 OPTIONS:
-	-i         Enter interactive mode to select packages to downgrade [default behavior]
-	-r         Automatically downgrade all packages from last upgrade
-	-t <num>   Specify the number of transactions to include for undoing selection [default 1]
+	-i         Enter interactive mode to select package operations to undo (default behavior)
+	-r         Non-interactively undo entire transactions
+	-t <num>   Specify the number of transactions to include (default: 1)
 	-d         Dry run, i.e. don't actually do anything
 	-h         Show this help information
 	-v         Print program version\n");
@@ -169,7 +169,8 @@ sub get_pkgmgr() {
 	my %pkgmgr = (
 		name           => $mgr,
 		bin            => $mgr_bin,
-		search         => "$sudo $mgr_bin -Ss",
+		search         => "$mgr_bin -Ss",
+		info           => "$mgr_bin -Si",
 		install_remote => "$sudo $mgr_bin -S",
 		install_local  => "$sudo $mgr_bin -U",
 		remove         => "$sudo $mgr_bin -R",
@@ -178,17 +179,33 @@ sub get_pkgmgr() {
 	return \%pkgmgr;
 }
 
-sub find_local_pkg($pkgmgr, $pkg_name, $pkg_ver) {
+sub find_local_pkg($pkgmgr, $pkg_name, $pkg_ver='') {
 	my $pkg_file = '';
-	my $aur_dir = "$ENV{'XDG_CACHE_HOME'}/yay/$pkg_name";
+	my $pkg_pat;
+	my $repo = `$pkgmgr->{info} $pkg_name | awk '{ if (\$1 == "Repository") print \$3; }'`;;
 
-	if ($pkgmgr->{name} eq 'yay' && -d $aur_dir) {
-		$pkg_file = `ls $aur_dir/$pkg_name-$pkg_ver-*.pkg.tar.zst | tail -n1`;
+	if ($pkg_ver ne '') {
+		$pkg_pat = "$pkg_name-$pkg_ver-*.pkg.tar.zst";
 	} else {
-		$pkg_file = `ls /var/cache/pacman/pkg/$pkg_name-$pkg_ver-*.pkg.tar.zst | tail -n1`;
+		$pkg_pat = "$pkg_name-*.pkg.tar.zst";
+	}
+
+	if ($repo eq 'aur') {
+		my $aur_dir;
+
+		if ($pkgmgr->{name} eq 'yay') {
+			$aur_dir = "$ENV{'XDG_CACHE_HOME'}/yay/$pkg_name";
+		} else {
+			return '';
+		}
+
+		$pkg_file = `ls $aur_dir/$pkg_pat | tail -n1`;
+	} else {
+		$pkg_file = `ls /var/cache/pacman/pkg/$pkg_pat | tail -n1`;
 	}
 
 	chomp($pkg_file);
+
 	return $pkg_file;
 }
 
@@ -227,8 +244,12 @@ foreach my $tx (@undo_txs) {
 	if ($tx->{action} eq 'installed') {
 		$remove_pkgs .= "$tx->{pkg_name} ";
 	} elsif ($tx->{action} eq 'removed') {
-		# TODO: install local package if available
+		my $pkg_file = &find_local_pkg($pkgmgr, $tx->{pkg_name});
+		if ($pkg_file eq '') {
 			$remote_install_pkgs .= "$tx->{pkg_name} ";
+		} else {
+			$local_install_pkgs .= "$pkg_file ";
+		}
 	} else {
 		my $pkg_file = &find_local_pkg($pkgmgr, $tx->{pkg_name}, $tx->{oldver});
 		if ($pkg_file eq '') {
@@ -239,6 +260,12 @@ foreach my $tx (@undo_txs) {
 	}
 }
 
-system("$pkgmgr->{remove} $remove_pkgs") if ($remove_pkgs ne '');
-system("$pkgmgr->{install_remote} $remote_install_pkgs") if ($remote_install_pkgs ne '');
-system("$pkgmgr->{install_local} $local_install_pkgs") if ($local_install_pkgs ne '');
+if ($dry_run) {
+	print("$pkgmgr->{remove} $remove_pkgs\n") if ($remove_pkgs ne '');
+	print("$pkgmgr->{install_remote} $remote_install_pkgs\n") if ($remote_install_pkgs ne '');
+	print("$pkgmgr->{install_local} $local_install_pkgs\n") if ($local_install_pkgs ne '');
+} else {
+	system("$pkgmgr->{remove} $remove_pkgs") if ($remove_pkgs ne '');
+	system("$pkgmgr->{install_remote} $remote_install_pkgs") if ($remote_install_pkgs ne '');
+	system("$pkgmgr->{install_local} $local_install_pkgs") if ($local_install_pkgs ne '');
+}