chapybara 0.4.0 → 0.6.0

package/README.md

@@ -91,24 +91,6 @@ async function getWebTechInfo() {
 getWebTechInfo();
 ```
 
-### Get Website Screenshot
-
-```javascript
-import fs from "fs/promises";
-
-async function getWebsiteScreenshot() {
-  try {
-    const imageBuffer = await chapybara.screenshot.get("apple.com");
-    await fs.writeFile("screenshot.png", imageBuffer);
-    console.log("Screenshot saved to screenshot.png");
-  } catch (error) {
-    console.error(`Error fetching screenshot: ${error.message}`);
-  }
-}
-
-getWebsiteScreenshot();
-```
-
 ### Get Account Information
 
 ```javascript
@@ -116,7 +98,6 @@ async function getAccountDetails() {
   try {
     const data = await chapybara.account.getInfo();
     console.log("Domain quota remaining:", data.quotas.domain.remaining);
-    console.log("Screenshot quota remaining:", data.quotas.screenshot.remaining);
   } catch (error) {
     console.error(`Error fetching account info: ${error.message}`);
   }

package/index.d.ts

@@ -1,5 +1,4 @@
 import type { LRUCache } from "lru-cache";
-import type { Buffer } from "node:buffer";
 
 declare module "chapybara" {
   interface ChapybaraClientOptions {
@@ -241,7 +240,6 @@ declare module "chapybara" {
       domain: { limit: number; used: number; remaining: number };
       ip: { limit: number; used: number; remaining: number };
       webtech: { limit: number; used: number; remaining: number };
-      screenshot: { limit: number; used: number; remaining: number };
       reset_date: string;
     };
     api_key: {
@@ -282,14 +280,12 @@ declare module "chapybara" {
       getScanner: (domain: string) => Promise<WebTechResponse>;
     };
 
-    screenshot: {
-      get: (domain: string) => Promise<Buffer>;
-    };
-
     account: {
       getInfo: () => Promise<AccountInfoResponse>;
     };
 
     getUserIP: () => Promise<{ ip: string }>;
+
+    close(): void;
   }
 }

package/index.js

@@ -1,11 +1,12 @@
 import { LRUCache } from "lru-cache";
+import { createSession, TimeoutError as NLcURLTimeout } from "nlcurl";
 import { APIError, AuthenticationError, BadRequestError, NotFoundError, RateLimitError, ServerError } from "./lib/errors.js";
 
 const DEFAULT_BASE_URL = "https://api.chapyapi.com/api/v1";
 const DEFAULT_RETRIES = 2;
 const DEFAULT_TIMEOUT = 30000;
 
-const SDK_VERSION = "0.4.0";
+const SDK_VERSION = "0.6.0";
 
 export class ChapybaraClient {
   constructor(options) {
@@ -18,6 +19,16 @@ export class ChapybaraClient {
     this.retries = options.retries ?? DEFAULT_RETRIES;
     this.timeout = options.timeout || DEFAULT_TIMEOUT;
 
+    this.session = createSession({
+      baseURL: this.baseUrl,
+      headers: {
+        "X-API-Key": this.apiKey,
+        "Content-Type": "application/json",
+        "User-Agent": `Chapybara-NodeJS-SDK/${SDK_VERSION}`,
+      },
+      timeout: this.timeout,
+    });
+
     if (options.cacheOptions) {
       this.cache = new LRUCache(options.cacheOptions);
     }
@@ -34,10 +45,6 @@ export class ChapybaraClient {
       getScanner: (domain) => this._request(`/webtech/${domain}`),
     };
 
-    this.screenshot = {
-      get: (domain) => this._request(`/screenshot/${domain}`),
-    };
-
     this.account = {
       getInfo: () => this._request("/account"),
     };
@@ -46,29 +53,14 @@ export class ChapybaraClient {
   }
 
   async _request(endpoint, attempt = 1) {
-    const url = `${this.baseUrl}${endpoint}`;
     const cacheKey = endpoint;
-    const isScreenshotRequest = endpoint.startsWith("/screenshot");
 
     if (this.cache?.has(cacheKey)) {
       return this.cache.get(cacheKey);
     }
 
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), this.timeout);
-
     try {
-      const response = await fetch(url, {
-        method: "GET",
-        headers: {
-          "X-API-Key": this.apiKey,
-          "Content-Type": "application/json",
-          "User-Agent": `Chapybara-NodeJS-SDK/${SDK_VERSION}`,
-        },
-        signal: controller.signal,
-      });
-
-      clearTimeout(timeoutId);
+      const response = await this.session.get(endpoint);
 
       if (!response.ok) {
         if (response.status >= 500 && attempt <= this.retries) {
@@ -76,16 +68,10 @@ export class ChapybaraClient {
           await new Promise((resolve) => setTimeout(resolve, delay));
           return this._request(endpoint, attempt + 1);
         }
-        await this._handleError(response);
+        this._handleError(response);
       }
 
-      let data;
-      if (isScreenshotRequest) {
-        const arrayBuffer = await response.arrayBuffer();
-        data = Buffer.from(arrayBuffer);
-      } else {
-        data = await response.json();
-      }
+      const data = response.json();
 
       if (this.cache) {
         this.cache.set(cacheKey, data);
@@ -93,18 +79,17 @@ export class ChapybaraClient {
 
       return data;
     } catch (error) {
-      clearTimeout(timeoutId);
-      if (error.name === "AbortError") {
+      if (error instanceof NLcURLTimeout) {
         throw new Error("Request timed out");
       }
       throw error;
     }
   }
 
-  async _handleError(response) {
+  _handleError(response) {
     let errorData;
     try {
-      errorData = await response.json();
+      errorData = response.json();
     } catch (e) {
       errorData = { error: `HTTP ${response.status} Error` };
     }
@@ -127,4 +112,8 @@ export class ChapybaraClient {
         throw new APIError(response.status, errorData);
     }
   }
+
+  close() {
+    this.session.close();
+  }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chapybara",
-  "version": "0.4.0",
+  "version": "0.6.0",
   "description": "Official NodeJS SDK for the Chapybara Domain & IP Intelligence API.",
   "main": "index.js",
   "types": "index.d.ts",
@@ -15,15 +15,15 @@
     "domain",
     "ip",
     "intelligence",
-    "security",
-    "screenshot"
+    "security"
   ],
   "author": "Alpha System",
   "license": "MIT",
   "dependencies": {
-    "lru-cache": "^11.2.2"
+    "lru-cache": "^11.2.6",
+    "nlcurl": "^0.2.0"
   },
   "engines": {
     "node": ">=18.0.0"
   }
-}
+}