Mercurial > cgi-bin > hgweb.cgi > PassMan
view src/main/kotlin/name/blackcap/passman/ReadSubcommand.kt @ 12:a38a2a1036c3
Add import subcommand.
author | David Barts <n5jrn@me.com> |
---|---|
date | Sun, 22 Jan 2023 09:22:53 -0800 |
parents | c69665ff37d0 |
children | ea65ab890f66 |
line wrap: on
line source
package name.blackcap.passman import org.apache.commons.cli.* import kotlin.system.exitProcess class ReadSubcommand(): Subcommand() { private companion object { const val CLIPBOARD = "clipboard" const val HELP = "help" const val LONG = "long" const val ALT_SB = "\u001b[?1049h" const val NORM_SB = "\u001b[?1049l" const val CLEAR = "\u001b[H\u001b[2J" } private lateinit var commandLine: CommandLine override fun run(args: Array<String>) { val options = Options().apply { addOption("c", CLIPBOARD, false, "Copy username and password into clipboard.") addOption("h", HELP, false, "Print this help message.") addOption("l", LONG, false, "Long format listing.") } try { commandLine = DefaultParser().parse(options, args) } catch (e: ParseException) { die(e.message ?: "syntax error", 2) } if (commandLine.hasOption(HELP)) { HelpFormatter().printHelp("$SHORTNAME read [options] name", options) exitProcess(0) } if (commandLine.args.isEmpty()) { die("expecting site name", 2) } if (commandLine.args.size > 1) { die("unexpected trailing arguments", 2) } val nameIn = commandLine.args[0]; val db = Database.open() val entry = Entry.fromDatabase(db, nameIn) if (entry == null) { die("no record matches ${see(nameIn)}") return // Kotlin is too stupid to realize we never get here } try { print(ALT_SB + CLEAR) val redaction = if (commandLine.hasOption(CLIPBOARD)) { "(in clipboard)" } else { null } if (commandLine.hasOption(LONG)) { entry.printLong(redaction) } else { entry.print(redaction) } if (commandLine.hasOption(CLIPBOARD)) { writeToClipboard(entry.password) } name.blackcap.passman.readLine("Press ENTER to continue: ") } finally { print(CLEAR + NORM_SB) entry.password.clear() } db.connection.prepareStatement("update passwords set accessed = ? where id = ?").use { it.setLong(1, System.currentTimeMillis()) it.setLong(2, db.makeKey(nameIn)) it.execute() } } }