Changing crypto keys working

main.go

@@ -279,6 +279,7 @@ func nodeCreate(w http.ResponseWriter, r *http.Request) {// {{{
 	})
 }// }}}
 func nodeUpdate(w http.ResponseWriter, r *http.Request) {// {{{
+	log.Println("/node/update")
 	var err error
 	var session Session
 

static/js/node.mjs

@@ -209,7 +209,8 @@ export class NodeUI extends Component {
 	}//}}}
 	saveNode() {//{{{
 		let content = this.nodeContent.current.contentDiv.current.value
-		this.node.value.save(content, ()=>this.props.app.nodeModified.value = false)
+		this.node.value.setContent(content)
+		this.node.value.save(()=>this.props.app.nodeModified.value = false)
 	}//}}}
 	renameNode() {//{{{
 		let name = prompt("New name")
@@ -397,21 +398,21 @@ export class Node {
 		})
 		.catch(this.app.responseError)
 	}//}}}
-	async save(content, callback) {//{{{
-		let update_content = content
-		/*
-		 * XXX - fix encrypting when saving
-		if(this.CryptoKeyID != 0)
-			update_content = await this.#encrypt(content)
-			*/
+	async save(callback) {//{{{
+		try {
+			await this.#encrypt()
 
-		this.app.request('/node/update', {
+			let req = {
 				NodeID: this.ID,
-			Content: update_content,
+				Content: this._content,
 				CryptoKeyID: this.CryptoKeyID,
-		})
+			}
+			this.app.request('/node/update', req)
 				.then(callback)
 				.catch(this.app.responseError)
+		} catch (err) {
+			this.app.responseError(err)
+		}
 	}//}}}
 	rename(name, callback) {//{{{
 		this.app.request('/node/rename', {
@@ -454,30 +455,26 @@ export class Node {
 		})
 	}//}}}
 	content() {//{{{
-		if(this.CryptoKeyID != 0 && !this._decrypted) {
+		if(this.CryptoKeyID != 0 && !this._decrypted)
 			this.#decrypt()
-		}
 		return this._content
 	}//}}}
-
-	async encrypt(obj_key) {//{{{
-		if(obj_key.ID != this.CryptoKeyID)
-			throw('Invalid key')
-
-		let crypto = new Crypto(obj_key.key)
+	setContent(new_content) {//{{{
+		this._content = new_content
+		if(this.CryptoKeyID == 0)
+			// Logic behind plaintext not being decrypted is that
+			// only encrypted values can be in a decrypted state.
 			this._decrypted = false
-
-		let counter = await obj_key.counter()
-
-		this.content = sjcl.codec.base64.fromBits(
-			crypto.encrypt(
-				sjcl.codec.utf8String.toBits(this.content),
-				counter,
-				false,
-			)
-		)
+		else
+			this._decrypted = true
+	}//}}}
+	async setCryptoKey(new_key) {//{{{
+		return this.#encrypt(true, new_key)
 	}//}}}
 	#decrypt() {//{{{
+		if(this.CryptoKeyID == 0 || this._decrypted)
+			return
+
 		let obj_key = this.app.nodeUI.current.getKey(this.CryptoKeyID)
 		if(obj_key === null || obj_key.ID != this.CryptoKeyID)
 			throw('Invalid key')
@@ -506,10 +503,25 @@ export class Node {
 			crypto.decrypt(this._content)
 		)
 	}//}}}
-	/*
-	async #encrypt(content) {//{{{
-		let obj_key = this.app.nodeUI.current.getKey(this.CryptoKeyID)
-		if(obj_key === null || obj_key.ID != this.CryptoKeyID)
+	async #encrypt(change_key = false, new_key = null) {//{{{
+		// Nothing to do if not changing key and already encrypted.
+		if(!change_key && this.CryptoKeyID != 0 && !this._decrypted)
+			return this._content
+
+		let content = this.content()
+
+		// Changing key to no encryption or already at no encryption -
+		// set to not decrypted (only encrypted values can be
+		// decrypted) and return plain value.
+		if((change_key && new_key === null) || (!change_key && this.CryptoKeyID == 0)) {
+			this._decrypted = false
+			this.CryptoKeyID = 0
+			return content
+		}
+
+		let key_id = change_key ? new_key.ID : this.CryptoKeyID
+		let obj_key = this.app.nodeUI.current.getKey(key_id)
+		if(obj_key === null || obj_key.ID != key_id)
 			throw('Invalid key')
 
 		if(obj_key.status() == 'locked')
@@ -518,9 +530,11 @@ export class Node {
 		let crypto = new Crypto(obj_key.key)
 		let content_bits = sjcl.codec.utf8String.toBits(content)
 		let counter = await this.app.nodeUI.current.keyCounter()
-		return crypto.encrypt(content_bits, counter, true)
+		this.CryptoKeyID = obj_key.ID
+		this._content = crypto.encrypt(content_bits, counter, true)
+		this._decrypted = false
+		return this._content
 	}//}}}
-	*/
 }
 
 class Menu extends Component {
@@ -681,38 +695,26 @@ class NodeProperties extends Component {
 			</div>
 		`
 	}//}}}
-	save() {//{{{
+	async save() {//{{{
 		let nodeui = this.props.nodeui
 		let node = nodeui.node.value
 
 		// Find the actual key object used for encryption
-		let encrypt_key = nodeui.getKey(this.selected_key_id)
-		let decrypt_key = nodeui.getKey(node.CryptoKeyID)
+		let new_key = nodeui.getKey(this.selected_key_id)
+		let current_key = nodeui.getKey(node.CryptoKeyID)
 
-		if(decrypt_key && decrypt_key.status() == 'locked') {
+		if(current_key && current_key.status() == 'locked') {
 			alert("Decryption key is locked and can not be used.")
 			return
 		}
 
-		if(encrypt_key && encrypt_key.status() == 'locked') {
+		if(new_key && new_key.status() == 'locked') {
 			alert("Key is locked and can not be used.")
 			return
 		}
 
-		// Currently not encrypted - encrypt with new key.
-		let crypto = new Crypto(selected_key.key)
-		if(node.CryptoKeyID == 0) {
-			let encrypted = crypto.encrypt(
-				sjcl.codec.utf8String.toBits(node.Content()), 
-				1n,
-			)
-			console.log(encrypted)
-		}
-
-		/*
-		crypto.encrypt(
-		)
-		*/
+		await node.setCryptoKey(new_key)
+		node.save(()=>this.props.nodeui.showPage('node'))
 	}//}}}
 }