chore: Clean up code related to headers.

src/ldp/UnsecureWebSocketsProtocol.ts

@@ -95,7 +95,7 @@ class WebSocketListener extends EventEmitter {
         throw new Error(`Mismatched protocol: ${resolved.protocol} instead of ${this.protocol}`);
       }
       // Subscribe to the URL
-      const url = resolved.toString();
+      const url = resolved.href;
       this.subscribedPaths.add(url);
       this.sendMessage('ack', url);
       this.logger.debug(`WebSocket subscribed to changes on ${url}`);

src/ldp/http/BasicTargetExtractor.ts

@@ -13,23 +13,25 @@ import { TargetExtractor } from './TargetExtractor';
 export class BasicTargetExtractor extends TargetExtractor {
   protected readonly logger = getLoggerFor(this);
 
-  public async handle(request: HttpRequest): Promise<ResourceIdentifier> {
+  public async handle({ url, headers: { host }, connection }: HttpRequest): Promise<ResourceIdentifier> {
-    if (!request.url) {
+    if (!url) {
       this.logger.error('The request has no URL');
       throw new Error('Missing URL');
     }
-    if (!request.headers.host) {
+    if (!host) {
       this.logger.error('The request has no Host header');
       throw new Error('Missing Host header');
     }
+    if (/[/\\*]/u.test(host)) {
+      throw new Error(`The request has an invalid Host header: ${host}`);
+    }
 
-    const isHttps = request.connection && (request.connection as TLSSocket).encrypted;
+    const isHttps = (connection as TLSSocket)?.encrypted;
     this.logger.debug(`Request is using HTTPS: ${isHttps}`);
 
     // URL object applies punycode encoding to domain
-    const base = `http${isHttps ? 's' : ''}://${request.headers.host}`;
+    const base = `http${isHttps ? 's' : ''}://${host}`;
-    const url = toCanonicalUriPath(request.url);
+    const path = new URL(toCanonicalUriPath(url), base).href;
-    const path = new URL(url, base).href;
 
     return { path };
   }

test/unit/ldp/http/BasicTargetExtractor.test.ts

@@ -15,18 +15,28 @@ describe('A BasicTargetExtractor', (): void => {
     await expect(extractor.handle({ url: 'url', headers: {}} as any)).rejects.toThrow('Missing Host header');
   });
 
+  it('errors if the host is invalid.', async(): Promise<void> => {
+    await expect(extractor.handle({ url: 'url', headers: { host: 'test.com/forbidden' }} as any))
+      .rejects.toThrow('The request has an invalid Host header: test.com/forbidden');
+  });
+
   it('returns the input URL.', async(): Promise<void> => {
     await expect(extractor.handle({ url: 'url', headers: { host: 'test.com' }} as any))
       .resolves.toEqual({ path: 'http://test.com/url' });
   });
 
+  it('supports host:port combinations.', async(): Promise<void> => {
+    await expect(extractor.handle({ url: 'url', headers: { host: 'localhost:3000' }} as any))
+      .resolves.toEqual({ path: 'http://localhost:3000/url' });
+  });
+
   it('uses https protocol if the connection is secure.', async(): Promise<void> => {
     await expect(extractor.handle(
       { url: 'url', headers: { host: 'test.com' }, connection: { encrypted: true } as any } as any,
     )).resolves.toEqual({ path: 'https://test.com/url' });
   });
 
-  it('encodes relevant characters.', async(): Promise<void> => {
+  it('encodes paths.', async(): Promise<void> => {
     await expect(extractor.handle({ url: '/a%20path%26/name', headers: { host: 'test.com' }} as any))
       .resolves.toEqual({ path: 'http://test.com/a%20path%26/name' });
 
@@ -36,4 +46,9 @@ describe('A BasicTargetExtractor', (): void => {
     await expect(extractor.handle({ url: '/path&%26/name', headers: { host: 'test.com' }} as any))
       .resolves.toEqual({ path: 'http://test.com/path%26%26/name' });
   });
+
+  it('encodes hosts.', async(): Promise<void> => {
+    await expect(extractor.handle({ url: '/', headers: { host: '點看' }} as any))
+      .resolves.toEqual({ path: 'http://xn--c1yn36f/' });
+  });
 });

test/unit/ldp/http/metadata/LinkRelMetadataWriter.test.ts

@@ -1,27 +1,16 @@
+import { createResponse } from 'node-mocks-http';
 import { LinkRelMetadataWriter } from '../../../../../src/ldp/http/metadata/LinkRelMetadataWriter';
 import { RepresentationMetadata } from '../../../../../src/ldp/representation/RepresentationMetadata';
-import * as util from '../../../../../src/util/HeaderUtil';
 import { LDP, RDF } from '../../../../../src/util/UriConstants';
 import { toNamedNode } from '../../../../../src/util/UriUtil';
 
 describe('A LinkRelMetadataWriter', (): void => {
   const writer = new LinkRelMetadataWriter({ [RDF.type]: 'type', dummy: 'dummy' });
-  let mock: jest.SpyInstance;
-  let addHeaderMock: jest.Mock;
-
-  beforeEach(async(): Promise<void> => {
-    addHeaderMock = jest.fn();
-    mock = jest.spyOn(util, 'addHeader').mockImplementation(addHeaderMock);
-  });
-
-  afterEach(async(): Promise<void> => {
-    mock.mockRestore();
-  });
 
   it('adds the correct link headers.', async(): Promise<void> => {
+    const response = createResponse();
     const metadata = new RepresentationMetadata({ [RDF.type]: toNamedNode(LDP.Resource), unused: 'text' });
-    await expect(writer.handle({ response: 'response' as any, metadata })).resolves.toBeUndefined();
+    await expect(writer.handle({ response, metadata })).resolves.toBeUndefined();
-    expect(addHeaderMock).toHaveBeenCalledTimes(1);
+    expect(response.getHeaders()).toEqual({ link: `<${LDP.Resource}>; rel="type"` });
-    expect(addHeaderMock).toHaveBeenLastCalledWith('response', 'link', [ `<${LDP.Resource}>; rel="type"` ]);
   });
 });

test/unit/ldp/http/metadata/MappedMetadataWriter.test.ts

@@ -1,26 +1,15 @@
+import { createResponse } from 'node-mocks-http';
 import { MappedMetadataWriter } from '../../../../../src/ldp/http/metadata/MappedMetadataWriter';
 import { RepresentationMetadata } from '../../../../../src/ldp/representation/RepresentationMetadata';
-import * as util from '../../../../../src/util/HeaderUtil';
 import { CONTENT_TYPE } from '../../../../../src/util/UriConstants';
 
 describe('A MappedMetadataWriter', (): void => {
   const writer = new MappedMetadataWriter({ [CONTENT_TYPE]: 'content-type', dummy: 'dummy' });
-  let mock: jest.SpyInstance;
-  let addHeaderMock: jest.Mock;
-
-  beforeEach(async(): Promise<void> => {
-    addHeaderMock = jest.fn();
-    mock = jest.spyOn(util, 'addHeader').mockImplementation(addHeaderMock);
-  });
-
-  afterEach(async(): Promise<void> => {
-    mock.mockRestore();
-  });
 
   it('adds metadata to the corresponding header.', async(): Promise<void> => {
+    const response = createResponse();
     const metadata = new RepresentationMetadata({ [CONTENT_TYPE]: 'text/turtle', unused: 'text' });
-    await expect(writer.handle({ response: 'response' as any, metadata })).resolves.toBeUndefined();
+    await expect(writer.handle({ response, metadata })).resolves.toBeUndefined();
-    expect(addHeaderMock).toHaveBeenCalledTimes(1);
+    expect(response.getHeaders()).toEqual({ 'content-type': 'text/turtle' });
-    expect(addHeaderMock).toHaveBeenLastCalledWith('response', 'content-type', [ 'text/turtle' ]);
   });
 });