|
8
|
1 package name.blackcap.passman
|
|
|
2
|
|
11
|
3 import org.apache.commons.cli.*
|
|
|
4 import java.sql.ResultSet
|
|
|
5 import java.util.*
|
|
|
6 import kotlin.system.exitProcess
|
|
|
7
|
|
8
|
8 class MergeSubcommand(): Subcommand() {
|
|
11
|
9 private companion object {
|
|
|
10 const val FORCE = "force"
|
|
|
11 const val HELP = "help"
|
|
|
12 }
|
|
|
13 private lateinit var commandLine: CommandLine
|
|
|
14 private lateinit var db: Database
|
|
|
15
|
|
8
|
16 override fun run(args: Array<String>) {
|
|
11
|
17 parseArguments(args)
|
|
|
18 db = Database.open()
|
|
|
19 doMerge()
|
|
|
20 }
|
|
|
21
|
|
|
22 private fun parseArguments(args: Array<String>) {
|
|
|
23 val options = Options().apply {
|
|
|
24 addOption("f", MergeSubcommand.FORCE, false, "Do not ask before overwriting.")
|
|
|
25 addOption("h", MergeSubcommand.HELP, false, "Print this help message.")
|
|
|
26 }
|
|
|
27 try {
|
|
|
28 commandLine = DefaultParser().parse(options, args)
|
|
|
29 } catch (e: ParseException) {
|
|
|
30 die(e.message ?: "syntax error", 2)
|
|
|
31 }
|
|
|
32 if (commandLine.hasOption(MergeSubcommand.HELP)) {
|
|
|
33 HelpFormatter().printHelp("$SHORTNAME merge [options] other_database", options)
|
|
|
34 exitProcess(0)
|
|
|
35 }
|
|
|
36 if (commandLine.args.isEmpty()) {
|
|
|
37 die("expecting other database name", 2)
|
|
|
38 }
|
|
|
39 if (commandLine.args.size > 1) {
|
|
|
40 die("unexpected trailing arguments", 2)
|
|
|
41 }
|
|
|
42 }
|
|
|
43
|
|
|
44 private fun doMerge() {
|
|
|
45 val otherFile = commandLine.args[0]
|
|
|
46 val otherDb = Database.open(
|
|
|
47 fileName = otherFile,
|
|
|
48 passwordPrompt = "Key for ${see(otherFile)}: ", create = false
|
|
|
49 )
|
|
|
50 otherDb.connection.prepareStatement("select name, username, password, notes, created, modified, accessed from passwords").use { stmt ->
|
|
|
51 val results = stmt.executeQuery()
|
|
|
52 while (results.next()) {
|
|
|
53 val otherEntry = makeEntry(results)
|
|
|
54 val thisEntry = getEntry(db, otherEntry.name)
|
|
|
55 if (thisEntry == null) {
|
|
|
56 doInsert(otherEntry)
|
|
|
57 } else {
|
|
|
58 doCompare(thisEntry, otherEntry)
|
|
|
59 thisEntry.password.clear()
|
|
|
60 }
|
|
|
61 otherEntry.password.clear()
|
|
|
62 }
|
|
|
63 }
|
|
8
|
64 }
|
|
11
|
65
|
|
|
66 private fun makeEntry(results: ResultSet) = Entry(
|
|
|
67 name = results.getDecryptedString(1, db.encryption)!!,
|
|
|
68 username = results.getDecryptedString(2, db.encryption)!!,
|
|
|
69 password = results.getDecrypted(3, db.encryption)!!,
|
|
|
70 notes = results.getDecryptedString(4, db.encryption),
|
|
|
71 created = results.getDate(5),
|
|
|
72 modified = results.getDate(6),
|
|
|
73 accessed = results.getDate(7)
|
|
|
74 )
|
|
|
75
|
|
|
76 private fun getEntry(otherDb: Database, name: String): Entry? {
|
|
|
77 otherDb.connection.prepareStatement("select name, username, password, notes, created, modified, accessed from passwords where id = ?").use { stmt ->
|
|
|
78 stmt.setLong(1, otherDb.makeKey(name))
|
|
|
79 val results = stmt.executeQuery()
|
|
|
80 return if (results.next()) makeEntry(results) else null
|
|
|
81 }
|
|
|
82 }
|
|
|
83
|
|
|
84 private fun doInsert(entry: Entry) {
|
|
|
85 db.connection.prepareStatement("insert into passwords (id, name, username, password, notes, created, modified, accessed) values (?, ?, ?, ?, ?, ?, ?, ?)")
|
|
|
86 .use {
|
|
|
87 it.setLong(1, db.makeKey(entry.name))
|
|
|
88 it.setEncryptedString(2, entry.name, db.encryption)
|
|
|
89 it.setEncryptedString(3, entry.username, db.encryption)
|
|
|
90 it.setEncrypted(4, entry.password, db.encryption)
|
|
|
91 it.setEncryptedString(5, entry.notes, db.encryption)
|
|
|
92 it.setLongOrNull(6, entry.created?.time)
|
|
|
93 it.setLongOrNull(7, entry.modified?.time)
|
|
|
94 it.setLongOrNull(8, entry.accessed?.time)
|
|
|
95 it.executeUpdate()
|
|
|
96 }
|
|
|
97 }
|
|
|
98
|
|
|
99 private fun doCompare(thisEntry: Entry, otherEntry: Entry) {
|
|
|
100 if (otherEntry.modifiedOrCreated.after(thisEntry.modifiedOrCreated) && okToChange(thisEntry, otherEntry)) {
|
|
|
101 db.connection.prepareStatement("update passwords set name = ?, username = ?, password = ?, notes = ?, modified = ? where id = ?").use {
|
|
|
102 it.setEncryptedString(1, otherEntry.name, db.encryption)
|
|
|
103 it.setEncryptedString(2, otherEntry.username, db.encryption)
|
|
|
104 it.setEncrypted(3, otherEntry.password, db.encryption)
|
|
|
105 it.setEncryptedString(4, otherEntry.notes, db.encryption)
|
|
|
106 it.setLong(5, otherEntry.modifiedOrCreated.time)
|
|
|
107 it.setLong(6, db.makeKey(thisEntry.name))
|
|
|
108 it.executeUpdate()
|
|
|
109 }
|
|
|
110 }
|
|
|
111 }
|
|
|
112
|
|
|
113 private fun okToChange(thisEntry: Entry, otherEntry: Entry): Boolean {
|
|
|
114 if (commandLine.hasOption(MergeSubcommand.FORCE)) {
|
|
|
115 return true
|
|
|
116 }
|
|
|
117 val REDACTED = "(redacted)"
|
|
|
118 println("EXISTING ENTRY:")
|
|
|
119 thisEntry.printLong(REDACTED)
|
|
|
120 println()
|
|
|
121 println("NEWER ENTRY:")
|
|
|
122 otherEntry.printLong(REDACTED)
|
|
|
123 println()
|
|
|
124 val answer = name.blackcap.passman.readLine("OK to overwrite existing entry? ")
|
|
|
125 println()
|
|
|
126 return answer.trimStart().firstOrNull()?.uppercaseChar() in setOf('T', 'Y')
|
|
|
127 }
|
|
|
128
|
|
8
|
129 }
|
