joplin/ReactNativeClient/lib/components/shared/encryption-config-shared.js

const EncryptionService = require('lib/services/EncryptionService');
const { _ } = require('lib/locale.js');
const BaseItem = require('lib/models/BaseItem.js');
const Setting = require('lib/models/Setting.js');

const shared = {};

shared.constructor = function(comp) {
	comp.state = {
		masterKeys: [],
		passwords: {},
		passwordChecks: {},
		stats: {
			encrypted: null,
			total: null,
		},
	};
	comp.isMounted_ = false;

	comp.refreshStatsIID_ = null;
}

shared.initState = function(comp, props) {
	comp.setState({
		masterKeys: props.masterKeys,
		passwords: props.passwords ? props.passwords : {},
	}, () => {
		comp.checkPasswords();
	});

	comp.refreshStats();

	if (comp.refreshStatsIID_) {
		clearInterval(comp.refreshStatsIID_);
		comp.refreshStatsIID_ = null;
	}

	comp.refreshStatsIID_ = setInterval(() => {
		if (!comp.isMounted_) {
			clearInterval(comp.refreshStatsIID_);
			comp.refreshStatsIID_ = null;
			return;
		}
		comp.refreshStats();
	}, 3000);
}

shared.refreshStats = async function(comp) {
	const stats = await BaseItem.encryptedItemsStats();
	comp.setState({ stats: stats });
}

shared.checkPasswords = async function(comp) {
	const passwordChecks = Object.assign({}, comp.state.passwordChecks);
	for (let i = 0; i < comp.state.masterKeys.length; i++) {
		const mk = comp.state.masterKeys[i];
		const password = comp.state.passwords[mk.id];
		const ok = password ? await EncryptionService.instance().checkMasterKeyPassword(mk, password) : false;
		passwordChecks[mk.id] = ok;
	}
	comp.setState({ passwordChecks: passwordChecks });
}

shared.decryptedStatText = function(comp) {
	const stats = comp.state.stats;
	const doneCount = stats.encrypted !== null ? (stats.total - stats.encrypted) : '-';
	const totalCount = stats.total !== null ? stats.total : '-';
	return _('Decrypted items: %s / %s', doneCount, totalCount);
}

shared.onSavePasswordClick = function(comp, mk) {
	const password = comp.state.passwords[mk.id];
	if (!password) {
		Setting.deleteObjectKey('encryption.passwordCache', mk.id);
	} else {
		Setting.setObjectKey('encryption.passwordCache', mk.id, password);
	}

	comp.checkPasswords();
}

shared.onPasswordChange = function(comp, mk, password) {
	const passwords = comp.state.passwords;
	passwords[mk.id] = password;
	comp.setState({ passwords: passwords });
}

module.exports = shared;