Add gitlab cog, needs more work, refs #30
This commit is contained in:
parent
af0bd71d89
commit
189e15b1dd
3 changed files with 157 additions and 1 deletions
|
|
@ -7,7 +7,8 @@
|
|||
password: pass
|
||||
host: localhost
|
||||
port: 27017
|
||||
api_urls:
|
||||
urls:
|
||||
url_name: url
|
||||
url_name2: url2
|
||||
max_messages: 1000
|
||||
gitlab_token: null
|
||||
|
|
|
|||
153
jarvis/cogs/gitlab.py
Normal file
153
jarvis/cogs/gitlab.py
Normal file
|
|
@ -0,0 +1,153 @@
|
|||
import re
|
||||
from datetime import datetime
|
||||
|
||||
import gitlab
|
||||
from discord.ext import commands
|
||||
from discord_slash import SlashContext, cog_ext
|
||||
from discord_slash.utils.manage_commands import create_option
|
||||
|
||||
from jarvis.config import get_config
|
||||
from jarvis.utils import build_embed
|
||||
from jarvis.utils.field import Field
|
||||
|
||||
guild_ids = [862402786116763668]
|
||||
|
||||
|
||||
class GitlabCog(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
config = get_config()
|
||||
self._gitlab = gitlab.Gitlab(
|
||||
"https://git.zevaryx.com", private_token=config.gitlab_token
|
||||
)
|
||||
# J.A.R.V.I.S. GitLab ID is 29
|
||||
self.project = self._gitlab.projects.get(29)
|
||||
|
||||
@cog_ext.cog_subcommand(
|
||||
base="gl",
|
||||
name="issue",
|
||||
description="Get an issue from GitLab",
|
||||
guild_ids=guild_ids,
|
||||
options=[
|
||||
create_option(
|
||||
name="id", description="Issue ID", option_type=4, required=True
|
||||
)
|
||||
],
|
||||
)
|
||||
async def _issue(self, ctx: SlashContext, id: int):
|
||||
try:
|
||||
issue = self.project.issues.get(int(id))
|
||||
except gitlab.exceptions.GitlabGetError:
|
||||
await ctx.send("Issue does not exist.", hidden=True)
|
||||
return
|
||||
assignee = issue.assignee
|
||||
if assignee:
|
||||
assignee = assignee["name"]
|
||||
|
||||
created_at = datetime.strptime(
|
||||
issue.created_at, "%Y-%m-%dT%H:%M:%S.%fZ"
|
||||
).strftime("%Y-%m-%d %H:%M:%S UTC")
|
||||
fields = [
|
||||
Field(
|
||||
name="State", value=issue.state[0].upper() + issue.state[1:]
|
||||
),
|
||||
Field(name="Assignee", value=assignee),
|
||||
Field(name="Labels", value="\n".join(issue.labels)),
|
||||
]
|
||||
color = self.project.labels.get(issue.labels[0]).color
|
||||
fields.append(Field(name="Created At", value=created_at))
|
||||
if issue.state == "closed":
|
||||
closed_at = datetime.strptime(
|
||||
issue.closed_at, "%Y-%m-%dT%H:%M:%S.%fZ"
|
||||
).strftime("%Y-%m-%d %H:%M:%S UTC")
|
||||
fields.append(Field(name="Closed At", value=closed_at))
|
||||
if issue.milestone:
|
||||
fields.append(
|
||||
Field(
|
||||
name="Milestone",
|
||||
value=f"[{issue.milestone['title']}]"
|
||||
+ f"({issue.milestone['web_url']})",
|
||||
inline=False,
|
||||
)
|
||||
)
|
||||
if len(issue.title) > 200:
|
||||
issue.title = issue.title[:200] + "..."
|
||||
embed = build_embed(
|
||||
title=f"[#{issue.iid}] {issue.title}",
|
||||
description=issue.description,
|
||||
fields=fields,
|
||||
color=color,
|
||||
url=issue.web_url,
|
||||
)
|
||||
embed.set_author(
|
||||
name=issue.author["name"],
|
||||
icon_url=issue.author["avatar_url"],
|
||||
url=issue.author["web_url"],
|
||||
)
|
||||
embed.set_thumbnail(
|
||||
url="https://about.gitlab.com/images"
|
||||
+ "/press/logo/png/gitlab-icon-rgb.png"
|
||||
)
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
@cog_ext.cog_subcommand(
|
||||
base="gl",
|
||||
name="milestone",
|
||||
description="Get a milestone from GitLab",
|
||||
guild_ids=guild_ids,
|
||||
options=[
|
||||
create_option(
|
||||
name="id",
|
||||
description="Milestone ID",
|
||||
option_type=4,
|
||||
required=True,
|
||||
)
|
||||
],
|
||||
)
|
||||
async def _milestone(self, ctx: SlashContext, id: int):
|
||||
try:
|
||||
milestone = self.project.milestones.get(int(id))
|
||||
except gitlab.exceptions.GitlabGetError:
|
||||
await ctx.send("Issue does not exist.", hidden=True)
|
||||
return
|
||||
|
||||
created_at = datetime.strptime(
|
||||
milestone.created_at, "%Y-%m-%dT%H:%M:%S.%fZ"
|
||||
).strftime("%Y-%m-%d %H:%M:%S UTC")
|
||||
|
||||
fields = [
|
||||
Field(
|
||||
name="State",
|
||||
value=milestone.state[0].upper() + milestone.state[1:],
|
||||
),
|
||||
Field(name="Start Date", value=milestone.start_date),
|
||||
Field(name="Due Date", value=milestone.due_date),
|
||||
Field(name="Created At", value=created_at),
|
||||
]
|
||||
|
||||
if milestone.updated_at:
|
||||
updated_at = datetime.strptime(
|
||||
milestone.updated_at, "%Y-%m-%dT%H:%M:%S.%fZ"
|
||||
).strftime("%Y-%m-%d %H:%M:%S UTC")
|
||||
fields.append(Field(name="Updated At", value=updated_at))
|
||||
|
||||
if len(milestone.title) > 200:
|
||||
milestone.title = milestone.title[:200] + "..."
|
||||
|
||||
embed = build_embed(
|
||||
title=f"[#{milestone.iid}] {milestone.title}",
|
||||
description=milestone.description,
|
||||
fields=fields,
|
||||
color="#00FFEE",
|
||||
url=milestone.web_url,
|
||||
)
|
||||
embed.set_thumbnail(
|
||||
url="https://about.gitlab.com/images"
|
||||
+ "/press/logo/png/gitlab-icon-rgb.png"
|
||||
)
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
|
||||
def setup(bot):
|
||||
if get_config().gitlab_token:
|
||||
bot.add_cog(GitlabCog(bot))
|
||||
|
|
@ -24,6 +24,7 @@ class Config(object):
|
|||
logo: str,
|
||||
mongo: dict,
|
||||
urls: dict,
|
||||
gitlab_token: str = None,
|
||||
max_messages: int = 1000,
|
||||
):
|
||||
self.token = token
|
||||
|
|
@ -32,6 +33,7 @@ class Config(object):
|
|||
self.mongo = mongo
|
||||
self.urls = urls
|
||||
self.max_messages = max_messages
|
||||
self.gitlab_token = gitlab_token
|
||||
db = DBManager(config=mongo).mongo.jarvis.config
|
||||
db_config = db.find()
|
||||
for item in db_config:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue