feat(bitbucket): integrate Bitbucket support with repository search and team management

Ayyanaruto · Ayyanaruto · commit 26f4adc97355 · 2025-09-06T18:48:56.000+05:30
diff --git a/backend/analytics_server/mhq/api/request_utils.py b/backend/analytics_server/mhq/api/request_utils.py
@@ -7,6 +7,7 @@
 from stringcase import snakecase
 from voluptuous import Invalid
 from werkzeug.exceptions import BadRequest
+from mhq.utils.log import LOG
 from mhq.store.models.code.repository import TeamRepos
 from mhq.service.code.models.org_repo import RawTeamOrgRepo
 from mhq.store.models.code import WorkflowFilter, CodeProvider
@@ -82,20 +83,24 @@ def coerce_workflow_filter(filter_data: str) -> WorkflowFilter:
 
 
 def coerce_org_repo(repo: Dict[str, str]) -> RawTeamOrgRepo:
-    return RawTeamOrgRepo(
-        team_id=repo.get("team_id"),
-        provider=CodeProvider(repo.get("provider")),
-        name=repo.get("name"),
-        org_name=repo.get("org"),
-        slug=repo.get("slug"),
-        idempotency_key=repo.get("idempotency_key"),
-        default_branch=repo.get("default_branch"),
-        deployment_type=(
-            TeamReposDeploymentType(repo.get("deployment_type"))
-            if repo.get("deployment_type")
-            else TeamReposDeploymentType.PR_MERGE
-        ),
-    )
+    try:
+        return RawTeamOrgRepo(
+            team_id=repo.get("team_id"),
+            provider=CodeProvider(repo.get("provider")),
+            name=repo.get("name"),
+            org_name=repo.get("org"),
+            slug=repo.get("slug"),
+            idempotency_key=repo.get("idempotency_key"),
+            default_branch=repo.get("default_branch"),
+            deployment_type=(
+                TeamReposDeploymentType(repo.get("deployment_type"))
+                if repo.get("deployment_type")
+                else TeamReposDeploymentType.PR_MERGE
+            ),
+        )
+    except Exception as e:
+        LOG.error(f"Error creating RawTeamOrgRepo with data: {repo}. Error: {str(e)}")
+        raise
 
 
 def coerce_org_repos(repos: List[Dict[str, str]]) -> List[RawTeamOrgRepo]:
diff --git a/backend/analytics_server/mhq/store/models/code/enums.py b/backend/analytics_server/mhq/store/models/code/enums.py
@@ -4,6 +4,7 @@
 class CodeProvider(Enum):
     GITHUB = "github"
     GITLAB = "gitlab"
+    BITBUCKET = "bitbucket"
 
 
 class CodeBookmarkType(Enum):
diff --git a/web-server/pages/api/internal/[org_id]/git_provider_org.ts b/web-server/pages/api/internal/[org_id]/git_provider_org.ts
@@ -1,6 +1,12 @@
 import * as yup from 'yup';
 
-import { gitlabSearch, searchGithubRepos, getGithubToken, getGitlabToken  } from '@/api/internal/[org_id]/utils';
+import {
+  gitlabSearch,
+  searchGithubRepos,
+  bitbucketSearch,
+  getGithubToken,
+  getGitlabToken
+} from '@/api/internal/[org_id]/utils';
 import { Endpoint } from '@/api-helpers/global';
 import { Integration } from '@/constants/integrations';
 
