Merge pull request #6 from saadmk11/private-repo-support

Added Support For Private Repositories

README.md

@@ -35,6 +35,8 @@ Put this inside your ``.github/workflows/workflow.yml`` file:
           # you can use secrets for it as well
           USERNAME:  'test'
           EMAIL:  '[email protected]'
+          # optional only required for ``private`` repositories
+          GITHUB_TOKEN: ${{secrets.GITHUB_TOKEN}}
 ```
 
 

scripts/changelog-ci.py

@@ -10,10 +10,15 @@
 
 class ChangelogCI:
 
-    def __init__(self, repository, event_path, filename='CHANGELOG.md'):
+    def __init__(
+        self, repository,
+        event_path, filename='CHANGELOG.md',
+        token=None
+    ):
         self.repository = repository
         self.event_path = event_path
         self.filename = filename
+        self.token = token
 
     def _pull_request_title(self):
         title = ''
@@ -23,14 +28,26 @@ def _pull_request_title(self):
 
         return title
 
+    def _get_request_headers(self):
+        headers = {
+            'Accept': 'application/vnd.github.v3+json'
+        }
+
+        if self.token:
+            headers.update({
+                'authorization': 'Bearer {token}'.format(token=self.token)
+            })
+
+        return headers
+
     def _get_latest_release_date(self):
         url = (
             'https://api.github.com/repos/{repo_name}/releases/latest'
         ).format(repo_name=self.repository)
 
         published_date = ''
 
-        response = requests.get(url)
+        response = requests.get(url, headers=self._get_request_headers())
 
         if response.status_code == 200:
             response_data = response.json()
@@ -87,7 +104,7 @@ def _get_pull_requests_after_last_release(self):
             merged_date_filter=merged_date_filter
         )
 
-        response = requests.get(url)
+        response = requests.get(url, headers=self._get_request_headers())
 
         if response.status_code == 200:
             response_data = response.json()
@@ -160,6 +177,7 @@ def write_changelog(self):
     event_path = os.environ['GITHUB_EVENT_PATH']
     repository = os.environ['GITHUB_REPOSITORY']
     filename = os.environ['INPUT_CHANGELOG_FILENAME']
+    token = os.environ.get('GITHUB_TOKEN')
 
-    ci = ChangelogCI(repository, event_path, filename=filename)
+    ci = ChangelogCI(repository, event_path, filename=filename, token=token)
     ci.write_changelog()