can work

Author: adela-bytebase

sql-editor/src/app/api/create-user/route.ts

@@ -1,4 +1,4 @@
-import { createBBIssueWorkflow, generateBBToken, grantUserRole } from "../utils";
+import { createBBIssueWorkflow, generateBBToken, grantUserRoleProjectOwner, switchWorkspaceMode } from "../utils";
 
 function generateRandomString(length: number): string {
   const characters = 'abcdefghijklmnopqrstuvwxyz';
@@ -70,20 +70,24 @@ export async function GET(request: Request) {
 
     console.log('User creation process completed successfully');
 
-    //console.log('now create db');
-    //const result = await createBBIssueWorkflow(username)
-    //console.log("after creating db", result)
+    console.log('now create db');
+    const createdDB = await createBBIssueWorkflow(username)
+    console.log("after creating db", createdDB)
 
 
     // Grant the user project querier role
-    const result = await grantUserRole(username);
-    console.log("after grantUserRole", result)
+    const grantresult = await grantUserRoleProjectOwner(username);
+    console.log("after grantUserRole", grantresult)
 
+    // Switch the workspace to Editor mode
+    const switchresult = await switchWorkspaceMode();
+    console.log("after switchWorkspaceMode", switchresult)
     // Return the credentials and created project
     return new Response(JSON.stringify({
       message: 'User and project created successfully',
       credentials: { username, email, password },
-      project: createdProject
+      project: createdProject,
+      database: createdDB
     }), {
       status: 200,
       headers: { 'Content-Type': 'application/json' },

sql-editor/src/app/api/utils.ts

@@ -49,7 +49,22 @@ export async function generateBBToken() {
     return token.token;
 }
 
-export async function grantUserRole(project: string) {
+export async function switchWorkspaceMode() {
+    const token = await generateBBToken();
+    const response = await fetchData(`${process.env.NEXT_PUBLIC_BB_HOST}/v1/settings/bb.workspace.profile?updateMask=value.workspace_profile_setting_value.database_change_mode`, token, {
+        method: 'PATCH',
+        body: JSON.stringify({
+            "value": { 
+                "workspaceProfileSettingValue": {
+                    "databaseChangeMode": "EDITOR"
+                }
+            }
+        })
+    });
+    return response;
+}
+
+export async function grantUserRoleProjectOwner(project: string) {
     const username = project;
     const token = await generateBBToken();
     const response = await fetchData(`${process.env.NEXT_PUBLIC_BB_HOST}/v1/projects/${project}:getIamPolicy`, token, {
@@ -59,7 +74,7 @@ export async function grantUserRole(project: string) {
     console.log("=============getIamPolicy", JSON.stringify(response));
 
     response.bindings.push({
-        "role": "roles/projectQuerier",
+        "role": "roles/projectOwner",
         "members": [`user:${username}@example.com`],
         "condition": {
           "expression": "",

sql-editor/src/app/page.tsx

@@ -49,7 +49,9 @@ export default function Home() {
 
   return (
     <main className="flex min-h-screen flex-col items-center justify-center p-24">
-      <h1 className="text-4xl font-bold mb-8">Welcome to Bytebase Demo</h1>
+      <h1 className="text-4xl font-bold mb-8"> Embedded Bytebase SQL Editor Demo</h1>
+      {!userData && (
+      <h2 className="text-2xl font-bold mb-4">Logout your Bytebase account in the same browser before click Start</h2>)}
       {!userData && (
         <button
           onClick={handleCreateUser}
@@ -62,13 +64,14 @@ export default function Home() {
       {error && <p className="text-red-500 mt-4">{error}</p>}
       {userData && (
         <div className="mt-8 w-full">
-          <h2 className="text-2xl font-bold mb-4">User and Project Created Successfully</h2>
+          <h2 className="text-2xl font-bold mb-4">User/Project/Database Created Successfully, click Sign in to start</h2>
           <div className="bg-gray-100 p-6 rounded-lg mb-4">
             <p><strong>Username:</strong> {userData.credentials.username}</p>
             <p><strong>Email:</strong> {userData.credentials.email}</p>
             <p><strong>Password:</strong> {userData.credentials.password}</p>
             <p><strong>Project Title:</strong> {userData.project.title}</p>
             <p><strong>Project Key:</strong> {userData.project.key}</p>
+            <p><strong>Database:</strong> {userData.project.key}</p>
           </div>
           <p className="mt-4 mb-4 text-sm text-gray-600">Please save these credentials and project details for future use.</p>
           <div className="w-full h-screen border border-gray-300 rounded-lg overflow-hidden">