feat: create a default branch on the first load of branches

Author: dkrizan

backend/data/src/main/kotlin/io/tolgee/development/testDataBuilder/data/BranchTestData.kt

@@ -1,6 +1,7 @@
 package io.tolgee.development.testDataBuilder.data
 
 import io.tolgee.component.CurrentDateProvider
+import io.tolgee.model.Project
 import io.tolgee.model.branching.Branch
 import io.tolgee.util.addDays
 
@@ -9,34 +10,44 @@ class BranchTestData(
 ) : BaseTestData("branch", "Project with branches") {
   lateinit var mainBranch: Branch
   lateinit var featureBranch: Branch
+  var secondProject: Project
   init {
-    projectBuilder.apply {
-      addBranch {
-        name = "main"
-        project = projectBuilder.self
-        isProtected = true
-        isDefault = true
-        createdAt = currentDateProvider.date
-      }.build {
-        mainBranch = self
+    this.root.apply {
+      projectBuilder.apply {
         addBranch {
-          name = "feature-branch"
+          name = "main"
           project = projectBuilder.self
-          isProtected = false
-          isDefault = false
-          originBranch = this
+          isProtected = true
+          isDefault = true
+          createdAt = currentDateProvider.date
         }.build {
-          featureBranch = self
-        }
-        addBranch {
-          name = "merged-and-deleted-branch"
-          project = projectBuilder.self
-          isProtected = false
-          isDefault = false
-          archivedAt = currentDateProvider.date.addDays(-1)
-          originBranch = this
+          mainBranch = self
+          addBranch {
+            name = "feature-branch"
+            project = projectBuilder.self
+            isProtected = false
+            isDefault = false
+            originBranch = this
+          }.build {
+            featureBranch = self
+          }
+          addBranch {
+            name = "merged-and-deleted-branch"
+            project = projectBuilder.self
+            isProtected = false
+            isDefault = false
+            archivedAt = currentDateProvider.date.addDays(-1)
+            originBranch = this
+          }
         }
       }
+      secondProject = addProject {
+        name = "empty-project"
+      }.build {
+        addKey {
+          name = "test"
+        }
+      }.self
     }
   }
 }

ee/backend/app/src/main/kotlin/io/tolgee/ee/service/branching/BranchServiceImpl.kt

@@ -15,6 +15,7 @@ import jakarta.transaction.Transactional
 import org.springframework.context.ApplicationContext
 import org.springframework.context.annotation.Primary
 import org.springframework.data.domain.Page
+import org.springframework.data.domain.PageImpl
 import org.springframework.data.domain.Pageable
 import org.springframework.stereotype.Service
 
@@ -26,9 +27,15 @@ class BranchServiceImpl(
   private val entityManager: EntityManager,
   private val branchCopyService: BranchCopyService,
   private val applicationContext: ApplicationContext,
+  private val defaultBranchCreator: DefaultBranchCreator,
 ) : BranchService {
   override fun getAllBranches(projectId: Long, page: Pageable, search: String?): Page<Branch> {
-    return branchRepository.getAllProjectBranches(projectId, page, search)
+    val branches = branchRepository.getAllProjectBranches(projectId, page, search)
+    if (branches.isEmpty) {
+      val defaultBranch = defaultBranchCreator.create(projectId)
+      return PageImpl(listOf(defaultBranch))
+    }
+    return branches
   }
 
   override fun getBranch(projectId: Long, branchId: Long): Branch {

ee/backend/app/src/main/kotlin/io/tolgee/ee/service/branching/DefaultBranchCreator.kt

@@ -0,0 +1,30 @@
+package io.tolgee.ee.service.branching
+
+import io.tolgee.model.Project
+import io.tolgee.model.branching.Branch
+import io.tolgee.service.key.KeyService
+import io.tolgee.service.project.ProjectService
+import jakarta.transaction.Transactional
+import org.springframework.stereotype.Service
+
+@Service
+class DefaultBranchCreator(
+  private val keyService: KeyService,
+  private val projectService: ProjectService,
+) {
+
+  @Transactional
+  fun create(projectId: Long): Branch {
+    val project = projectService.get(projectId)
+    return create(project)
+  }
+
+  @Transactional
+  fun create(project: Project): Branch {
+    val branch = Branch.createMainBranch(project)
+    keyService.getAll(project.id).forEach { it ->
+      it.branch = branch
+    }
+    return branch
+  }
+}

ee/backend/tests/src/test/kotlin/io/tolgee/ee/api/v2/controllers/branching/BranchControllerTest.kt

@@ -52,6 +52,28 @@ class BranchControllerTest : ProjectAuthControllerTest("/v2/projects/") {
     }
   }
 
+  @Test
+  @ProjectJWTAuthTestMethod
+  fun `creates default branch on project without branches`() {
+    testData.projectBuilder.self = testData.secondProject
+    performProjectAuthGet("branches").andAssertThatJson {
+      node("page.totalElements").isNumber.isEqualTo(BigDecimal(1))
+      node("_embedded.branches") {
+        node("[0].name").isEqualTo("main")
+        node("[0].active").isEqualTo(true)
+        node("[0].isDefault").isEqualTo(true)
+      }
+    }
+    keyService.getAll(testData.secondProject.id).first().branch!!.id.let { it ->
+      it.assert.isNotNull()
+      branchRepository.findByIdOrNull(it)!!.let { branch ->
+        branch.name.assert.isEqualTo(Branch.DEFAULT_BRANCH_NAME)
+        branch.isDefault.assert.isTrue
+        branch.isProtected.assert.isTrue
+      }
+    }
+  }
+
   @Test()
   @ProjectJWTAuthTestMethod
   fun `creates branch`() {

ee/backend/tests/src/test/kotlin/io/tolgee/ee/api/v2/controllers/branching/BranchCopyIntegrationTest.kt

@@ -8,6 +8,7 @@ import io.tolgee.fixtures.andIsOk
 import io.tolgee.fixtures.waitForNotThrowing
 import io.tolgee.model.key.Key
 import io.tolgee.repository.KeyMetaRepository
+import io.tolgee.repository.KeyRepository
 import io.tolgee.repository.KeyScreenshotReferenceRepository
 import io.tolgee.testing.annotations.ProjectJWTAuthTestMethod
 import io.tolgee.testing.assert
@@ -37,6 +38,9 @@ class BranchCopyIntegrationTest : ProjectAuthControllerTest("/v2/projects/") {
   @Autowired
   lateinit var keyScreenshotReferenceRepository: KeyScreenshotReferenceRepository
 
+  @Autowired
+  lateinit var keyRepository: KeyRepository
+
   @BeforeEach
   fun setup() {
     testData = BranchTranslationsTestData()