update index.ts to include custom domain support

Author: AlexanderEpolite

src/index.ts

@@ -11,10 +11,29 @@ const BASE_URL = "https://api.tempmail.lol";
  * Create a new Inbox.
  * @param cb {function} Callback function.
  * @param rush {boolean} (optional) Enable Rush Mode (see https://tempmail.lol/news/2022/08/03/introducing-rush-mode-for-tempmail/).
+ * @param domain {string} (optional) use a specific domain from the website.
+ * 
+ * @throws {Error} if the custom domain does not exist or `rush` and `domain` are present.
  */
-function createInbox(cb: (inbox: Inbox | undefined, err: Error | null) => any, rush = false): void {
-    fetch(`${BASE_URL}/generate${rush ? "/rush" : ""}`).then(res => res.json()).then((json) => {
+function createInbox(cb: (inbox: Inbox | undefined, err: Error | null) => any, rush = false, domain: string | undefined = undefined): void {
+    
+    if(rush && domain !== undefined) {
+        throw new Error("You cannot specify both a custom domain and rush mode.");
+    }
+    
+    let url = `${BASE_URL}/generate`;
+    
+    if(rush) {
+        url += "/rush";
+    } else if(domain !== undefined) {
+        url += `/${domain}`;
+    }
+    
+    fetch(url).then(res => res.json()).then((json) => {
         const inbox = new Inbox(json.address, json.token);
+        if(json.error) {
+            cb(undefined, new Error(json.error));
+        }
         cb(inbox, null);
     }).catch((err) => {
         cb(undefined, err);
@@ -25,11 +44,28 @@ function createInbox(cb: (inbox: Inbox | undefined, err: Error | null) => any, r
  * Create a new Inbox asynchronously.
  * 
  * @param rush {boolean} (optional) Enable Rush Mode (see https://tempmail.lol/news/2022/08/03/introducing-rush-mode-for-tempmail/).
+ * @param domain {string} (optional) use a specific domain from the website.
  * @returns {Promise<Inbox>} Promise with the Inbox.
+ * @throws {Error} if the custom domain does not exist or `rush` and `domain` are present.
  */
-async function createInboxAsync(rush: boolean = false): Promise<Inbox> {
-    const res = await fetch(`${BASE_URL}/generate${rush ? "/rush" : ""}`);
+async function createInboxAsync(rush: boolean = false, domain: string | undefined = undefined): Promise<Inbox> {
+    if(rush && domain !== undefined) {
+        throw new Error("You cannot specify both a custom domain and rush mode.");
+    }
+    
+    let url = `${BASE_URL}/generate`;
+    
+    if(rush) {
+        url += "/rush";
+    } else if(domain !== undefined) {
+        url += `/${domain}`;
+    }
+    
+    const res = await fetch(url);
     const json = await res.json();
+    
+    if(json.error) throw new Error(json.error);
+    
     return new Inbox(json.address, json.token);
 }
 
@@ -82,10 +118,43 @@ async function checkInboxAsync(inbox: Inbox | string): Promise<Email[]> {
     return json.email.map((email: Email) => new Email(email.from, email.to, email.subject, email.body, email.html, email.date, email.ip));
 }
 
+/**
+ * Check a custom inbox.
+ * 
+ * NOTE: this method will not return anything indicating if the token is invalid.
+ * 
+ * @param domain {string} Domain to check.
+ * @param key {string} The key for the domain generated by the website.
+ * @param cb {function} Callback function.
+ */
+function checkCustomInbox(domain: string, key: string, cb: (emails: Email[]) => any): void {
+    fetch(`${BASE_URL}/custom/${key}/${domain}`).then(res => res.json()).then(json => {
+        const emails = json.email.map((email: Email) => new Email(email.from, email.to, email.subject, email.body, email.html, email.date, email.ip));
+        cb(emails);
+    });
+}
+
+/**
+ * Check a custom inbox asynchronously.
+ * 
+ * NOTE: this method will not return anything indicating if the token is invalid.
+ * 
+ * @param domain {string} Domain to check.
+ * @param key {string} The key for the domain generated by the website.
+ * 
+ * @returns {Promise<Email[]>} Promise with the emails.
+ */
+async function checkCustomInboxAsync(domain: string, key: string): Promise<Email[]> {
+    const res = await fetch(`${BASE_URL}/custom/${key}/${domain}`);
+    const json = await res.json();
+    return json.email.map((email: Email) => new Email(email.from, email.to, email.subject, email.body, email.html, email.date, email.ip));
+}
+
 export {
     Inbox,
     Email,
 
     createInbox, createInboxAsync,
     checkInbox, checkInboxAsync,
+    checkCustomInbox, checkCustomInboxAsync
 };