1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2024-11-30 16:40:57 -05:00
denoland-deno/http/file_server.ts

242 lines
5.7 KiB
TypeScript
Raw Normal View History

2018-12-09 15:35:26 -05:00
#!/usr/bin/env deno --allow-net
// This program serves files in the current directory over HTTP.
// TODO Stream responses instead of reading them into memory.
// TODO Add tests like these:
// https://github.com/indexzero/http-server/blob/master/test/http-server-test.js
import {
listenAndServe,
ServerRequest,
setContentLength,
Response
} from "./mod.ts";
import { cwd, DenoError, ErrorKind, args, stat, readDir, open } from "deno";
2019-01-10 17:11:44 -05:00
import { extname } from "../fs/path.ts";
2019-01-11 00:16:47 -05:00
import { contentType } from "../media_types/mod.ts";
const dirViewerTemplate = `
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>Deno File Server</title>
<style>
td {
padding: 0 1rem;
}
td.mode {
font-family: Courier;
}
</style>
</head>
<body>
<h1>Index of <%DIRNAME%></h1>
<table>
<tr><th>Mode</th><th>Size</th><th>Name</th></tr>
<%CONTENTS%>
</table>
</body>
</html>
`;
2018-11-07 13:16:07 -05:00
2018-12-23 22:50:49 -05:00
const serverArgs = args.slice();
let CORSEnabled = false;
// TODO: switch to flags if we later want to add more options
for (let i = 0; i < serverArgs.length; i++) {
if (serverArgs[i] === "--cors") {
CORSEnabled = true;
serverArgs.splice(i, 1);
break;
}
}
let currentDir = cwd();
2018-12-23 22:50:49 -05:00
const target = serverArgs[1];
if (target) {
currentDir = `${currentDir}/${target}`;
}
2018-12-23 22:50:49 -05:00
const addr = `0.0.0.0:${serverArgs[2] || 4500}`;
2018-12-09 15:35:26 -05:00
const encoder = new TextEncoder();
function modeToString(isDir: boolean, maybeMode: number | null) {
const modeMap = ["---", "--x", "-w-", "-wx", "r--", "r-x", "rw-", "rwx"];
2018-11-07 13:16:07 -05:00
if (maybeMode === null) {
return "(unknown mode)";
}
const mode = maybeMode!.toString(8);
if (mode.length < 3) {
return "(unknown mode)";
}
let output = "";
mode
.split("")
.reverse()
.slice(0, 3)
.forEach(v => {
output = modeMap[+v] + output;
});
output = `(${isDir ? "d" : "-"}${output})`;
return output;
}
function fileLenToString(len: number) {
const multipler = 1024;
let base = 1;
const suffix = ["B", "K", "M", "G", "T"];
let suffixIndex = 0;
while (base * multipler < len) {
if (suffixIndex >= suffix.length - 1) {
break;
2018-12-09 15:35:26 -05:00
}
base *= multipler;
suffixIndex++;
2018-12-09 15:35:26 -05:00
}
return `${(len / base).toFixed(2)}${suffix[suffixIndex]}`;
}
function createDirEntryDisplay(
name: string,
path: string,
size: number | null,
mode: number | null,
isDir: boolean
) {
const sizeStr = size === null ? "" : "" + fileLenToString(size!);
return `
<tr><td class="mode">${modeToString(
isDir,
mode
)}</td><td>${sizeStr}</td><td><a href="${path}">${name}${
isDir ? "/" : ""
}</a></td>
</tr>
`;
}
// TODO: simplify this after deno.stat and deno.readDir are fixed
async function serveDir(req: ServerRequest, dirPath: string, dirName: string) {
// dirname has no prefix
const listEntry: string[] = [];
const fileInfos = await readDir(dirPath);
for (const info of fileInfos) {
if (info.name === "index.html" && info.isFile()) {
// in case index.html as dir...
2018-12-12 12:47:58 -05:00
return await serveFile(req, info.path);
}
// Yuck!
let mode = null;
try {
mode = (await stat(info.path)).mode;
} catch (e) {}
listEntry.push(
createDirEntryDisplay(
info.name,
dirName + "/" + info.name,
info.isFile() ? info.len : null,
mode,
info.isDirectory()
)
);
}
const page = new TextEncoder().encode(
dirViewerTemplate
.replace("<%DIRNAME%>", dirName + "/")
.replace("<%CONTENTS%>", listEntry.join(""))
);
2018-12-09 15:35:26 -05:00
const headers = new Headers();
headers.set("content-type", "text/html");
const res = {
status: 200,
body: page,
headers
};
setContentLength(res);
2018-12-12 04:38:46 -05:00
return res;
}
async function serveFile(req: ServerRequest, filename: string) {
const file = await open(filename);
const fileInfo = await stat(filename);
const headers = new Headers();
headers.set("content-length", fileInfo.len.toString());
2019-01-11 00:16:47 -05:00
headers.set("content-type", contentType(extname(filename)) || "text/plain");
2018-12-09 15:35:26 -05:00
const res = {
status: 200,
body: file,
headers
};
2018-12-12 04:38:46 -05:00
return res;
}
async function serveFallback(req: ServerRequest, e: Error) {
if (
e instanceof DenoError &&
(e as DenoError<any>).kind === ErrorKind.NotFound
) {
return {
status: 404,
body: encoder.encode("Not found")
2018-12-12 04:38:46 -05:00
};
} else {
2018-12-12 04:38:46 -05:00
return {
status: 500,
body: encoder.encode("Internal server error")
2018-12-12 04:38:46 -05:00
};
}
}
2018-12-12 04:38:46 -05:00
function serverLog(req: ServerRequest, res: Response) {
const d = new Date().toISOString();
const dateFmt = `[${d.slice(0, 10)} ${d.slice(11, 19)}]`;
const s = `${dateFmt} "${req.method} ${req.url} ${req.proto}" ${res.status}`;
console.log(s);
}
2018-12-23 22:50:49 -05:00
function setCORS(res: Response) {
if (!res.headers) {
res.headers = new Headers();
}
res.headers!.append("access-control-allow-origin", "*");
res.headers!.append(
"access-control-allow-headers",
"Origin, X-Requested-With, Content-Type, Accept, Range"
);
}
listenAndServe(addr, async req => {
const fileName = req.url.replace(/\/$/, "");
const filePath = currentDir + fileName;
2018-12-12 04:38:46 -05:00
let response: Response;
try {
const fileInfo = await stat(filePath);
if (fileInfo.isDirectory()) {
// Bug with deno.stat: name and path not populated
// Yuck!
2018-12-12 04:38:46 -05:00
response = await serveDir(req, filePath, fileName);
} else {
2018-12-12 04:38:46 -05:00
response = await serveFile(req, filePath);
}
} catch (e) {
2018-12-12 04:38:46 -05:00
response = await serveFallback(req, e);
} finally {
2018-12-23 22:50:49 -05:00
if (CORSEnabled) {
setCORS(response);
}
2018-12-12 04:38:46 -05:00
serverLog(req, response);
req.respond(response);
}
2018-11-07 13:16:07 -05:00
});
console.log(`HTTP server listening on http://${addr}/`);