46 lines
2.2 KiB
Python
46 lines
2.2 KiB
Python
from chalicelib.core import integration_github, integration_jira_cloud
|
|
from chalicelib.utils import pg_client
|
|
|
|
SUPPORTED_TOOLS = [integration_github.PROVIDER, integration_jira_cloud.PROVIDER]
|
|
|
|
|
|
async def get_available_integrations(user_id):
|
|
async with pg_client.cursor() as cur:
|
|
await cur.execute(
|
|
cur.mogrify(f"""\
|
|
SELECT EXISTS((SELECT 1
|
|
FROM public.oauth_authentication
|
|
WHERE user_id = %(user_id)s
|
|
AND provider = 'github')) AS github,
|
|
EXISTS((SELECT 1
|
|
FROM public.jira_cloud
|
|
WHERE user_id = %(user_id)s)) AS jira;""",
|
|
{"user_id": user_id})
|
|
)
|
|
current_integrations = await cur.fetchone()
|
|
return dict(current_integrations)
|
|
|
|
|
|
async def __get_default_integration(user_id):
|
|
current_integrations = await get_available_integrations(user_id)
|
|
return integration_github.PROVIDER if current_integrations["github"] else integration_jira_cloud.PROVIDER if \
|
|
current_integrations["jira"] else None
|
|
|
|
|
|
async def get_integration(tenant_id, user_id, tool=None, for_delete=False):
|
|
if tool is None:
|
|
tool = await __get_default_integration(user_id=user_id)
|
|
if tool is None:
|
|
return {"errors": [f"no issue tracking tool found"]}, None
|
|
tool = tool.upper()
|
|
if tool not in SUPPORTED_TOOLS:
|
|
return {"errors": [f"issue tracking tool not supported yet, available: {SUPPORTED_TOOLS}"]}, None
|
|
if tool == integration_jira_cloud.PROVIDER:
|
|
integration = integration_jira_cloud.JIRAIntegration(tenant_id=tenant_id, user_id=user_id)
|
|
await integration.init()
|
|
if not for_delete and integration.integration is not None and not integration.integration.get("valid", True):
|
|
return {"errors": ["JIRA: connexion issue/unauthorized"]}, integration
|
|
return None, integration
|
|
elif tool == integration_github.PROVIDER:
|
|
return None, integration_github.GitHubIntegration(tenant_id=tenant_id, user_id=user_id)
|
|
return {"errors": ["lost integration"]}, None
|