feat: add is_public property to projects

Author: haydnli-shopify

src/dstack/_internal/core/models/projects.py

@@ -25,3 +25,4 @@ class Project(CoreModel):
     created_at: Optional[datetime] = None
     backends: List[BackendInfo]
     members: List[Member]
+    is_public: bool = False

src/dstack/_internal/server/migrations/versions/35f732ee4cf5_add_projectmodel_is_public.py

@@ -0,0 +1,39 @@
+"""Add ProjectModel.is_public
+
+Revision ID: 35f732ee4cf5
+Revises: bca2fdf130bf
+Create Date: 2025-06-06 13:04:02.912032
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "35f732ee4cf5"
+down_revision = "bca2fdf130bf"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    # Add is_public column as nullable first
+    with op.batch_alter_table("projects", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("is_public", sa.Boolean(), nullable=True))
+
+    # Set is_public to False for existing projects
+    op.execute(sa.sql.text("UPDATE projects SET is_public = FALSE"))
+
+    # Make is_public non-nullable with default value
+    with op.batch_alter_table("projects", schema=None) as batch_op:
+        batch_op.alter_column("is_public", nullable=False, server_default=sa.false())
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    # Remove is_public column
+    with op.batch_alter_table("projects", schema=None) as batch_op:
+        batch_op.drop_column("is_public")
+    # ### end Alembic commands ###

src/dstack/_internal/server/models.py

@@ -202,6 +202,7 @@ class ProjectModel(BaseModel):
     name: Mapped[str] = mapped_column(String(50), unique=True)
     created_at: Mapped[datetime] = mapped_column(NaiveDateTime, default=get_current_datetime)
     deleted: Mapped[bool] = mapped_column(Boolean, default=False)
+    is_public: Mapped[bool] = mapped_column(Boolean, default=False)
 
     owner_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"))
     owner: Mapped[UserModel] = relationship(lazy="joined")

src/dstack/_internal/server/schemas/projects.py

@@ -1,25 +1,28 @@
-from typing import Annotated, List
+from typing import Annotated, List, Optional
 
 from pydantic import Field
 
-from dstack._internal.core.models.common import CoreModel
+from dstack._internal.core.models.base import CoreModel
+from dstack._internal.core.models.projects import Project
 from dstack._internal.core.models.users import ProjectRole
 
 
 class CreateProjectRequest(CoreModel):
     project_name: str
+    is_public: Optional[bool] = False
 
 
 class DeleteProjectsRequest(CoreModel):
     projects_names: List[str]
 
 
+class GetProjectRequest(CoreModel):
+    project_name: str
+
+
 class MemberSetting(CoreModel):
-    username: Annotated[
-        str,
-        Field(description="The username or email of the user"),
-    ]
-    project_role: ProjectRole
+    username: str
+    project_role: Annotated[ProjectRole, Field(discriminator="project_role")]
 
 
 class SetProjectMembersRequest(CoreModel):