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 5cdf5adc 2022-05-08 op sub readpass {
103 5cdf5adc 2022-05-08 op 	# todo some stty black magic to avoid echo
104 5cdf5adc 2022-05-08 op 	print shift if -t;
105 5cdf5adc 2022-05-08 op 	my $pass = <>;
106 5cdf5adc 2022-05-08 op 	die "failed to read stdin: $!" unless defined($pass);
107 5cdf5adc 2022-05-08 op 	chomp $pass;
108 5cdf5adc 2022-05-08 op 	return $pass;
109 1f89d66a 2022-06-29 op }
110 1f89d66a 2022-06-29 op 
111 1f89d66a 2022-06-29 op sub mkdirs {
112 1f89d66a 2022-06-29 op 	my $dir = shift;
113 1f89d66a 2022-06-29 op 	my $parent = dirname $dir;
114 1f89d66a 2022-06-29 op 	mkdirs($parent) unless -d $parent || $parent eq '/';
115 1f89d66a 2022-06-29 op 	mkdir $dir or die "mkdir $dir: $!"
116 1f89d66a 2022-06-29 op 	    unless -d $dir;
117 5cdf5adc 2022-05-08 op }
118 5cdf5adc 2022-05-08 op 
119 5cdf5adc 2022-05-08 op sub writepass {
120 5cdf5adc 2022-05-08 op 	my ($file, $pass) = @_;
121 1f89d66a 2022-06-29 op 
122 1f89d66a 2022-06-29 op 	mkdirs(dirname $file);
123 5cdf5adc 2022-05-08 op 
124 945f0333 2022-06-29 op 	# temporary redirect stdout to $file
125 945f0333 2022-06-29 op 	open(my $stdout, '>&', STDOUT);
126 945f0333 2022-06-29 op 	open(STDOUT, '>', $file);
127 945f0333 2022-06-29 op 
128 5cdf5adc 2022-05-08 op 	my @args = ($gpg, @gpg_flags, '-e', '-r', recipient(),
129 945f0333 2022-06-29 op 	    '-o', '-');
130 5cdf5adc 2022-05-08 op 	open my $fh, '|-', @args;
131 5cdf5adc 2022-05-08 op 	say $fh "$pass";
132 5cdf5adc 2022-05-08 op 	close($fh);
133 945f0333 2022-06-29 op 	my $ok = !$?;
134 945f0333 2022-06-29 op 
135 945f0333 2022-06-29 op 	open(STDOUT, '>&', $stdout); # restore stdout
136 945f0333 2022-06-29 op 
137 945f0333 2022-06-29 op 	die "failed to run $gpg\n" unless $ok;
138 5cdf5adc 2022-05-08 op }
139 5cdf5adc 2022-05-08 op 
140 5cdf5adc 2022-05-08 op sub recipient {
141 5cdf5adc 2022-05-08 op 	open my $fh, '<', "$store/.gpg-id"
142 5cdf5adc 2022-05-08 op 	    or die "can't open recipient file";
143 5cdf5adc 2022-05-08 op 	my $r = <$fh>;
144 5cdf5adc 2022-05-08 op 	chomp $r;
145 5cdf5adc 2022-05-08 op 	close($fh);
146 5cdf5adc 2022-05-08 op 	return $r;
147 5cdf5adc 2022-05-08 op }
148 5cdf5adc 2022-05-08 op 
149 5cdf5adc 2022-05-08 op sub passfind {
150 5cdf5adc 2022-05-08 op 	my $pattern = shift;
151 5cdf5adc 2022-05-08 op 	my @entries;
152 5cdf5adc 2022-05-08 op 
153 5cdf5adc 2022-05-08 op 	find({
154 5cdf5adc 2022-05-08 op 		wanted => sub {
155 5cdf5adc 2022-05-08 op 			if (m,/.git$, || m,/.got$,) {
156 5cdf5adc 2022-05-08 op 				$File::Find::prune = 1;
157 5cdf5adc 2022-05-08 op 				return;
158 5cdf5adc 2022-05-08 op 			}
159 5cdf5adc 2022-05-08 op 
160 5cdf5adc 2022-05-08 op 			return if defined($pattern) && ! m/$pattern/;
161 5cdf5adc 2022-05-08 op 			return unless -f && m,.gpg$,;
162 5cdf5adc 2022-05-08 op 
163 5cdf5adc 2022-05-08 op 			s,^$store/*,,;
164 5cdf5adc 2022-05-08 op 			s,.gpg$,,;
165 5cdf5adc 2022-05-08 op 			push @entries, $_;
166 5cdf5adc 2022-05-08 op 		},
167 5cdf5adc 2022-05-08 op 		no_chdir => 1,
168 5cdf5adc 2022-05-08 op 		follow_fast => 1,
169 5cdf5adc 2022-05-08 op 	     }, ($store));
170 5cdf5adc 2022-05-08 op 	return sort(@entries);
171 5cdf5adc 2022-05-08 op }
172 5cdf5adc 2022-05-08 op 
173 5cdf5adc 2022-05-08 op sub got {
174 87197963 2022-06-29 op 	# discard stdout
175 5cdf5adc 2022-05-08 op 	open my $fh, '-|', ($got, @_);
176 5cdf5adc 2022-05-08 op 	close($fh);
177 f96edc25 2022-06-29 op 	return !$?;
178 5cdf5adc 2022-05-08 op }
179 5cdf5adc 2022-05-08 op 
180 5cdf5adc 2022-05-08 op sub got_add {
181 f96edc25 2022-06-29 op 	return got 'add', '-I', shift;
182 5cdf5adc 2022-05-08 op }
183 5cdf5adc 2022-05-08 op 
184 5cdf5adc 2022-05-08 op sub got_rm {
185 f96edc25 2022-06-29 op 	got 'remove', '-f', shift
186 f96edc25 2022-06-29 op 	    or exit(1);
187 5cdf5adc 2022-05-08 op }
188 5cdf5adc 2022-05-08 op 
189 5cdf5adc 2022-05-08 op sub got_ci {
190 afdbc7b0 2022-06-29 op 	my $pid = fork;
191 afdbc7b0 2022-06-29 op 	die "failed to fork: $!" unless defined $pid;
192 afdbc7b0 2022-06-29 op 
193 f9b1eaa1 2022-06-29 op 	if ($pid != 0) {
194 afdbc7b0 2022-06-29 op 		wait;
195 afdbc7b0 2022-06-29 op 		die "failed to commit changes" if $?;
196 afdbc7b0 2022-06-29 op 		return;
197 afdbc7b0 2022-06-29 op 	}
198 afdbc7b0 2022-06-29 op 
199 afdbc7b0 2022-06-29 op 	open (STDOUT, ">&", \*STDERR)
200 afdbc7b0 2022-06-29 op 	    or die "can't redirect stdout to stderr";
201 afdbc7b0 2022-06-29 op 	exec ($got, 'commit', '-m', shift)
202 afdbc7b0 2022-06-29 op 	    or die "failed to exec $got: $!";
203 5cdf5adc 2022-05-08 op }
204 5cdf5adc 2022-05-08 op 
205 5cdf5adc 2022-05-08 op 
206 5cdf5adc 2022-05-08 op # cmds
207 5cdf5adc 2022-05-08 op 
208 5cdf5adc 2022-05-08 op sub cmd_cat {
209 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
210 5cdf5adc 2022-05-08 op 	usage unless @ARGV;
211 5cdf5adc 2022-05-08 op 
212 5cdf5adc 2022-05-08 op 	while (@ARGV) {
213 5cdf5adc 2022-05-08 op 		my $file = name2file(shift @ARGV);
214 5cdf5adc 2022-05-08 op 		system ($gpg, @gpg_flags, '-d', $file);
215 5cdf5adc 2022-05-08 op 		die "failed to exec $gpg: $!" if $? == -1;
216 5cdf5adc 2022-05-08 op 	}
217 5cdf5adc 2022-05-08 op }
218 5cdf5adc 2022-05-08 op 
219 5cdf5adc 2022-05-08 op sub cmd_find {
220 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
221 f9b1eaa1 2022-06-29 op 	usage if @ARGV > 1;
222 5cdf5adc 2022-05-08 op 
223 5cdf5adc 2022-05-08 op 	map { say $_ } passfind(shift @ARGV);
224 5cdf5adc 2022-05-08 op }
225 5cdf5adc 2022-05-08 op 
226 5cdf5adc 2022-05-08 op sub cmd_gen {
227 5cdf5adc 2022-05-08 op 	my $chars = $default_chars;
228 5cdf5adc 2022-05-08 op 	my $length = $default_length;
229 9d75d6d8 2022-06-29 op 	my $nop;
230 9d75d6d8 2022-06-29 op 	my $q;
231 5cdf5adc 2022-05-08 op 
232 5cdf5adc 2022-05-08 op 	GetOptions(
233 5cdf5adc 2022-05-08 op 		'c=s' => sub { $chars = $_[1] },
234 5cdf5adc 2022-05-08 op 		'h|?' => \&usage,
235 5cdf5adc 2022-05-08 op 		'l=i' => sub { $length = $_[1] },
236 9d75d6d8 2022-06-29 op 		'n'   => \$nop,
237 9d75d6d8 2022-06-29 op 		'q'   => \$q,
238 5cdf5adc 2022-05-08 op 	    ) or usage;
239 8bd06def 2022-10-01 op 
240 8bd06def 2022-10-01 op 	my $pass = gen($chars, $length);
241 8bd06def 2022-10-01 op 	if ($nop) {
242 8bd06def 2022-10-01 op 		say $pass;
243 8bd06def 2022-10-01 op 		return;
244 8bd06def 2022-10-01 op 	}
245 8bd06def 2022-10-01 op 
246 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 1;
247 5cdf5adc 2022-05-08 op 
248 5cdf5adc 2022-05-08 op 	my $name = shift @ARGV;
249 5cdf5adc 2022-05-08 op 	my $file = name2file $name;
250 bd71caf6 2022-06-29 op 	my $renamed = -f $file;
251 5cdf5adc 2022-05-08 op 
252 8bd06def 2022-10-01 op 	writepass($file, $pass);
253 8bd06def 2022-10-01 op 	got_add $file;
254 8bd06def 2022-10-01 op 	got_ci($renamed ? "update $name" : "+$name");
255 9d75d6d8 2022-06-29 op 	say $pass unless $q;
256 5cdf5adc 2022-05-08 op }
257 5cdf5adc 2022-05-08 op 
258 5cdf5adc 2022-05-08 op # TODO: handle moving directories?
259 5cdf5adc 2022-05-08 op sub cmd_mv {
260 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
261 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 2;
262 5cdf5adc 2022-05-08 op 
263 5cdf5adc 2022-05-08 op 	my $a = shift @ARGV;
264 5cdf5adc 2022-05-08 op 	my $b = shift @ARGV;
265 5cdf5adc 2022-05-08 op 
266 5cdf5adc 2022-05-08 op 	my $pa = name2file $a;
267 5cdf5adc 2022-05-08 op 	my $pb = name2file $b;
268 5cdf5adc 2022-05-08 op 
269 5cdf5adc 2022-05-08 op 	die "source password doesn't exist" unless -f $pa;
270 5cdf5adc 2022-05-08 op 	die "target password exists" if -f $pb;
271 5cdf5adc 2022-05-08 op 
272 4568ada0 2022-06-29 op 	mkdirs(dirname $pb);
273 5cdf5adc 2022-05-08 op 	rename $pa, $pb or die "can't rename $a to $b: $!";
274 5cdf5adc 2022-05-08 op 
275 5cdf5adc 2022-05-08 op 	got_rm $pa;
276 bd71caf6 2022-06-29 op 	got_add $pb or die "can't add $pb\n";
277 5cdf5adc 2022-05-08 op 	got_ci "mv $a $b";
278 5cdf5adc 2022-05-08 op }
279 5cdf5adc 2022-05-08 op 
280 5cdf5adc 2022-05-08 op sub cmd_rm {
281 5cdf5adc 2022-05-08 op 	GetOptions('h|?' => \&usage) or usage;
282 6a455fdf 2022-06-29 op 	usage unless @ARGV;
283 5cdf5adc 2022-05-08 op 
284 6a455fdf 2022-06-29 op 	while (@ARGV) {
285 6a455fdf 2022-06-29 op 		my $name = shift @ARGV;
286 6a455fdf 2022-06-29 op 		my $file = name2file $name;
287 5cdf5adc 2022-05-08 op 
288 6a455fdf 2022-06-29 op 		got_rm $file;
289 6a455fdf 2022-06-29 op 		got_ci "-$name";
290 6a455fdf 2022-06-29 op 	}
291 5cdf5adc 2022-05-08 op }
292 5cdf5adc 2022-05-08 op 
293 afdbc7b0 2022-06-29 op sub cmd_tee {
294 afdbc7b0 2022-06-29 op 	my $q;
295 afdbc7b0 2022-06-29 op 	GetOptions(
296 afdbc7b0 2022-06-29 op 		'h|?' => \&usage,
297 afdbc7b0 2022-06-29 op 		'q'   => \$q,
298 afdbc7b0 2022-06-29 op 	    ) or usage;
299 f9b1eaa1 2022-06-29 op 	usage if @ARGV != 1;
300 5cdf5adc 2022-05-08 op 
301 5cdf5adc 2022-05-08 op 	my $name = shift @ARGV;
302 5cdf5adc 2022-05-08 op 	my $file = name2file $name;
303 5cdf5adc 2022-05-08 op 
304 d69b7902 2022-05-15 op 	my $pass = readpass "Enter the password: ";
305 d69b7902 2022-05-15 op 	writepass($file, $pass);
306 5cdf5adc 2022-05-08 op 
307 5cdf5adc 2022-05-08 op 	got_add $file;
308 afdbc7b0 2022-06-29 op 	got_ci (-f $file ? "update $name" : "+$name");
309 b2653662 2022-06-29 op 	say $pass unless $q;
310 5cdf5adc 2022-05-08 op }