code-server/test/ui/tree/server.js

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

const fs = require('mz/fs');
const path = require('path');
const Koa = require('koa');
const _ = require('koa-route');
const serve = require('koa-static');
const mount = require('koa-mount');

const app = new Koa();
const root = path.dirname(path.dirname(__dirname));

async function getTree(fsPath, level) {
	const element = path.basename(fsPath);
	const stat = await fs.stat(fsPath);

	if (!stat.isDirectory() || element === '.git' || element === '.build' || level >= 4) {
		return { element };
	}

	const childNames = await fs.readdir(fsPath);
	const children = await Promise.all(childNames.map(async childName => await getTree(path.join(fsPath, childName), level + 1)));
	return { element, collapsible: true, collapsed: false, children };
}

async function readdir(relativePath) {
	const absolutePath = relativePath ? path.join(root, relativePath) : root;
	const childNames = await fs.readdir(absolutePath);
	const childStats = await Promise.all(childNames.map(async name => await fs.stat(path.join(absolutePath, name))));
	const result = [];

	for (let i = 0; i < childNames.length; i++) {
		const name = childNames[i];
		const path = relativePath ? `${relativePath}/${name}` : name;
		const stat = childStats[i];

		if (stat.isFile()) {
			result.push({ type: 'file', name, path });
		} else if (!stat.isDirectory() || name === '.git' || name === '.build') {
			continue;
		} else {
			result.push({ type: 'dir', name, path });
		}
	}

	return result;
}

app.use(serve('public'));
app.use(mount('/static', serve('../../out')));
app.use(_.get('/api/ls', async ctx => {
	const relativePath = ctx.query.path;
	const absolutePath = path.join(root, relativePath);

	ctx.body = await getTree(absolutePath, 0);
}));

app.use(_.get('/api/readdir', async ctx => {
	const relativePath = ctx.query.path;

	ctx.body = await readdir(relativePath);
}));

app.listen(3000);
console.log('http://localhost:3000');