1 5cdf5adc 2022-05-08 op #!/usr/bin/env perl
  2 5cdf5adc 2022-05-08 op #
  3 5cdf5adc 2022-05-08 op # Copyright (c) 2022 Omar Polo <op@omarpolo.com>
  4 5cdf5adc 2022-05-08 op #
  5 5cdf5adc 2022-05-08 op # Permission to use, copy, modify, and distribute this software for any
  6 5cdf5adc 2022-05-08 op # purpose with or without fee is hereby granted, provided that the above
  7 5cdf5adc 2022-05-08 op # copyright notice and this permission notice appear in all copies.
  8 5cdf5adc 2022-05-08 op #
  9 5cdf5adc 2022-05-08 op # THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 10 5cdf5adc 2022-05-08 op # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 11 5cdf5adc 2022-05-08 op # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 12 5cdf5adc 2022-05-08 op # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 13 5cdf5adc 2022-05-08 op # WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 14 5cdf5adc 2022-05-08 op # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 15 5cdf5adc 2022-05-08 op # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 16 5cdf5adc 2022-05-08 op 
 17 5cdf5adc 2022-05-08 op use strict;
 18 5cdf5adc 2022-05-08 op use warnings;
 19 5cdf5adc 2022-05-08 op use v5.32;
 20 5cdf5adc 2022-05-08 op 
 21 5cdf5adc 2022-05-08 op use open ":std", ":encoding(UTF-8)";
 22 5cdf5adc 2022-05-08 op 
 23 5cdf5adc 2022-05-08 op use Getopt::Long qw(:config bundling require_order);
 24 5cdf5adc 2022-05-08 op use File::Basename;
 25 5cdf5adc 2022-05-08 op use File::Find;
 26 5cdf5adc 2022-05-08 op 
 27 50c751f3 2022-06-29 op my $store = $ENV{'PLASS_STORE'} // $ENV{'HOME'}.'/.password-store';
 28 5cdf5adc 2022-05-08 op 
 29 50c751f3 2022-06-29 op my $got = $ENV{'PLASS_GOT'} // 'got';
 30 5cdf5adc 2022-05-08 op 
 31 d16aedaf 2022-09-09 op my $gpg = $ENV{'PLASS_GPG'} // 'gpg';
 32 12ad62fd 2022-06-29 op my @gpg_flags = qw(--quiet --compress-algo=none --no-encrypt-to --use-agent);
 33 5cdf5adc 2022-05-08 op 
 34 8dc394df 2022-06-29 op my %subcmd = (
 35 8dc394df 2022-06-29 op 	cat	=> [\&cmd_cat,		"entries..."],
 36 8dc394df 2022-06-29 op 	find	=> [\&cmd_find,		"[pattern]"],
 37 8dc394df 2022-06-29 op 	mv	=> [\&cmd_mv,		"from to"],
 38 8dc394df 2022-06-29 op 	rm	=> [\&cmd_rm,		"entries..."],
 39 8dc394df 2022-06-29 op 	tee	=> [\&cmd_tee,		"[-q] entry"],
 40 8dc394df 2022-06-29 op     );
 41 8dc394df 2022-06-29 op 
 42 f19c01bc 2022-09-09 op my $usage = "[-h] command [argument ...]";
 43 5f8e7670 2022-06-29 op my $cmd;
 44 5f8e7670 2022-06-29 op sub usage {
 45 5f8e7670 2022-06-29 op 	my $prog = basename $0;
 46 d1bd3477 2022-06-29 op 	if (defined($cmd) and defined($subcmd{$cmd})) {
 47 5f8e7670 2022-06-29 op 		say STDERR "Usage: $prog $cmd $usage";
 48 5f8e7670 2022-06-29 op 	} else {
 49 5f8e7670 2022-06-29 op 		say STDERR "Usage: $prog $usage";
 50 d1bd3477 2022-06-29 op 		say STDERR "unknown command $cmd" if defined($cmd);
 51 8dc394df 2022-06-29 op 		say STDERR "commands: ", join(' ', sort(keys %subcmd));
 52 5f8e7670 2022-06-29 op 	}
 53 5f8e7670 2022-06-29 op 	exit 1;
 54 5f8e7670 2022-06-29 op }
 55 5cdf5adc 2022-05-08 op 
 56 5f8e7670 2022-06-29 op GetOptions("h|?" => \&usage) or usage();
 57 5cdf5adc 2022-05-08 op 
 58 5f8e7670 2022-06-29 op $cmd = shift // 'find';
 59 5f8e7670 2022-06-29 op usage() unless defined $subcmd{$cmd};
 60 5f8e7670 2022-06-29 op my $fn;
 61 5f8e7670 2022-06-29 op ($fn, $usage) = @{$subcmd{$cmd}};
 62 5cdf5adc 2022-05-08 op chdir $store;
 63 5cdf5adc 2022-05-08 op $fn->();
 64 5cdf5adc 2022-05-08 op exit 0;
 65 5cdf5adc 2022-05-08 op 
 66 5cdf5adc 2022-05-08 op 
 67 5cdf5adc 2022-05-08 op # utils
 68 5cdf5adc 2022-05-08 op 
 69 5cdf5adc 2022-05-08 op sub name2file {
 70 5cdf5adc 2022-05-08 op 	my $f = shift;
 71 5cdf5adc 2022-05-08 op 	$f .= ".gpg" unless $f =~ m,\.gpg$,;
 72 5cdf5adc 2022-05-08 op 	return $f;
 73 5cdf5adc 2022-05-08 op }
 74 5cdf5adc 2022-05-08 op 
 75 1f89d66a 2022-06-29 op sub mkdirs {
 76 1f89d66a 2022-06-29 op 	my $dir = shift;
 77 1f89d66a 2022-06-29 op 	my $parent = dirname $dir;
 78 1f89d66a 2022-06-29 op 	mkdirs($parent) unless -d $parent || $parent eq '/';
 79 1f89d66a 2022-06-29 op 	mkdir $dir or die "mkdir $dir: $!"
 80 1f89d66a 2022-06-29 op 	    unless -d $dir;
 81 5cdf5adc 2022-05-08 op }
 82 5cdf5adc 2022-05-08 op 
 83 5cdf5adc 2022-05-08 op sub writepass {
 84 5cdf5adc 2022-05-08 op 	my ($file, $pass) = @_;
 85 1f89d66a 2022-06-29 op 
 86 1f89d66a 2022-06-29 op 	mkdirs(dirname $file);
 87 5cdf5adc 2022-05-08 op 
 88 945f0333 2022-06-29 op 	# temporary redirect stdout to $file
 89 945f0333 2022-06-29 op 	open(my $stdout, '>&', STDOUT);
 90 945f0333 2022-06-29 op 	open(STDOUT, '>', $file);
 91 945f0333 2022-06-29 op 
 92 5cdf5adc 2022-05-08 op 	my @args = ($gpg, @gpg_flags, '-e', '-r', recipient(),
 93 945f0333 2022-06-29 op 	    '-o', '-');
 94 5cdf5adc 2022-05-08 op 	open my $fh, '|-', @args;
 95 5eba1883 2022-10-20 op 	print $fh "$pass";
 96 5cdf5adc 2022-05-08 op 	close($fh);
 97 945f0333 2022-06-29 op 	my $ok = !$?;
 98 945f0333 2022-06-29 op 
 99 945f0333 2022-06-29 op 	open(STDOUT, '>&', $stdout); # restore stdout
100 945f0333 2022-06-29 op 
101 945f0333 2022-06-29 op 	die "failed to run $gpg\n" unless $ok;
102 5cdf5adc 2022-05-08 op }
103 5cdf5adc 2022-05-08 op 
104 5cdf5adc 2022-05-08 op sub recipient {
105 5cdf5adc 2022-05-08 op 	open my $fh, '<', "$store/.gpg-id"
106 5cdf5adc 2022-05-08 op 	    or die "can't open recipient file";
107 5cdf5adc 2022-05-08 op 	my $r = <$fh>;
108 5cdf5adc 2022-05-08 op 	chomp $r;
109 5cdf5adc 2022-05-08 op 	close($fh);
110 5cdf5adc 2022-05-08 op 	return $r;
111 5cdf5adc 2022-05-08 op }
112 5cdf5adc 2022-05-08 op 
113 5cdf5adc 2022-05-08 op sub passfind {
114 5cdf5adc 2022-05-08 op 	my $pattern = shift;
115 5cdf5adc 2022-05-08 op 	my @entries;
116 5cdf5adc 2022-05-08 op 
117 5cdf5adc 2022-05-08 op 	find({
118 5cdf5adc 2022-05-08 op 		wanted => sub {
119 5cdf5adc 2022-05-08 op 			if (m,/.git$, || m,/.got$,) {
120 5cdf5adc 2022-05-08 op 				$File::Find::prune = 1;
121 5cdf5adc 2022-05-08 op 				return;
122 5cdf5adc 2022-05-08 op 			}
123 5cdf5adc 2022-05-08 op 			return unless -f && m,.gpg$,;
124 5cdf5adc 2022-05-08 op 
125 5cdf5adc 2022-05-08 op 			s,^$store/*,,;
126 5cdf5adc 2022-05-08 op 			s,.gpg$,,;
127 fb70a412 2022-10-21 op 
128 fb70a412 2022-10-21 op 			return if defined($pattern) && ! m/$pattern/;
129 5cdf5adc 2022-05-08 op 			push @entries, $_;
130 5cdf5adc 2022-05-08 op 		},
131 5cdf5adc 2022-05-08 op 		no_chdir => 1,
132 5cdf5adc 2022-05-08 op 		follow_fast => 1,
133 5cdf5adc 2022-05-08 op 	     }, ($store));
134 5cdf5adc 2022-05-08 op 	return sort(@entries);
135 5cdf5adc 2022-05-08 op }
136 5cdf5adc 2022-05-08 op 
137 5cdf5adc 2022-05-08 op sub got {
138 87197963 2022-06-29 op 	# discard stdout
139 5cdf5adc 2022-05-08 op 	open my $fh, '-|', ($got, @_);
140 5cdf5adc 2022-05-08 op 	close($fh);
141 f96edc25 2022-06-29 op 	return !$?;
142 5cdf5adc 2022-05-08 op }
143 5cdf5adc 2022-05-08 op 
144 5cdf5adc 2022-05-08 op sub got_add {
145 f96edc25 2022-06-29 op 	return got 'add', '-I', shift;
146 5cdf5adc 2022-05-08 op }
147 5cdf5adc 2022-05-08 op 
148 5cdf5adc 2022-05-08 op sub got_rm {
149 f96edc25 2022-06-29 op 	got 'remove', '-f', shift
150 f96edc25 2022-06-29 op 	    or exit(1);
151 5cdf5adc 2022-05-08 op }
152 5cdf5adc 2022-05-08 op 
153 5cdf5adc 2022-05-08 op sub got_ci {
154 afdbc7b0 2022-06-29 op 	my $pid = fork;
155 afdbc7b0 2022-06-29 op 	die "failed to fork: $!" unless defined $pid;
156 afdbc7b0 2022-06-29 op 
157 f9b1eaa1 2022-06-29 op 	if ($pid != 0) {
158 afdbc7b0 2022-06-29 op 		wait;
159 afdbc7b0 2022-06-29 op 		die "failed to commit changes" if $?;
160 afdbc7b0 2022-06-29 op 		return;
161 afdbc7b0 2022-06-29 op 	}
162 afdbc7b0 2022-06-29 op 
163 afdbc7b0 2022-06-29 op 	open (STDOUT, ">&", \*STDERR)
164 afdbc7b0 2022-06-29 op 	    or die "can't redirect stdout to stderr";
165 afdbc7b0 2022-06-29 op 	exec ($got, 'commit', '-m', shift)
166 afdbc7b0 2022-06-29 op 	    or die "failed to exec $got: $!";
167 5cdf5adc 2022-05-08 op }
168 5cdf5adc 2022-05-08 op 
169 5cdf5adc 2022-05-08 op 
170 5cdf5adc 2022-05-08 op # cmds
171 5cdf5adc 2022-05-08 op 
172 5cdf5adc 2022-05-08 op sub cmd_cat {
173 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
174 5cdf5adc 2022-05-08 op 	usage unless @ARGV;
175 5cdf5adc 2022-05-08 op 
176 5cdf5adc 2022-05-08 op 	while (@ARGV) {
177 5cdf5adc 2022-05-08 op 		my $file = name2file(shift @ARGV);
178 5cdf5adc 2022-05-08 op 		system ($gpg, @gpg_flags, '-d', $file);
179 5cdf5adc 2022-05-08 op 		die "failed to exec $gpg: $!" if $? == -1;
180 5cdf5adc 2022-05-08 op 	}
181 5cdf5adc 2022-05-08 op }
182 5cdf5adc 2022-05-08 op 
183 5cdf5adc 2022-05-08 op sub cmd_find {
184 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
185 f9b1eaa1 2022-06-29 op 	usage if @ARGV > 1;
186 5cdf5adc 2022-05-08 op 
187 5cdf5adc 2022-05-08 op 	map { say $_ } passfind(shift @ARGV);
188 5cdf5adc 2022-05-08 op }
189 5cdf5adc 2022-05-08 op 
190 5cdf5adc 2022-05-08 op # TODO: handle moving directories?
191 5cdf5adc 2022-05-08 op sub cmd_mv {
192 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
193 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 2;
194 5cdf5adc 2022-05-08 op 
195 5cdf5adc 2022-05-08 op 	my $a = shift @ARGV;
196 5cdf5adc 2022-05-08 op 	my $b = shift @ARGV;
197 5cdf5adc 2022-05-08 op 
198 5cdf5adc 2022-05-08 op 	my $pa = name2file $a;
199 5cdf5adc 2022-05-08 op 	my $pb = name2file $b;
200 5cdf5adc 2022-05-08 op 
201 5cdf5adc 2022-05-08 op 	die "source password doesn't exist" unless -f $pa;
202 5cdf5adc 2022-05-08 op 	die "target password exists" if -f $pb;
203 5cdf5adc 2022-05-08 op 
204 4568ada0 2022-06-29 op 	mkdirs(dirname $pb);
205 5cdf5adc 2022-05-08 op 	rename $pa, $pb or die "can't rename $a to $b: $!";
206 5cdf5adc 2022-05-08 op 
207 5cdf5adc 2022-05-08 op 	got_rm $pa;
208 bd71caf6 2022-06-29 op 	got_add $pb or die "can't add $pb\n";
209 5cdf5adc 2022-05-08 op 	got_ci "mv $a $b";
210 5cdf5adc 2022-05-08 op }
211 5cdf5adc 2022-05-08 op 
212 5cdf5adc 2022-05-08 op sub cmd_rm {
213 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
214 6a455fdf 2022-06-29 op 	usage unless @ARGV;
215 5cdf5adc 2022-05-08 op 
216 6a455fdf 2022-06-29 op 	while (@ARGV) {
217 6a455fdf 2022-06-29 op 		my $name = shift @ARGV;
218 6a455fdf 2022-06-29 op 		my $file = name2file $name;
219 5cdf5adc 2022-05-08 op 
220 6a455fdf 2022-06-29 op 		got_rm $file;
221 6a455fdf 2022-06-29 op 		got_ci "-$name";
222 6a455fdf 2022-06-29 op 	}
223 5cdf5adc 2022-05-08 op }
224 5cdf5adc 2022-05-08 op 
225 afdbc7b0 2022-06-29 op sub cmd_tee {
226 afdbc7b0 2022-06-29 op 	my $q;
227 afdbc7b0 2022-06-29 op 	GetOptions(
228 afdbc7b0 2022-06-29 op 		'h|?' => \&usage,
229 afdbc7b0 2022-06-29 op 		'q'   => \$q,
230 afdbc7b0 2022-06-29 op 	    ) or usage;
231 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 1;
232 5cdf5adc 2022-05-08 op 
233 5cdf5adc 2022-05-08 op 	my $name = shift @ARGV;
234 5cdf5adc 2022-05-08 op 	my $file = name2file $name;
235 5cdf5adc 2022-05-08 op 
236 5eba1883 2022-10-20 op 	my $msg = -f $file ? "update $name" : "+$name";
237 5eba1883 2022-10-20 op 
238 5eba1883 2022-10-20 op 	my $pass = "";
239 5eba1883 2022-10-20 op 	$pass .= $_ while <STDIN>;
240 dd204b00 2022-10-20 op 	die "No content!\n" if $pass eq "";
241 5eba1883 2022-10-20 op 
242 d69b7902 2022-05-15 op 	writepass($file, $pass);
243 5cdf5adc 2022-05-08 op 
244 5cdf5adc 2022-05-08 op 	got_add $file;
245 5eba1883 2022-10-20 op 	got_ci $msg;
246 5eba1883 2022-10-20 op 	print $pass unless $q;
247 5cdf5adc 2022-05-08 op }