umami/pages/api/website.js

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;
        console.log('exising id', 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);
};