gitlit/app/main.js

const {app, BrowserWindow, ipcMain} = require('electron');
const path = require('path');
const url = require('url');
const electronLocalshortcut = require('electron-localshortcut');
const exec = require('child_process').exec;
const args = require('minimist')(process.defaultApp ? process.argv.slice(2) : process.argv.slice(1), {
	default: {
		_: process.cwd()
	}
});

const repoDir = path.resolve(path.normalize(args._.join(' ')));

function getLfsFileList(dir, cb) {
	exec('git lfs ls-files', {maxBuffer: 1024 * 1024}, {
		cwd: dir
	},
	(error, stdout, stderr) => {
		if (error) {
			cb(error);
			return;
		}

		let parsedFiles = [];
		if (stdout) {
			let files = stdout.split('\n');
			files.forEach((file) => {
				file = file.split(' * ');
				if (file[1]) {
					parsedFiles.push(file[1].trim());
				}
			});

			cb(null, parsedFiles);
		} else {
			cb(null, parsedFiles);
		}
	});
};

function getLfsLocks(dir, cb) {
	exec('git lfs locks', {
		cwd: dir
	},
	(error, stdout, stderr) => {
		if (error) {
			cb(error);
			return;
		}

		let parsedFiles = [];
		if (stdout) {
			let files = stdout.split('\n');
			files.forEach((file) => {
				if (file) {
					let fileName = file.split('\t')[0].trim();
					let lockedBy = file.split('\t')[1].trim();
					let id = file.split('ID:')[1].trim();
					parsedFiles.push({
						file: fileName,
						lockedBy: lockedBy,
						id: id
					});
				}
			});

			cb(null, parsedFiles);
		} else {
			cb(null, parsedFiles);
		}
	});
};

function getArrayObjectByKey(array, key, value, defaultKeyValue) {
	let o = array.filter((e) => {
		return e[key] === value;
	});
	if (o.length > 0) {
		return defaultKeyValue ? o[0][defaultKeyValue] : o[0];
	}
	return undefined;
}

function createWindow() {
	// Create the browser window.
	win = new BrowserWindow({width: 800, height: 600});
	win.setMenu(null);

	// and load the index.html of the app.
	win.loadURL(url.format({
		pathname: path.join(__dirname, 'index.html'),
		protocol: 'file:',
		slashes: true
	}))

	electronLocalshortcut.register(win, 'F12', () => {
		win.webContents.toggleDevTools();
	});

	win.webContents.on('did-finish-load', () => {
		win.webContents.send('repoDir', repoDir);

		getLfsFileList(repoDir, (err, files) => {
			if (err) {
				console.error(err);

				win.webContents.send('isNoGitLfsRepo', repoDir);
				return;
			}
			getLfsLocks(repoDir, (err, lockedFiles) => {
				if (err) {
					console.error(err);

					win.webContents.send('isNoGitLfsRepo', repoDir);
					return;
				}
				let allFiles = [];

				files.forEach((file) => {
					const t = {
						file: file,
						lockedBy: getArrayObjectByKey(lockedFiles, 'file', file, 'lockedBy'),
						id: getArrayObjectByKey(lockedFiles, 'file', file, 'id')
					};

					allFiles.push(t);
				});

				win.webContents.send('fileList', allFiles);
			});
		});
	});
}

ipcMain.on('unlock', (event, file) => {
	exec('git lfs unlock "' + file + '"', {maxBuffer: 1024 * 1024}, {
		cwd: repoDir
	},
	(error, stdout, stderr) => {
		let notification = {
			message:  (error && error.message) || stderr,
			type: 'error'
		};

		if (stdout) {
			notification = {
				file: file,
				event: 'unlock',
				type: 'info'
			};
		}

		win.webContents.send('notification', notification);
	});
});

ipcMain.on('lock', (event, file) => {
	exec('git lfs lock --json "' + file + '"', {maxBuffer: 1024 * 1024}, {
		cwd: repoDir
	},
	(error, stdout, stderr) => {
		let notification = {
			message:  (error && error.message) || stderr,
			type: 'error'
		};

		if (stdout) {
			notification = {
				file: file,
				event: 'lock',
				data: JSON.parse(stdout),
				type: 'info'
			};
		}

		win.webContents.send('notification', notification);
	});
});

app.on('ready', createWindow);

app.on('window-all-closed', function() {
	if (process.platform != 'darwin') {
		app.quit();
	}
});