feat: support multi domain (#18)

README.md

@@ -106,7 +106,7 @@ node_compat = true
 
 [vars]
 PREFIX = "tmp" # 要处理的邮箱名称前缀
-DOMAIN = "xxx.xxx" # 你的域名
+DOMAINS = ["xxx.xxx1" , "xxx.xxx2"] # 你的域名
 JWT_SECRET = "xxx" # 用于生成 jwt 的密钥
 BLACK_LIST = "" # 黑名单，用于过滤发件人，逗号分隔
 

README_EN.md

@@ -41,7 +41,7 @@ pnpm install
 # copy wrangler.toml.template to wrangler.toml
 # and add your d1 config and these config
 # PREFIX = "tmp" - the email create will be like tmp<xxxxx>@DOMAIN
-# DOMAIN = "xxx.xxx" - you domain name
+# DOMAINS = ["xxx.xxx1" , "xxx.xxx2"] you domain name
 # JWT_SECRET = "xxx"
 # BLACK_LIST = ""
 cp wrangler.toml.template wrangler.toml

frontend/src/Content.vue

@@ -1,6 +1,6 @@
 <script setup>
 import { NSpace, NAlert, NSwitch, NCard, NInput, NInputGroupLabel } from 'naive-ui'
-import { NSpin, NButton, NLayout, NInputGroup, NModal } from 'naive-ui'
+import { NSpin, NButton, NLayout, NInputGroup, NModal, NSelect } from 'naive-ui'
 import { NList, NListItem, NThing, NTag, NNumberAnimation } from 'naive-ui'
 import { watch, onMounted, ref } from "vue";
 import { useStorage } from '@vueuse/core'
@@ -21,9 +21,13 @@ const timer = ref(null)
 const showPassword = ref(false)
 const showNewEmail = ref(false)
 const emailName = ref("")
+const emailDomain = ref("")
 const openSettings = ref({
   prefix: 'test',
-  domain: 'test.com'
+  domains: [{
+    label: 'test.com',
+    value: 'test.com'
+  }]
 })
 
 const { t, locale } = useI18n({
@@ -119,9 +123,8 @@ const newEmail = async () => {
   try {
     loading.value = true;
     let url = `${API_BASE}/api/new_address`;
-    if (emailName.value) {
-      url = `${url}?name=${emailName.value}`;
-    }
+    url = `${url}?name=${emailName.value || ''}`;
+    url = `${url}&domain=${emailDomain.value || ''}`;
     const response = await fetch(url, {
       method: "GET",
       headers: {
@@ -159,7 +162,16 @@ const getOpenSettings = async (jwt) => {
     return;
   }
   let res = await response.json();
-  openSettings.value = res;
+  openSettings.value = {
+    prefix: res["prefix"] || "",
+    domains: res["domains"].map((domain) => {
+      return {
+        label: domain,
+        value: domain
+      }
+    })
+  };
+  emailDomain.value = openSettings.value.domains[0].value;
 }
 
 const getSettings = async (jwt) => {
@@ -266,9 +278,8 @@ onMounted(async () => {
           {{ openSettings.prefix }}
         </n-input-group-label>
         <n-input v-model:value="emailName" />
-        <n-input-group-label>
-          @{{ openSettings.domain }}
-        </n-input-group-label>
+        <n-input-group-label>@</n-input-group-label>
+        <n-select v-model:value="emailDomain" :consistent-menu-width="false" :options="openSettings.domains" />
       </n-input-group>
       <template #action>
         <n-button @click="showNewEmail = false">

worker/src/router.js

@@ -21,20 +21,26 @@ api.get('/api/settings', async (c) => {
 api.get('/open_api/settings', async (c) => {
     return c.json({
         "prefix": c.env.PREFIX,
-        "domain": c.env.DOMAIN,
+        "domains": c.env.DOMAINS,
     });
 })
 
 api.get('/api/new_address', async (c) => {
-    // insert new address
-    let { name } = await c.req.query();
+    let { name, domain } = await c.req.query();
+    // if no name, generate random name
     if (!name) {
-        name = Math.random().toString(36).substring(2, 15)
+        name = Math.random().toString(36).substring(2, 15);
     }
+    // check domain, generate random domain
+    if (!domain || !c.env.DOMAINS.includes(domain)) {
+        domain = c.env.DOMAINS[Math.floor(Math.random() * c.env.DOMAINS.length)];
+    }
+    // create address
+    const emailAddress = c.env.PREFIX + name + "@" + domain
     try {
         const { success } = await c.env.DB.prepare(
             `INSERT INTO address (name) VALUES (?)`
-        ).bind(name).run();
+        ).bind(name + "@" + domain).run();
         if (!success) {
             return c.text("Failed to create address", 500)
         }
@@ -46,7 +52,7 @@ api.get('/api/new_address', async (c) => {
     }
     // create jwt
     const jwt = await Jwt.sign({
-        address: c.env.PREFIX + name + "@" + c.env.DOMAIN
+        address: emailAddress
     }, c.env.JWT_SECRET)
     return c.json({
         jwt: jwt

worker/wrangler.toml.template

@@ -5,7 +5,7 @@ node_compat = true
 
 [vars]
 PREFIX = "tmp"
-DOMAIN = "xxx.xxx"
+DOMAINS = ["xxx.xxx1" , "xxx.xxx2"]
 JWT_SECRET = "xxx"
 BLACK_LIST = ""