tldr/scripts/send-to-bot.py

#!/usr/bin/env python3
# SPDX-License-Identifier: MIT

import os
import sys
import requests

BOT_URL = 'https://tldr-bot.starbeamrainbowlabs.com'

COMMENT_ERROR="""
The [build](https://github.com/tldr-pages/tldr/actions/runs/{build_id}) for this PR failed with the following error(s):

```
{content}
```

Please fix the error(s) and push again.
"""

COMMENT_CHECK="""
Hello! I've noticed something unusual when checking this PR:

{content}

Is this intended? If so, just ignore this comment. Otherwise, please double-check the commits.
"""

################################################################################

def post_comment(pr_id, body, once):
  endpoint = BOT_URL + '/comment'

  if once:
    endpoint += '/once'

  data = {'pr_id': pr_id, 'body': body}

  try:
    with requests.post(endpoint, json=data) as r:
      if r.status_code != requests.codes.ok:
        print('Error: tldr-bot responded with code', r.status_code, file=sys.stderr)
        print(r.text, file=sys.stderr)
        return False
  except requests.exceptions.RequestException as e:
    print('Error sending data to tldr-bot:', str(e), file=sys.stderr)
    return False
    
  return True

def main(action):
  if action not in ('report-errors', 'report-check-results'):
    print('Unknown action:', action, file=sys.stderr)
    sys.exit(1)

  content = sys.stdin.read().strip()

  if action == 'report-errors':
    comment_body = COMMENT_ERROR.format(build_id=BUILD_ID, content=content)
    comment_once = False
  elif action == 'report-check-results':
    comment_body = COMMENT_CHECK.format(content=content)
    comment_once = True

  if post_comment(PR_ID, comment_body, comment_once):
    print('Success.')
  else:
    print('Error sending data to tldr-bot!', file=sys.stderr)

################################################################################

if __name__ == '__main__':
  REPO_SLUG = os.environ.get('GITHUB_REPOSITORY')
  PR_ID = os.environ.get('PULL_REQUEST_ID')
  BUILD_ID = os.environ.get('GITHUB_RUN_ID')

  if PR_ID is None or BUILD_ID is None or REPO_SLUG is None:
    print('Needed environment variables are not set.', file=sys.stderr)
    sys.exit(1)

  if PR_ID is None or PR_ID == 'false':
    print('Not a pull request, refusing to run.', file=sys.stderr)
    sys.exit(0)

  if len(sys.argv) != 2:
    print('Usage:', sys.argv[0], '<ACTION>', file=sys.stderr)
    sys.exit(1)

  main(sys.argv[1])