azot 0.7.1 → 0.8.0

package/dist/cli/main.cjs

@@ -7820,8 +7820,8 @@ license.help = help2;
 // package.json
 var package_default = {
   name: "azot",
-  version: "0.7.1",
-  description: "Swiss Army knife for pentesting DRMs",
+  version: "0.8.0",
+  description: "Swiss Army knife for pentesting DRM systems",
   type: "module",
   files: [
     "dist",
@@ -7830,7 +7830,7 @@ var package_default = {
   main: "dist/lib/main.cjs",
   module: "dist/lib/main.js",
   bin: {
-    azot: "dist/cli/main.js"
+    azot: "dist/cli/main.cjs"
   },
   scripts: {
     prepublishOnly: "npm run build:lib && npm run build:cli",

package/dist/lib/main.cjs

@@ -34,6 +34,7 @@ __export(main_exports, {
   Client: () => Client,
   Key: () => Key,
   Session: () => Session,
+  connect: () => connect,
   fetchDecryptionKeys: () => fetchDecryptionKeys,
   fromBase64: () => fromBase64,
   fromBinary: () => fromBinary,
@@ -7400,6 +7401,47 @@ var defaultResponseFilter = (response, log = console) => {
   return response;
 };
 
+// src/lib/connect.ts
+var connect = ({ baseUrl, secret }) => {
+  const headers = {
+    "x-secret-key": secret,
+    "content-type": "application/json"
+  };
+  const json = (data) => JSON.stringify(data);
+  const post = async (route, body) => {
+    const response = await fetch(`${baseUrl}${route}`, {
+      method: "POST",
+      headers,
+      body: json(body)
+    });
+    const contentLength = response.headers.get("content-length");
+    if (contentLength === "0") return;
+    return response.json();
+  };
+  const get = async (route) => {
+    const response = await fetch(`${baseUrl}${route}`, {
+      method: "GET",
+      headers
+    });
+    return response.json();
+  };
+  const createSession = async (client) => {
+    const data = await post(`/session`, { client });
+    const sessionId = data.id;
+    const generateRequest = async (initData, initDataType) => {
+      const data2 = await post(`/session/${sessionId}/generate-request`, {
+        initDataType,
+        initData
+      });
+      return data2.licenseRequest;
+    };
+    const update = async (response) => post(`/session/${sessionId}/update`, { response });
+    const keys = async () => get(`/session/${sessionId}/keys`);
+    return { sessionId, generateRequest, update, keys };
+  };
+  return { createSession };
+};
+
 // src/lib/wvd.ts
 var VERSION = 2;
 var MAGIC = fromText("WVD").toBuffer();
@@ -7720,6 +7762,7 @@ var fetchDecryptionKeys = async (params) => {
   Client,
   Key,
   Session,
+  connect,
   fetchDecryptionKeys,
   fromBase64,
   fromBinary,

package/dist/lib/main.d.cts

@@ -2396,6 +2396,19 @@ declare class Client {
     };
 }
 
+type ConnectParams = {
+    baseUrl: string;
+    secret: string;
+};
+declare const connect: ({ baseUrl, secret }: ConnectParams) => {
+    createSession: (client: string) => Promise<{
+        sessionId: any;
+        generateRequest: (initData: string, initDataType: string) => Promise<any>;
+        update: (response: string) => Promise<any>;
+        keys: () => Promise<any>;
+    }>;
+};
+
 declare const fromText: (data: string) => {
     toBase64: () => string;
     toHex: () => string;
@@ -2435,4 +2448,4 @@ interface FetchDecryptionKeysParams {
 }
 declare const fetchDecryptionKeys: (params: FetchDecryptionKeysParams) => Promise<Key[]>;
 
-export { CLIENT_TYPE, Client, type ClientType, Key, SecurityLevel, Session, fetchDecryptionKeys, fromBase64, fromBinary, fromBuffer, fromHex, fromText };
+export { CLIENT_TYPE, Client, type ClientType, Key, SecurityLevel, Session, connect, fetchDecryptionKeys, fromBase64, fromBinary, fromBuffer, fromHex, fromText };

package/dist/lib/main.d.ts

@@ -2396,6 +2396,19 @@ declare class Client {
     };
 }
 
+type ConnectParams = {
+    baseUrl: string;
+    secret: string;
+};
+declare const connect: ({ baseUrl, secret }: ConnectParams) => {
+    createSession: (client: string) => Promise<{
+        sessionId: any;
+        generateRequest: (initData: string, initDataType: string) => Promise<any>;
+        update: (response: string) => Promise<any>;
+        keys: () => Promise<any>;
+    }>;
+};
+
 declare const fromText: (data: string) => {
     toBase64: () => string;
     toHex: () => string;
@@ -2435,4 +2448,4 @@ interface FetchDecryptionKeysParams {
 }
 declare const fetchDecryptionKeys: (params: FetchDecryptionKeysParams) => Promise<Key[]>;
 
-export { CLIENT_TYPE, Client, type ClientType, Key, SecurityLevel, Session, fetchDecryptionKeys, fromBase64, fromBinary, fromBuffer, fromHex, fromText };
+export { CLIENT_TYPE, Client, type ClientType, Key, SecurityLevel, Session, connect, fetchDecryptionKeys, fromBase64, fromBinary, fromBuffer, fromHex, fromText };

package/dist/lib/main.js

@@ -7355,6 +7355,47 @@ var defaultResponseFilter = (response, log = console) => {
   return response;
 };
 
+// src/lib/connect.ts
+var connect = ({ baseUrl, secret }) => {
+  const headers = {
+    "x-secret-key": secret,
+    "content-type": "application/json"
+  };
+  const json = (data) => JSON.stringify(data);
+  const post = async (route, body) => {
+    const response = await fetch(`${baseUrl}${route}`, {
+      method: "POST",
+      headers,
+      body: json(body)
+    });
+    const contentLength = response.headers.get("content-length");
+    if (contentLength === "0") return;
+    return response.json();
+  };
+  const get = async (route) => {
+    const response = await fetch(`${baseUrl}${route}`, {
+      method: "GET",
+      headers
+    });
+    return response.json();
+  };
+  const createSession = async (client) => {
+    const data = await post(`/session`, { client });
+    const sessionId = data.id;
+    const generateRequest = async (initData, initDataType) => {
+      const data2 = await post(`/session/${sessionId}/generate-request`, {
+        initDataType,
+        initData
+      });
+      return data2.licenseRequest;
+    };
+    const update = async (response) => post(`/session/${sessionId}/update`, { response });
+    const keys = async () => get(`/session/${sessionId}/keys`);
+    return { sessionId, generateRequest, update, keys };
+  };
+  return { createSession };
+};
+
 // src/lib/wvd.ts
 var VERSION = 2;
 var MAGIC = fromText("WVD").toBuffer();
@@ -7674,6 +7715,7 @@ export {
   Client,
   Key,
   Session,
+  connect,
   fetchDecryptionKeys,
   fromBase64,
   fromBinary,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "azot",
-  "version": "0.7.1",
-  "description": "Swiss Army knife for pentesting DRMs",
+  "version": "0.8.0",
+  "description": "Swiss Army knife for pentesting DRM systems",
   "type": "module",
   "files": [
     "dist",
@@ -10,7 +10,7 @@
   "main": "dist/lib/main.cjs",
   "module": "dist/lib/main.js",
   "bin": {
-    "azot": "dist/cli/main.js"
+    "azot": "dist/cli/main.cjs"
   },
   "scripts": {
     "prepublishOnly": "npm run build:lib && npm run build:cli",