def __init__(self, dispatcher: Dispatcher):

self.dispatcher = dispatcher

self.logger = logging.getLogger(self.__class__.__qualname__)

def handle_auth_update(self, update: GithubAuthUpdate, context: CallbackContext):

user_id = update.state[0]

message_id = update.state[1]

# noinspection PyProtectedMember

context.user_data = self.dispatcher.user_data[user_id]

access_token = github_api.get_oauth_access_token(update.code, update.raw_state)

context.user_data['access_token'] = access_token

from bot.settings import login_menu

context.menu_stack = ['settings', 'login']

edit_menu_by_id(user_id, message_id, context, login_menu)

def handle_update(self, update: GithubUpdate, context: CallbackContext):

return getattr(self, update.event, self.unknown)(update, context)

def unknown(self, update, _):

self.logger.warning('Unknown event type %s. Data: %s', update.event, update.payload)

def ping(self, update, _):

self.logger.info('PING: %s', update.payload)

def _iter_repos(self, repository):

repo_id = repository['id']

for chat_id, chat_data in self.dispatcher.chat_data.items():

if 'repos' in chat_data:

for repo in chat_data['repos'].values():

if repo.id == repo_id:

yield chat_id, chat_data, repo

def _send(self, repo, text, check_repo: Callable[[Repo], bool], suffix=REPLY_MESSAGE):

truncated_text = {}

for chat_id, chat_data, repo in self._iter_repos(repo):

if check_repo(repo):

truncation_limit = chat_data.get('truncation_limit', DEFAULT_TRUNCATION_LIMIT)

try:

message_text = truncated_text[truncation_limit]

except KeyError:

message_text = truncate(text, TRUNCATED_MESSAGE, suffix, max_length=truncation_limit)

try:

self.dispatcher.bot.send_message(chat_id=chat_id, text=message_text,

parse_mode=ParseMode.HTML, disable_web_page_preview=True, disable_notification=True)

except TelegramError:

logging.error('error while sending github update', exc_info=1)

def issues(self, update, _):

# Issue opened, edited, closed, reopened, assigned, unassigned, labeled,

# unlabeled, milestoned, or demilestoned.

# TODO: Possibly support editing, closing, reopening, etc. of issues

if update.payload['action'] in ('opened', 'closed', 'reopened'):

issue = update.payload['issue']

author = issue['user']

repo = update.payload['repository']

if update.payload['action'] == 'opened':

desc = '🐛 New issue'

if issue['body']:

text = "\n\n" + render_github_markdown(issue['body'], repo['full_name'])

else:

text = "\n\nNo description provided."

elif update.payload['action'] == 'closed':

desc = '✅ Issue closed'

text = ""

elif update.payload['action'] == 'reopened':

desc = '🔄 Issue reopened'

text = ""

issue_link = link(issue['html_url'], f'{repo["full_name"]}#{issue["number"]} {issue["title"]}')

author_link = link(author['html_url'], '@' + author['login'])

data_link = encode_data_link(('issue', repo['full_name'], issue['number'], author['login']))

text = f'{data_link}{desc} {issue_link}\nby {author_link}{text}'

self._send(repo, text, lambda r: r.issues)

def issue_comment(self, update, context):

# Any time a comment on an issue or pull request is created, edited, or deleted.

# TODO: Possibly support editing and closing of comments?

if update.payload['action'] == 'created':

issue = update.payload['issue']

comment = update.payload['comment']

author = comment['user']

repo = update.payload['repository']

is_pull_request = 'pull_request' in issue

text = render_github_markdown(comment['body'], repo['full_name'])

issue_link = link(issue['html_url'], f'{repo["full_name"]}#{issue["number"]} {issue["title"]}')

author_link = link(author['html_url'], '@' + author['login'])

data_link = encode_data_link(('pull request' if is_pull_request else 'issue',

repo['full_name'], issue['number'], author['login']))

text = f'{data_link}💬 New comment on {issue_link}\nby {author_link}\n\n{text}'

self._send(repo, text, lambda r: r.pull_comments if is_pull_request else r.issue_comments)

def discussion(self, update, _):

# Discussion created, edited, deleted, pinned, unpinned, locked, unlocked, transferred,

# category_changed, answered, unanswered, labeled, or unlabeled

if update.payload['action'] == 'created':

discussion = update.payload['discussion']

author = discussion['user']

repo = update.payload['repository']

text = render_github_markdown(discussion['body'], repo['full_name'])

issue_link = link(discussion['html_url'], f'{repo["full_name"]}#{discussion["number"]} {discussion["title"]}')

author_link = link(author['html_url'], '@' + author['login'])

data_link = encode_data_link(('discussion', repo['full_name'], discussion['number'], author['login']))

text = f'{data_link}⁉️ New discussion {issue_link}\nby {author_link}\n\n{text}'

self._send(repo, text, lambda r: r.issues, suffix="")

def pull_request(self, update, context):

# Pull request opened, closed, reopened, edited, assigned, unassigned, review requested,

# review request removed, labeled, unlabeled, or synchronized.

# TODO: Possibly support closed, reopened, edited, assigned etc.

if update.payload['action'] == 'opened':

pull_request = update.payload['pull_request']

author = pull_request['user']

repo = update.payload['repository']

if pull_request['body']:

text = render_github_markdown(pull_request['body'], repo['full_name'])

else:

text = "No description provided"

pull_request_link = link(pull_request['html_url'],

f'{repo["full_name"]}#{pull_request["number"]} {pull_request["title"]}')

author_link = link(author['html_url'], '@' + author['login'])

