rearrange lib

LiviaMedeiros · Aug 26, 2023 · 67f492d · 67f492d
1 parent 9a8321a
commit 67f492d
Show file tree

Hide file tree

Showing 4 changed files with 86 additions and 78 deletions.
diff --git a/lib/generic.mjs b/lib/generic.mjs
@@ -1,13 +1,7 @@
-import { sep } from 'node:path';
-import { cwd } from 'node:process';
-import { pathToFileURL } from 'node:url';
-
 const requestConstructor = Reflect.construct.bind(Reflect, Request);
 const responseConstructor = Reflect.construct.bind(Reflect, Response);
-const getCwdURL = () => pathToFileURL(cwd() + sep).href;
 
 export {
-  getCwdURL,
   requestConstructor,
   responseConstructor,
 };
diff --git a/lib/http.mjs b/lib/http.mjs
@@ -1,6 +1,9 @@
 import { readlink } from 'node:fs/promises';
 import { pathToFileURL } from 'node:url';
 
+const _params = Object.fromEntries(new URL(import.meta.url).searchParams.entries());
+const PREFIX = `X-${_params.prefix ?? 'Poteto'}-`;
+
 const errorAsUndefined = () => {};
 
 const HEADERS_ALLOW = Object.freeze(new Headers({
@@ -73,6 +76,26 @@ const genericResponse = (status = 200, {
     ...opts,
   });
 
+const adjustHeaders = ({ headers }) =>
+  new Headers([
+    ...getGenericHeaders(),
+    ...Object.entries(headers).map(([$, _]) => [
+      `${PREFIX}${$}`,
+      _ instanceof Date ? _.toISOString() : `${_}`,
+    ]),
+
+    // TODO: Etag?
+    // TODO: Content-Type?
+    ['Content-Length', headers.size],
+    ['Last-Modified', headers.mtime?.toUTCString()],
+  ].filter(([, $]) => $ !== undefined));
+
+const statsAsOptions = async (statsOrError, { status } = {}) =>
+  Promise.resolve(statsOrError)
+    .catch($ => $)
+    .then(headers => ({ ...errAsStatus(headers, status), headers }))
+    .then($ => Object.assign($, { headers: adjustHeaders($) }));
+
 const blockingHooks = [
   // if request.redirect is not follow,
   async (url, { redirect }) => {
@@ -100,9 +123,8 @@ const getRanges = headers =>
   headers.get('Range')?.split(';')[0].split('=')[1]?.split(',').map($ => $.trim()) ?? [];
 
 export {
-  errAsStatus,
   genericResponse,
-  getGenericHeaders,
   getRanges,
   preHooks,
+  statsAsOptions,
 };
diff --git a/lib/poteto.mjs b/lib/poteto.mjs
@@ -2,18 +2,16 @@ import fs from 'node:fs/promises';
 import { Writable } from 'node:stream';
 
 import {
-  errAsStatus,
   genericResponse,
-  getGenericHeaders,
   getRanges,
   preHooks,
+  statsAsOptions,
 } from './http.mjs';
 import {
   STAT_OPTS,
   READDIR_OPTS,
 } from './constants.mjs';
 import {
-  getCwdURL,
   responseConstructor,
 } from './generic.mjs';
 import {
@@ -23,74 +21,11 @@ import {
   readRange,
   writeRange,
 } from './fs.mjs';
+import {
+  PotetoRequest,
+  ProxyRequest,
+} from './request.mjs';
 
-const _params = Object.fromEntries(new URL(import.meta.url).searchParams.entries());
-const cwdURL = _params.persistCwd
-  ? getCwdURL()
-  : { [Symbol.toPrimitive]: getCwdURL };
-const PREFIX = `X-${_params.prefix ?? 'Poteto'}-`;
-
-const adjustHeaders = ({ headers }) =>
-  new Headers([
-    ...getGenericHeaders(),
-    ...Object.entries(headers).map(([$, _]) => [
-      `${PREFIX}${$}`,
-      _ instanceof Date ? _.toISOString() : `${_}`,
-    ]),
-
-    // TODO: Etag?
-    // TODO: Content-Type?
-    ['Content-Length', headers.size],
-    ['Last-Modified', headers.mtime?.toUTCString()],
-  ].filter(([, $]) => $ !== undefined));
-
-const statsAsOptions = async (statsOrError, { status } = {}) =>
-  Promise.resolve(statsOrError)
-    .catch($ => $)
-    .then(headers => ({ ...errAsStatus(headers, status), headers }))
-    .then($ => Object.assign($, { headers: adjustHeaders($) }));
-
-const requestInitOptions = new Set([
-  'method',
-  'headers',
-  'body',
-  'referrer',
-  'referrerPolicy',
-  'mode',
-  'credentials',
-  'cache',
-  'redirect',
-  'integrity',
-  'keepalive',
-  'signal',
-  'duplex',
-  'priority',
-  'window',
-]);
-
-const potetoRequestInit = ([ resource, options ], requestInit = {}, url = null) => {
-  if (resource instanceof Request) {
-    requestInitOptions.forEach($ => requestInit[$] = resource[$]);
-    resource = resource.url;
-  }
-  return [
-    (url = new URL(resource, cwdURL)).protocol === 'file:'
-      ? url
-      : resource,
-    { ...requestInit, ...options },
-  ];
-};
-
-class PotetoRequest extends Request {
-  constructor(...$) {
-    super(...potetoRequestInit($));
-  }
-}
-
-const ProxyRequest = new Proxy(Request, {
-  construct: (target, argumentsList) =>
-    Reflect.construct(target, potetoRequestInit(argumentsList))
-});
 
 const DELETE = async url =>
   fs.rm(url).then(() => genericResponse(204));

diff --git a/lib/request.mjs b/lib/request.mjs
@@ -0,0 +1,57 @@
+import { sep } from 'node:path';
+import { cwd } from 'node:process';
+import { pathToFileURL } from 'node:url';
+
+const getCwdURL = () => pathToFileURL(cwd() + sep).href;
+
+const _params = Object.fromEntries(new URL(import.meta.url).searchParams.entries());
+const cwdURL = _params.persistCwd
+  ? getCwdURL()
+  : { [Symbol.toPrimitive]: getCwdURL };
+
+const requestInitOptions = new Set([
+  'method',
+  'headers',
+  'body',
+  'referrer',
+  'referrerPolicy',
+  'mode',
+  'credentials',
+  'cache',
+  'redirect',
+  'integrity',
+  'keepalive',
+  'signal',
+  'duplex',
+  'priority',
+  'window',
+]);
+
+const potetoRequestInit = ([ resource, options ], requestInit = {}, url = null) => {
+  if (resource instanceof Request) {
+    requestInitOptions.forEach($ => requestInit[$] = resource[$]);
+    resource = resource.url;
+  }
+  return [
+    (url = new URL(resource, cwdURL)).protocol === 'file:'
+      ? url
+      : resource,
+    { ...requestInit, ...options },
+  ];
+};
+
+class PotetoRequest extends Request {
+  constructor(...$) {
+    super(...potetoRequestInit($));
+  }
+}
+
+const ProxyRequest = new Proxy(Request, {
+  construct: (target, argumentsList) =>
+    Reflect.construct(target, potetoRequestInit(argumentsList))
+});
+
+export {
+  PotetoRequest,
+  ProxyRequest,
+};