import { updateWebsite, createWebsite, getWebsiteById } from 'lib/queries'; import { useAuth } from 'lib/middleware'; import { uuid, getRandomChars } from 'lib/crypto'; import { ok, unauthorized, methodNotAllowed } from 'lib/response'; export default async (req, res) => { await useAuth(req, res); const { user_id, is_admin } = req.auth; const { website_id, enable_share_url } = req.body; if (req.method === 'POST') { const { name, domain } = req.body; if (website_id) { const website = await getWebsiteById(website_id); if (website.user_id === user_id || is_admin) { let { share_id } = website; if (enable_share_url) { share_id = share_id ? share_id : getRandomChars(8); } else { share_id = null; } await updateWebsite(website_id, { name, domain, share_id }); return ok(res); } return unauthorized(res); } else { const website_uuid = uuid(); const share_id = enable_share_url ? getRandomChars(8) : null; const website = await createWebsite(user_id, { website_uuid, name, domain, share_id }); return ok(res, website); } } return methodNotAllowed(res); };