data_link = encode_data_link(('pull request', repo['full_name'], pull_request['number'], author['login']))

text = f'{data_link}🔌 New pull request {pull_request_link}\nby {author_link}\n\n{text}'

self._send(repo, text, lambda r: r.pulls)

def pull_request_review(self, update, context):

# Pull request review submitted, edited, or dismissed.

# TODO: Possibly support edited and dismissed?

if update.payload['action'] == 'submitted':

review = update.payload['review']

pull_request = update.payload['pull_request']

author = review['user']

repo = update.payload['repository']

# Warning: Currently only works for public repos. Otherwise, we need an access token

# to access the comments.

comments = github_api.get_pull_request_review_comments(

repo['owner']['login'],

repo['name'],

pull_request['number'],

review['id'],

)

all_replies = all(comment.get("in_reply_to_id") for comment in comments)

if review['body']:

text = "\n\n" + render_github_markdown(review['body'], repo['full_name'])

else:

text = ""

if not text and all_replies:

# Don't ping when review only contains replies

return

review_link = link(review['html_url'],

f'{repo["full_name"]}#{pull_request["number"]} {pull_request["title"]}')

author_link = link(author['html_url'], '@' + author['login'])

data_link = encode_data_link(('pull request', repo['full_name'], pull_request['number'], author['login']))

if review['state'] in ('commented', 'approved', 'request_changes', 'changes_requested'):

if review['state'] == 'commented':

state = 'Commented'

emoji = '💬'

elif review['state'] == 'approved':

state = 'Approved'

emoji = '✅'

elif review['state'] in ('request_changes', 'changes_requested'):

state = 'Changes requested'

emoji = '‼️'

text = f'{data_link}{emoji} New pull request review {review_link}\n{state} by {author_link}{text}'

self._send(repo, text, lambda r: r.pull_reviews)

def pull_request_review_comment(self, update, context):

# Pull request diff comment created, edited, or deleted.

if update.payload['action'] == 'created':

pull_request = update.payload['pull_request']

comment = update.payload['comment']

author = comment['user']

repo = update.payload['repository']

diff_hunk = f'<pre>{comment["path"]}\n{comment["diff_hunk"]}</pre>'

text = render_github_markdown(comment['body'], repo['full_name'])

issue_link = link(comment['html_url'],

f'{repo["full_name"]}#{pull_request["number"]} {pull_request["title"]}')

author_link = link(author['html_url'], '@' + author['login'])

data_link = encode_data_link(('pull request review comment',

repo['full_name'],

pull_request['number'],

comment['in_reply_to_id'] if 'in_reply_to_id' in comment else comment['id'],

author['login'],))

text = f'{data_link}💬 New pull request review comment {issue_link}\nby {author_link}\n{diff_hunk}\n\n{text}'

self._send(repo, text, lambda r: r.pull_review_comments)

def push(self, update, context):

# Triggered on a push to a repository branch.

# Branch pushes and repository tag pushes also trigger webhook push events.

commits = update.payload['commits']

ref = update.payload['ref']

if commits and ref.startswith('refs/heads/'):

branch = ref[len('refs/heads/'):]

repo = update.payload['repository']

compare = update.payload['compare']

text = f'🔨 <a href="{compare}">{len(commits)} new commits</a> to {repo["full_name"]}:{branch}\n\n'

for commit in commits:

text += f'<a href="{commit["url"]}">{commit["id"][:7]}</a>: {commit["message"]} by {commit["author"]["name"]}\n'

self._send(repo, text, lambda r: (r.push_main or r.push) if branch == repo["default_branch"] else r.push,

suffix='')

def gollum(self, update, context):

# Wiki page is created or updated.

pages = update.payload['pages']

repo = update.payload['repository']

sender = update.payload['sender']

text = f'🔨 {len(pages)} {repo["full_name"]} wiki page{"s" if len(pages) > 1 else ""} were updated '

sender_link = link(sender['html_url'], '@' + sender['login'])

text += f'by {sender_link}\n\n'

for page in pages:

compare_url = f'{page["html_url"]}/_compare/{page["sha"]}'

text += f'<a href="{page["html_url"]}">{page["title"]}</a> (<a href="{compare_url}">compare</a>)\n'

self._send(repo, text, lambda r: r.wiki_pages, suffix='')

def commit_comment(self, update, context):

if update.payload['action'] == 'created':

repo = update.payload['repository']

comment = update.payload['comment']

author = comment['user']

author_link = link(author['html_url'], '@' + author['login'])

text = f'💬 <a href="{comment["html_url"]}">New comment</a> on commit {comment["commit_id"][:7]} by {author_link}'

position, line, path = comment['position'], comment['line'], comment['path']

if path:

text += f'\nPath: {path}'

if line:

text += f'\nLine: {line}'

if position == 1:

text += ' (before)'

elif position == 2:

text += ' (after)'

text += f'\n\n{comment["body"]}'

self._send(repo, text, lambda r: r.commit_comments, suffix='')

# def integration_installation_repositories(self, update, context):

# new_repos = [{'id': repo['id'], 'full_name': repo['full_name']} for repo in

# update.payload['repositories_added']]

# # TODO: Implement repositories_removed

#

# self.logger.debug('New installation repos: %s', new_repos)

#

# context.github_data.setdefault('repos', []).extend(new_repos)

#

# def integration_installation(self, update, context):

# new_repos = [{'id': repo['id'], 'full_name': repo['full_name']} for repo in update.payload['repositories']]

#

# self.logger.debug('New installation. Repos: %s', new_repos)

#