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 50c751f3 2022-06-29 op my $default_chars = $ENV{'PLASS_CHARS'} // '!-~';
 35 5cdf5adc 2022-05-08 op my $default_length = $ENV{'PLASS_LENGTH'};
 36 f9b1eaa1 2022-06-29 op if (!defined($default_length) || $default_length <= 0) {
 37 5cdf5adc 2022-05-08 op 	$default_length = 32;
 38 5cdf5adc 2022-05-08 op }
 39 5cdf5adc 2022-05-08 op 
 40 8dc394df 2022-06-29 op my %subcmd = (
 41 8dc394df 2022-06-29 op 	cat	=> [\&cmd_cat,		"entries..."],
 42 8dc394df 2022-06-29 op 	find	=> [\&cmd_find,		"[pattern]"],
 43 8dc394df 2022-06-29 op 	gen	=> [\&cmd_gen,		"[-nq] [-c chars] [-l length] entry"],
 44 8dc394df 2022-06-29 op 	mv	=> [\&cmd_mv,		"from to"],
 45 8dc394df 2022-06-29 op 	rm	=> [\&cmd_rm,		"entries..."],
 46 8dc394df 2022-06-29 op 	tee	=> [\&cmd_tee,		"[-q] entry"],
 47 8dc394df 2022-06-29 op     );
 48 8dc394df 2022-06-29 op 
 49 f19c01bc 2022-09-09 op my $usage = "[-h] command [argument ...]";
 50 5f8e7670 2022-06-29 op my $cmd;
 51 5f8e7670 2022-06-29 op sub usage {
 52 5f8e7670 2022-06-29 op 	my $prog = basename $0;
 53 d1bd3477 2022-06-29 op 	if (defined($cmd) and defined($subcmd{$cmd})) {
 54 5f8e7670 2022-06-29 op 		say STDERR "Usage: $prog $cmd $usage";
 55 5f8e7670 2022-06-29 op 	} else {
 56 5f8e7670 2022-06-29 op 		say STDERR "Usage: $prog $usage";
 57 d1bd3477 2022-06-29 op 		say STDERR "unknown command $cmd" if defined($cmd);
 58 8dc394df 2022-06-29 op 		say STDERR "commands: ", join(' ', sort(keys %subcmd));
 59 5f8e7670 2022-06-29 op 	}
 60 5f8e7670 2022-06-29 op 	exit 1;
 61 5f8e7670 2022-06-29 op }
 62 5cdf5adc 2022-05-08 op 
 63 5f8e7670 2022-06-29 op GetOptions("h|?" => \&usage) or usage();
 64 5cdf5adc 2022-05-08 op 
 65 5f8e7670 2022-06-29 op $cmd = shift // 'find';
 66 5f8e7670 2022-06-29 op usage() unless defined $subcmd{$cmd};
 67 5f8e7670 2022-06-29 op my $fn;
 68 5f8e7670 2022-06-29 op ($fn, $usage) = @{$subcmd{$cmd}};
 69 5cdf5adc 2022-05-08 op chdir $store;
 70 5cdf5adc 2022-05-08 op $fn->();
 71 5cdf5adc 2022-05-08 op exit 0;
 72 5cdf5adc 2022-05-08 op 
 73 5cdf5adc 2022-05-08 op 
 74 5cdf5adc 2022-05-08 op # utils
 75 5cdf5adc 2022-05-08 op 
 76 5cdf5adc 2022-05-08 op sub name2file {
 77 5cdf5adc 2022-05-08 op 	my $f = shift;
 78 5cdf5adc 2022-05-08 op 	$f .= ".gpg" unless $f =~ m,\.gpg$,;
 79 5cdf5adc 2022-05-08 op 	return $f;
 80 5cdf5adc 2022-05-08 op }
 81 5cdf5adc 2022-05-08 op 
 82 5cdf5adc 2022-05-08 op # tr -cd -- $chars < /dev/random | dd bs=$len count=1 status=none
 83 5cdf5adc 2022-05-08 op sub gen {
 84 5cdf5adc 2022-05-08 op 	my ($chars, $length) = @_;
 85 5cdf5adc 2022-05-08 op 	my $pass = "";
 86 5cdf5adc 2022-05-08 op 
 87 5cdf5adc 2022-05-08 op 	open(my $fh, '<:raw', '/dev/random')
 88 5cdf5adc 2022-05-08 op 	    or die "can't open /dev/random: $!";
 89 5cdf5adc 2022-05-08 op 	my $l = $length;
 90 f9b1eaa1 2022-06-29 op 	while ($l >= 0) {
 91 5cdf5adc 2022-05-08 op 		read($fh, my $t, $length * 4)
 92 5cdf5adc 2022-05-08 op 		    or die "failed to read /dev/random: $!";
 93 5cdf5adc 2022-05-08 op 		$t =~ s/[^$chars]//g;
 94 5cdf5adc 2022-05-08 op 		$l -= length($t);
 95 5cdf5adc 2022-05-08 op 		$pass .= $t;
 96 5cdf5adc 2022-05-08 op 	}
 97 5cdf5adc 2022-05-08 op 	close($fh);
 98 5cdf5adc 2022-05-08 op 
 99 5cdf5adc 2022-05-08 op 	return substr($pass, 0, $length);
100 5cdf5adc 2022-05-08 op }
101 5cdf5adc 2022-05-08 op 
102 1f89d66a 2022-06-29 op sub mkdirs {
103 1f89d66a 2022-06-29 op 	my $dir = shift;
104 1f89d66a 2022-06-29 op 	my $parent = dirname $dir;
105 1f89d66a 2022-06-29 op 	mkdirs($parent) unless -d $parent || $parent eq '/';
106 1f89d66a 2022-06-29 op 	mkdir $dir or die "mkdir $dir: $!"
107 1f89d66a 2022-06-29 op 	    unless -d $dir;
108 5cdf5adc 2022-05-08 op }
109 5cdf5adc 2022-05-08 op 
110 5cdf5adc 2022-05-08 op sub writepass {
111 5cdf5adc 2022-05-08 op 	my ($file, $pass) = @_;
112 1f89d66a 2022-06-29 op 
113 1f89d66a 2022-06-29 op 	mkdirs(dirname $file);
114 5cdf5adc 2022-05-08 op 
115 945f0333 2022-06-29 op 	# temporary redirect stdout to $file
116 945f0333 2022-06-29 op 	open(my $stdout, '>&', STDOUT);
117 945f0333 2022-06-29 op 	open(STDOUT, '>', $file);
118 945f0333 2022-06-29 op 
119 5cdf5adc 2022-05-08 op 	my @args = ($gpg, @gpg_flags, '-e', '-r', recipient(),
120 945f0333 2022-06-29 op 	    '-o', '-');
121 5cdf5adc 2022-05-08 op 	open my $fh, '|-', @args;
122 5eba1883 2022-10-20 op 	print $fh "$pass";
123 5cdf5adc 2022-05-08 op 	close($fh);
124 945f0333 2022-06-29 op 	my $ok = !$?;
125 945f0333 2022-06-29 op 
126 945f0333 2022-06-29 op 	open(STDOUT, '>&', $stdout); # restore stdout
127 945f0333 2022-06-29 op 
128 945f0333 2022-06-29 op 	die "failed to run $gpg\n" unless $ok;
129 5cdf5adc 2022-05-08 op }
130 5cdf5adc 2022-05-08 op 
131 5cdf5adc 2022-05-08 op sub recipient {
132 5cdf5adc 2022-05-08 op 	open my $fh, '<', "$store/.gpg-id"
133 5cdf5adc 2022-05-08 op 	    or die "can't open recipient file";
134 5cdf5adc 2022-05-08 op 	my $r = <$fh>;
135 5cdf5adc 2022-05-08 op 	chomp $r;
136 5cdf5adc 2022-05-08 op 	close($fh);
137 5cdf5adc 2022-05-08 op 	return $r;
138 5cdf5adc 2022-05-08 op }
139 5cdf5adc 2022-05-08 op 
140 5cdf5adc 2022-05-08 op sub passfind {
141 5cdf5adc 2022-05-08 op 	my $pattern = shift;
142 5cdf5adc 2022-05-08 op 	my @entries;
143 5cdf5adc 2022-05-08 op 
144 5cdf5adc 2022-05-08 op 	find({
145 5cdf5adc 2022-05-08 op 		wanted => sub {
146 5cdf5adc 2022-05-08 op 			if (m,/.git$, || m,/.got$,) {
147 5cdf5adc 2022-05-08 op 				$File::Find::prune = 1;
148 5cdf5adc 2022-05-08 op 				return;
149 5cdf5adc 2022-05-08 op 			}
150 5cdf5adc 2022-05-08 op 			return unless -f && m,.gpg$,;
151 5cdf5adc 2022-05-08 op 
152 5cdf5adc 2022-05-08 op 			s,^$store/*,,;
153 5cdf5adc 2022-05-08 op 			s,.gpg$,,;
154 fb70a412 2022-10-21 op 
155 fb70a412 2022-10-21 op 			return if defined($pattern) && ! m/$pattern/;
156 5cdf5adc 2022-05-08 op 			push @entries, $_;
157 5cdf5adc 2022-05-08 op 		},
158 5cdf5adc 2022-05-08 op 		no_chdir => 1,
159 5cdf5adc 2022-05-08 op 		follow_fast => 1,
160 5cdf5adc 2022-05-08 op 	     }, ($store));
161 5cdf5adc 2022-05-08 op 	return sort(@entries);
162 5cdf5adc 2022-05-08 op }
163 5cdf5adc 2022-05-08 op 
164 5cdf5adc 2022-05-08 op sub got {
165 87197963 2022-06-29 op 	# discard stdout
166 5cdf5adc 2022-05-08 op 	open my $fh, '-|', ($got, @_);
167 5cdf5adc 2022-05-08 op 	close($fh);
168 f96edc25 2022-06-29 op 	return !$?;
169 5cdf5adc 2022-05-08 op }
170 5cdf5adc 2022-05-08 op 
171 5cdf5adc 2022-05-08 op sub got_add {
172 f96edc25 2022-06-29 op 	return got 'add', '-I', shift;
173 5cdf5adc 2022-05-08 op }
174 5cdf5adc 2022-05-08 op 
175 5cdf5adc 2022-05-08 op sub got_rm {
176 f96edc25 2022-06-29 op 	got 'remove', '-f', shift
177 f96edc25 2022-06-29 op 	    or exit(1);
178 5cdf5adc 2022-05-08 op }
179 5cdf5adc 2022-05-08 op 
180 5cdf5adc 2022-05-08 op sub got_ci {
181 afdbc7b0 2022-06-29 op 	my $pid = fork;
182 afdbc7b0 2022-06-29 op 	die "failed to fork: $!" unless defined $pid;
183 afdbc7b0 2022-06-29 op 
184 f9b1eaa1 2022-06-29 op 	if ($pid != 0) {
185 afdbc7b0 2022-06-29 op 		wait;
186 afdbc7b0 2022-06-29 op 		die "failed to commit changes" if $?;
187 afdbc7b0 2022-06-29 op 		return;
188 afdbc7b0 2022-06-29 op 	}
189 afdbc7b0 2022-06-29 op 
190 afdbc7b0 2022-06-29 op 	open (STDOUT, ">&", \*STDERR)
191 afdbc7b0 2022-06-29 op 	    or die "can't redirect stdout to stderr";
192 afdbc7b0 2022-06-29 op 	exec ($got, 'commit', '-m', shift)
193 afdbc7b0 2022-06-29 op 	    or die "failed to exec $got: $!";
194 5cdf5adc 2022-05-08 op }
195 5cdf5adc 2022-05-08 op 
196 5cdf5adc 2022-05-08 op 
197 5cdf5adc 2022-05-08 op # cmds
198 5cdf5adc 2022-05-08 op 
199 5cdf5adc 2022-05-08 op sub cmd_cat {
200 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
201 5cdf5adc 2022-05-08 op 	usage unless @ARGV;
202 5cdf5adc 2022-05-08 op 
203 5cdf5adc 2022-05-08 op 	while (@ARGV) {
204 5cdf5adc 2022-05-08 op 		my $file = name2file(shift @ARGV);
205 5cdf5adc 2022-05-08 op 		system ($gpg, @gpg_flags, '-d', $file);
206 5cdf5adc 2022-05-08 op 		die "failed to exec $gpg: $!" if $? == -1;
207 5cdf5adc 2022-05-08 op 	}
208 5cdf5adc 2022-05-08 op }
209 5cdf5adc 2022-05-08 op 
210 5cdf5adc 2022-05-08 op sub cmd_find {
211 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
212 f9b1eaa1 2022-06-29 op 	usage if @ARGV > 1;
213 5cdf5adc 2022-05-08 op 
214 5cdf5adc 2022-05-08 op 	map { say $_ } passfind(shift @ARGV);
215 5cdf5adc 2022-05-08 op }
216 5cdf5adc 2022-05-08 op 
217 5cdf5adc 2022-05-08 op sub cmd_gen {
218 5cdf5adc 2022-05-08 op 	my $chars = $default_chars;
219 5cdf5adc 2022-05-08 op 	my $length = $default_length;
220 9d75d6d8 2022-06-29 op 	my $nop;
221 9d75d6d8 2022-06-29 op 	my $q;
222 5cdf5adc 2022-05-08 op 
223 5cdf5adc 2022-05-08 op 	GetOptions(
224 5cdf5adc 2022-05-08 op 		'c=s' => sub { $chars = $_[1] },
225 5cdf5adc 2022-05-08 op 		'h|?' => \&usage,
226 5cdf5adc 2022-05-08 op 		'l=i' => sub { $length = $_[1] },
227 9d75d6d8 2022-06-29 op 		'n'   => \$nop,
228 9d75d6d8 2022-06-29 op 		'q'   => \$q,
229 5cdf5adc 2022-05-08 op 	    ) or usage;
230 8bd06def 2022-10-01 op 
231 5eba1883 2022-10-20 op 	my $pass = gen($chars, $length) . "\n";
232 8bd06def 2022-10-01 op 	if ($nop) {
233 8bd06def 2022-10-01 op 		say $pass;
234 8bd06def 2022-10-01 op 		return;
235 8bd06def 2022-10-01 op 	}
236 8bd06def 2022-10-01 op 
237 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 1;
238 5cdf5adc 2022-05-08 op 
239 5cdf5adc 2022-05-08 op 	my $name = shift @ARGV;
240 5cdf5adc 2022-05-08 op 	my $file = name2file $name;
241 bd71caf6 2022-06-29 op 	my $renamed = -f $file;
242 5cdf5adc 2022-05-08 op 
243 8bd06def 2022-10-01 op 	writepass($file, $pass);
244 8bd06def 2022-10-01 op 	got_add $file;
245 8bd06def 2022-10-01 op 	got_ci($renamed ? "update $name" : "+$name");
246 9d75d6d8 2022-06-29 op 	say $pass unless $q;
247 5cdf5adc 2022-05-08 op }
248 5cdf5adc 2022-05-08 op 
249 5cdf5adc 2022-05-08 op # TODO: handle moving directories?
250 5cdf5adc 2022-05-08 op sub cmd_mv {
251 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
252 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 2;
253 5cdf5adc 2022-05-08 op 
254 5cdf5adc 2022-05-08 op 	my $a = shift @ARGV;
255 5cdf5adc 2022-05-08 op 	my $b = shift @ARGV;
256 5cdf5adc 2022-05-08 op 
257 5cdf5adc 2022-05-08 op 	my $pa = name2file $a;
258 5cdf5adc 2022-05-08 op 	my $pb = name2file $b;
259 5cdf5adc 2022-05-08 op 
260 5cdf5adc 2022-05-08 op 	die "source password doesn't exist" unless -f $pa;
261 5cdf5adc 2022-05-08 op 	die "target password exists" if -f $pb;
262 5cdf5adc 2022-05-08 op 
263 4568ada0 2022-06-29 op 	mkdirs(dirname $pb);
264 5cdf5adc 2022-05-08 op 	rename $pa, $pb or die "can't rename $a to $b: $!";
265 5cdf5adc 2022-05-08 op 
266 5cdf5adc 2022-05-08 op 	got_rm $pa;
267 bd71caf6 2022-06-29 op 	got_add $pb or die "can't add $pb\n";
268 5cdf5adc 2022-05-08 op 	got_ci "mv $a $b";
269 5cdf5adc 2022-05-08 op }
270 5cdf5adc 2022-05-08 op 
271 5cdf5adc 2022-05-08 op sub cmd_rm {
272 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
273 6a455fdf 2022-06-29 op 	usage unless @ARGV;
274 5cdf5adc 2022-05-08 op 
275 6a455fdf 2022-06-29 op 	while (@ARGV) {
276 6a455fdf 2022-06-29 op 		my $name = shift @ARGV;
277 6a455fdf 2022-06-29 op 		my $file = name2file $name;
278 5cdf5adc 2022-05-08 op 
279 6a455fdf 2022-06-29 op 		got_rm $file;
280 6a455fdf 2022-06-29 op 		got_ci "-$name";
281 6a455fdf 2022-06-29 op 	}
282 5cdf5adc 2022-05-08 op }
283 5cdf5adc 2022-05-08 op 
284 afdbc7b0 2022-06-29 op sub cmd_tee {
285 afdbc7b0 2022-06-29 op 	my $q;
286 afdbc7b0 2022-06-29 op 	GetOptions(
287 afdbc7b0 2022-06-29 op 		'h|?' => \&usage,
288 afdbc7b0 2022-06-29 op 		'q'   => \$q,
289 afdbc7b0 2022-06-29 op 	    ) or usage;
290 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 1;
291 5cdf5adc 2022-05-08 op 
292 5cdf5adc 2022-05-08 op 	my $name = shift @ARGV;
293 5cdf5adc 2022-05-08 op 	my $file = name2file $name;
294 5cdf5adc 2022-05-08 op 
295 5eba1883 2022-10-20 op 	my $msg = -f $file ? "update $name" : "+$name";
296 5eba1883 2022-10-20 op 
297 5eba1883 2022-10-20 op 	my $pass = "";
298 5eba1883 2022-10-20 op 	$pass .= $_ while <STDIN>;
299 dd204b00 2022-10-20 op 	die "No content!\n" if $pass eq "";
300 5eba1883 2022-10-20 op 
301 d69b7902 2022-05-15 op 	writepass($file, $pass);
302 5cdf5adc 2022-05-08 op 
303 5cdf5adc 2022-05-08 op 	got_add $file;
304 5eba1883 2022-10-20 op 	got_ci $msg;
305 5eba1883 2022-10-20 op 	print $pass unless $q;
306 5cdf5adc 2022-05-08 op }