set custom headers

jest.config.ts

@@ -3,7 +3,7 @@ const config: Config = {
     testPathIgnorePatterns: ["/node_modules/", "/dist/"],
     testMatch: [
         "**/__tests__/**/*.[jt]s?(x)",
-        "**/?(*.)+(spec|test|test.e2e).[jt]s?(x)",
+        "**/?(*.)+(spec|test).[jt]s?(x)",
     ],
 };
 

package.json

@@ -9,7 +9,7 @@
         "doc": "jsdoc -c ./jsdoc.config.json",
         "style": "npx prettier ./src --write",
         "test": "npx jest -c jest.config.ts",
-        "e2eTest": "npx jest -c jest.config.e2e.ts",
+        "test:e2e": "npx jest -c jest.config.e2e.ts",
         "unitTest": "npx jest -c jest.config.unit.ts"
     },
     "author": "keivan",

src/lib/InfoScraper/index.ts

@@ -68,4 +68,5 @@ export const getPodcastInfo = async (url: string): Promise<PodcastInfo> => {
     return infoAndUrl;
 };
 
+export {setRequestHeaders} from "./utils"
 export * from "./interfaces";

src/lib/InfoScraper/interfaces.ts

@@ -57,6 +57,9 @@ export interface PodcastInfo extends MusicAndVideoCommonInfo {
     relatedTracks: Track[];
 }
 
+export type headersInputType = {[key in string]:string}
+
+
 /**
  * Music details
  *

src/lib/InfoScraper/utils/__test__/setRequestHeaders.test.ts

@@ -0,0 +1,25 @@
+import { setRequestHeaders } from '../index';
+import { getRequestHeaders } from '../index';
+import { headersInputType } from '../../interfaces';
+describe("setRequestHeaders()", () => {
+
+
+    it("should use default headers when not called setRequestHeaders()", () => {
+        expect(getRequestHeaders())
+            .toEqual({
+               "user-agent":
+                 "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36",
+             })
+    })
+
+    it("should set custom headers", () => {
+        const headers:headersInputType = {
+            "user-agent": "xxxx xxx xxx",
+            "user-jwt":"xxxxxx.xxxxxxxx.xxxx"
+        }
+        setRequestHeaders(headers)
+        expect(getRequestHeaders()).toEqual(headers)
+    })
+
+
+})

src/lib/InfoScraper/utils/index.ts

@@ -1,14 +1,39 @@
 import axios from "axios";
 import { JSDOM } from "jsdom";
+import { headersInputType } from "../interfaces";
 export * from "./PageScraper";
+
+let requestHeaders:unknown = {
+    "user-agent":
+        "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36",
+}
+
+
+export function getRequestHeaders() {
+    return requestHeaders
+}
+
+/**
+ * Used to set custom headers for requests
+ *
+ * @example
+ * setRequestHeaders({
+ *      "user-agent":"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36"
+ *      "Authorization":"token"
+ * })
+ * @param {headersInputType} headers
+ */
+export function setRequestHeaders(headers:headersInputType) {
+    requestHeaders = headers
+}
+
+
 export const getSourceCodeDOMDocument = async (url: string) => {
+    const headers = getRequestHeaders()
     const sourceCode = (
         await axios.get(url, {
             method: "GET",
-            headers: {
-                "user-agent":
-                    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36",
-            },
+            headers: headers,
         })
     ).data;
     const dom = new JSDOM(sourceCode);

src/lib/index.ts

@@ -24,6 +24,7 @@ import {
     AlbumInfo,
 } from "./InfoScraper";
 import { Music, Podcast, Video } from "./interfaces";
+import { setRequestHeaders } from './InfoScraper/utils/index';
 
 export {
     formatURL,
@@ -145,3 +146,5 @@ export const getAlbum = async (url: string): Promise<AlbumInfo> =>
  */
 export const getPlaylist = async (url: string): Promise<PlaylistInfo> =>
     getPlaylistInfo(url);
+
+export {setRequestHeaders} from "./InfoScraper"