@@ -42,6 +48,42 @@ endpoint.handle.GET(getSchema, async (req, res) => {
 
 export default endpoint.serve();
 
+const getGithubToken = async (org_id: ID) => {
+  return await db('Integration')
+    .select()
+    .where({
+      org_id,
+      name: Integration.GITHUB
+    })
+    .returning('*')
+    .then(getFirstRow)
+    .then((r) => dec(r.access_token_enc_chunks));
+};
+
+const getGitlabToken = async (org_id: ID) => {
+  return await db('Integration')
+    .select()
+    .where({
+      org_id,
+      name: Integration.GITLAB
+    })
+    .returning('*')
+    .then(getFirstRow)
+    .then((r) => dec(r.access_token_enc_chunks));
+};
+
+const getBitbucketToken = async (org_id: ID) => {
+  return await db('Integration')
+    .select()
+    .where({
+      org_id,
+      name: Integration.BITBUCKET
+    })
+    .returning('*')
+    .then(getFirstRow)
+    .then((r) => dec(r.access_token_enc_chunks));
+};
+
 const fetchMap = [
   {
     provider: Integration.GITHUB,
@@ -52,5 +94,10 @@ const fetchMap = [
     provider: Integration.GITLAB,
     search: gitlabSearch,
     getToken: getGitlabToken
+  },
+  {
+    provider: Integration.BITBUCKET,
+    search: bitbucketSearch,
+    getToken: getBitbucketToken
   }
 ];
diff --git a/web-server/pages/api/internal/[org_id]/utils.ts b/web-server/pages/api/internal/[org_id]/utils.ts
@@ -3,10 +3,10 @@ import { head } from 'ramda';
 
 import { Row } from '@/constants/db';
 import { Integration } from '@/constants/integrations';
-import { BaseRepo } from '@/types/resources';
-import { db, getFirstRow } from '@/utils/db';
 import { DEFAULT_GH_URL } from '@/constants/urls';
+import { BaseRepo } from '@/types/resources';
 import { dec } from '@/utils/auth-supplementary';
+import { db, getFirstRow } from '@/utils/db';
 
 type GithubRepo = {
   name: string;
@@ -271,6 +271,134 @@ export const gitlabSearch = async (pat: string, searchString: string) => {
   return searchGitlabRepos(pat, search);
 };
 
+// Bitbucket functions
+
+type BitbucketRepo = {
+  uuid: string;
+  name: string;
+  full_name: string;
+  description?: string;
+  language?: string;
+  mainbranch?: {
+    name: string;
+  };
+  links: {
+    html: {
+      href: string;
+    };
+  };
+  owner: {
+    username: string;
+  };
+};
+
+type BitbucketResponse = {
+  values: BitbucketRepo[];
+  next?: string;
+};
+
+const BITBUCKET_API_URL = 'https://api.bitbucket.org/2.0';
+
+export const searchBitbucketRepos = async (
+  credentials: string,
+  searchString: string
+): Promise<BaseRepo[]> => {
+  let urlString = convertUrlToQuery(searchString);
+  if (urlString !== searchString && urlString.includes('/')) {
+    try {
+      return await searchBitbucketRepoWithURL(credentials, urlString);
+    } catch (e) {
+      return await searchBitbucketReposWithNames(credentials, urlString);
+    }
+  }
+  return await searchBitbucketReposWithNames(credentials, urlString);
+};
+
+const searchBitbucketRepoWithURL = async (
+  credentials: string,
+  searchString: string
+): Promise<BaseRepo[]> => {
+  const apiUrl = `${BITBUCKET_API_URL}/repositories/${searchString}`;
+
+  const response = await fetch(apiUrl, {
+    method: 'GET',
+    headers: {
+      Authorization: `Basic ${credentials}`,
+      'Content-Type': 'application/json'
+    }
+  });
+
+  if (!response.ok) {
+    throw new Error(`Bitbucket API error: ${response.statusText}`);
+  }
+
+  const repo = (await response.json()) as BitbucketRepo;
+
+  return [
+    {
+      id: repo.uuid.replace(/[{}]/g, ''),
+      name: repo.name,
+      desc: repo.description,
+      slug: repo.name,
+      parent: repo.owner.username,
+      web_url: repo.links.html.href,
+      branch: repo.mainbranch?.name,
+      language: repo.language,
+      provider: Integration.BITBUCKET
+    }
+  ] as BaseRepo[];
+};
+
+const searchBitbucketReposWithNames = async (
+  credentials: string,
+  searchString: string
+): Promise<BaseRepo[]> => {
+  const apiUrl = `${BITBUCKET_API_URL}/repositories`;
+  const params = new URLSearchParams({
+    q: `name~"${searchString}"`,
+    role: 'member',
+    pagelen: '50'
+  });
+
+  const response = await fetch(`${apiUrl}?${params}`, {
+    method: 'GET',
+    headers: {
+      Authorization: `Basic ${credentials}`,
+      'Content-Type': 'application/json'
+    }
+  });
+
+  if (!response.ok) {
+    throw new Error(`Bitbucket API error: ${response.statusText}`);
+  }
+
+  const responseBody = (await response.json()) as BitbucketResponse;
+  const repositories = responseBody.values || [];
+
+  return repositories.map(
+    (repo) =>
+      ({
+        id: repo.uuid.replace(/[{}]/g, ''),
+        name: repo.name,
+        desc: repo.description,
+        slug: repo.name,
+        parent: repo.owner.username,
+        web_url: repo.links.html.href,
+        branch: repo.mainbranch?.name,
+        language: repo.language || null,
+        provider: Integration.BITBUCKET
+      }) as BaseRepo
+  );
+};
+
+export const bitbucketSearch = async (
+  credentials: string,
+  searchString: string
+): Promise<BaseRepo[]> => {
+  let search = convertUrlToQuery(searchString);
+  return searchBitbucketRepos(credentials, search);
+};
+
 const convertUrlToQuery = (url: string) => {
   let query = url;
   try {
@@ -282,6 +410,7 @@ const convertUrlToQuery = (url: string) => {
     query = query.replace('http://', '');
     query = query.replace('github.com/', '');
     query = query.replace('gitlab.com/', '');
+    query = query.replace('bitbucket.org/', '');
     query = query.startsWith('www.') ? query.slice(4) : query;
     query = query.endsWith('/') ? query.slice(0, -1) : query;
   }
@@ -315,26 +444,27 @@ export const getGitHubCustomDomain = async (): Promise<string | null> => {
 
     return head(provider_meta || [])?.custom_domain || null;
   } catch (error) {
-    console.error('Error occured while getting custom domain from database:', error);
+    console.error(
+      'Error occured while getting custom domain from database:',
+      error
+    );
     return null;
   }
 };
 
-const normalizeSlashes = (url: string) =>
-  url.replace(/(?<!:)\/{2,}/g, '/');
+const normalizeSlashes = (url: string) => url.replace(/(?<!:)\/{2,}/g, '/');
 
 export const getGitHubRestApiUrl = async (path: string) => {
   const customDomain = await getGitHubCustomDomain();
-  const base = customDomain
-    ? `${customDomain}/api/v3`
-    : DEFAULT_GH_URL;
+  const base = customDomain ? `${customDomain}/api/v3` : DEFAULT_GH_URL;
   return normalizeSlashes(`${base}/${path}`);
 };
 
-
 export const getGitHubGraphQLUrl = async (): Promise<string> => {
   const customDomain = await getGitHubCustomDomain();
-  return customDomain ? `${customDomain}/api/graphql` : `${DEFAULT_GH_URL}/graphql`;
+  return customDomain
+    ? `${customDomain}/api/graphql`
+    : `${DEFAULT_GH_URL}/graphql`;
 };
 
 export const getGithubToken = async (org_id: ID) => {
diff --git a/web-server/pages/api/resources/orgs/[org_id]/teams/v2.ts b/web-server/pages/api/resources/orgs/[org_id]/teams/v2.ts
@@ -91,7 +91,7 @@ endpoint.handle.GET(getSchema, async (req, res) => {
     org_id,
     providers?.length
       ? (providers as Integration[])
-      : [Integration.GITHUB, Integration.GITLAB]
+      : [Integration.GITHUB, Integration.GITLAB,Integration.BITBUCKET]
   );
 
   res.send({
@@ -116,6 +116,7 @@ endpoint.handle.POST(postSchema, async (req, res) => {
       } as any as ReqRepoWithProvider);
     });
   }, org_repos);
+  console.log('orgReposList in POST', orgReposList);
   const [team, onboardingState] = await Promise.all([
     createTeam(org_id, name, []),
     getOnBoardingState(org_id)
@@ -165,7 +166,7 @@ endpoint.handle.PATCH(patchSchema, async (req, res) => {
       } as any as ReqRepoWithProvider);
     });
   }, org_repos);
-
+console.log('orgReposList in Patch', orgReposList);
   const [team] = await Promise.all([
     updateTeam(id, name, []),
     handleRequest<(Row<'TeamRepos'> & Row<'OrgRepo'>)[]>(`/teams/${id}/repos`, {
@@ -301,7 +302,7 @@ const updateReposWorkflows = async (
       .whereIn('name', reposForWorkflows)
       .where('org_id', org_id)
       .andWhere('is_active', true)
-      .and.whereIn('provider', [Integration.GITHUB, Integration.GITLAB]);
+      .and.whereIn('provider', [Integration.GITHUB, Integration.GITLAB, Integration.BITBUCKET]);
 
     const groupedRepos = groupBy(dbReposForWorkflows, 'name');
 
diff --git a/web-server/src/components/Teams/CreateTeams.tsx b/web-server/src/components/Teams/CreateTeams.tsx
@@ -34,6 +34,7 @@ import { DeploymentWorkflowSelector } from '@/components/WorkflowSelector';
 import { Integration } from '@/constants/integrations';
 import { useModal } from '@/contexts/ModalContext';
 import { useBoolState, useEasyState } from '@/hooks/useEasyState';
+import BitbucketIcon from '@/mocks/icons/bitbucket.svg';
 import GitlabIcon from '@/mocks/icons/gitlab.svg';
 import { BaseRepo, DeploymentSources } from '@/types/resources';
 import { trimWithEllipsis } from '@/utils/stringFormatting';
@@ -311,6 +312,8 @@ const TeamRepos: FC = () => {
                   <FlexBox gap={1 / 2} alignCenter>
                     {option.provider === Integration.GITHUB ? (
                       <GitHub sx={{ fontSize: '14px' }} />
+                    ) : option.provider === Integration.BITBUCKET ? (
+                      <BitbucketIcon height={12} width={12} />
                     ) : (
                       <GitlabIcon height={12} width={12} />
                     )}
@@ -447,6 +450,8 @@ const DisplayRepos: FC = () => {
                     >
                       {repo.provider === Integration.GITHUB ? (
                         <GitHub />
+                      ) : repo.provider === Integration.BITBUCKET ? (
+                        <BitbucketIcon height={14} width={14} />
                       ) : (
                         <GitlabIcon height={14} width={14} />
                       )}
