[Bitbucket]: Pull Requests: Support Bitbucket Server

.dockerignore

@@ -2,3 +2,4 @@ node_modules/
 shields.env
 .git/
 .gitignore
+.vscode/

.gitignore

@@ -1,4 +1,5 @@
 # Project-specific files
+/.vscode
 /Verdana.ttf
 /analytics.json
 /coverage.html

package.json

@@ -96,6 +96,7 @@
     "heroku-postbuild": "run-s --silent build",
     "analyze": "cross-env ANALYZE=true LONG_CACHE=false BASE_URL=https://img.shields.io npm run build",
     "start:server": "cross-env HANDLE_INTERNAL_ERRORS=false RATE_LIMIT=false node server 8080 ::",
+    "start:server:debug": "cross-env ALLOWED_ORIGIN=http://localhost:3000 HANDLE_INTERNAL_ERRORS=false RATE_LIMIT=false node --nolazy --inspect=9229 server 8080 ::",
     "now-start": "node server",
     "heroku-start": "node scripts/export-heroku-secrets-cli.js && node server",
     "prestart": "run-s --silent depcheck defs features",

secret.tpl.json

@@ -1,8 +1,12 @@
-{
-  "gh_client_id": "${GH_CLIENT_ID}",
-  "gh_client_secret": "${GH_CLIENT_SECRET}",
-  "shieldsIps": [ "${SHIELDS_IP}" ],
-  "gh_token": "${GH_TOKEN}",
-  "npm_token": "${NPM_TOKEN}",
-  "wheelmap_token": "${WHEELMAP_TOKEN}"
-}
+{
+  "gh_client_id": "${GH_CLIENT_ID}",
+  "gh_client_secret": "${GH_CLIENT_SECRET}",
+  "shieldsIps": [ "${SHIELDS_IP}" ],
+  "gh_token": "${GH_TOKEN}",
+  "npm_token": "${NPM_TOKEN}",
+  "wheelmap_token": "${WHEELMAP_TOKEN}",
+  "bitbucket_username": "${BITBUCKET_USERNAME}",
+  "bitbucket_password": "${BITBUCKET_PASSWORD}",
+  "bitbucket_server_username": "${BITBUCKET_SERVER_USERNAME}",
+  "bitbucket_server_password": "${BITBUCKET_SERVER_PASSWORD}"
+}

services/bitbucket/bitbucket-pull-request.service.js

@@ -2,6 +2,7 @@
 
 const Joi = require('joi')
 const BaseJsonService = require('../base-json')
+const serverSecrets = require('../../lib/server-secrets')
 const { metric } = require('../../lib/text-formatters')
 const { nonNegativeInteger } = require('../validators')
 
@@ -14,14 +15,64 @@ function pullRequestClassGenerator(raw) {
   const badgeSuffix = raw ? '' : ' open'
 
   return class BitbucketPullRequests extends BaseJsonService {
-    async fetch({ user, repo }) {
-      const url = `https://bitbucket.org/api/2.0/repositories/${user}/${repo}/pullrequests/`
-      return this._requestJson({
-        url,
+    async fetchCloud({ args, user, repo }) {
+      args.url = `https://bitbucket.org/api/2.0/repositories/${user}/${repo}/pullrequests/`
+      args.options = { qs: { state: 'OPEN', limit: 0 } }
+
+      if (
+        serverSecrets &&
+        serverSecrets.bitbucket_username &&
+        serverSecrets.bitbucket_password
+      ) {
+        args.options.auth = {
+          user: serverSecrets.bitbucket_username,
+          pass: serverSecrets.bitbucket_password,
+        }
+      }
+
+      return this._requestJson(args)
+    }
+
+    async fetchServer({ args, proto, hostAndPath, user, repo }) {
+      args.url = `${proto}://${hostAndPath}/rest/api/1.0/projects/${user}/repos/${repo}/pull-requests`
+      args.options = {
+        qs: {
+          state: 'OPEN',
+          limit: 100,
+          withProperties: false,
+          withAttributes: false,
+        },
+      }
+
+      if (
+        serverSecrets &&
+        serverSecrets.bitbucket_server_username &&
+        serverSecrets.bitbucket_server_password
+      ) {
+        args.options.auth = {
+          user: serverSecrets.bitbucket_server_username,
+          pass: serverSecrets.bitbucket_server_password,
+        }
+      }
+
+      return this._requestJson(args)
+    }
+
+    async fetch({ proto, hostAndPath, user, repo }) {
+      const args = {
         schema: bitbucketPullRequestsSchema,
-        options: { qs: { state: 'OPEN', limit: 0 } },
-        errorMessages: { 403: 'private repo' },
-      })
+        errorMessages: {
+          401: 'invalid credentials',
+          403: 'private repo',
+          404: 'not found',
+        },
+      }
+
+      if (hostAndPath !== undefined) {
+        return this.fetchServer({ args, proto, hostAndPath, user, repo })
+      } else {
+        return this.fetchCloud({ args, user, repo })
+      }
     }
 
     static render({ prs }) {
@@ -31,8 +82,8 @@ function pullRequestClassGenerator(raw) {
       }
     }
 
-    async handle({ user, repo }) {
-      const data = await this.fetch({ user, repo })
+    async handle({ proto, hostAndPath, user, repo }) {
+      const data = await this.fetch({ proto, hostAndPath, user, repo })
       return this.constructor.render({ prs: data.size })
     }
 
@@ -47,19 +98,33 @@ function pullRequestClassGenerator(raw) {
     static get route() {
       return {
         base: `bitbucket/${routePrefix}`,
-        format: '([^/]+)/([^/]+)',
-        capture: ['user', 'repo'],
+        format: `(?:(http|https)/(.+)/)?([^/]+)/([^/]+)`,
+        capture: ['proto', 'hostAndPath', 'user', 'repo'],
       }
     }
 
     static get examples() {
       return [
         {
           title: 'Bitbucket open pull requests',
-          exampleUrl: 'atlassian/python-bitbucket',
+          namedParams: {
+            user: 'atlassian',
+            repo: 'python-bitbucket',
+          },
           pattern: ':user/:repo',
           staticExample: this.render({ prs: 22 }),
         },
+        {
+          title: 'Bitbucket Server open pull requests',
+          namedParams: {
+            proto: 'https',
+            hostAndPath: 'bitbucket.mydomain.net',
+            user: 'foo',
+            repo: 'bar',
+          },
+          pattern: ':proto/:hostAndPath/:user/:repo',
+          staticExample: this.render({ prs: 42 }),
+        },
       ]
     }
   }

services/bitbucket/bitbucket.tester.js

@@ -85,6 +85,26 @@ t.create('pr (private repo)')
   .get('/pr/chris48s/example-private-repo.json')
   .expectJSON({ name: 'pull requests', value: 'private repo' })
 
+t.create('pr (server)')
+  .get('/pr/https/bitbucket.mydomain.net/project/repo.json')
+  .intercept(nock =>
+    nock('https://bitbucket.mydomain.net/rest/api/1.0/projects')
+      .get('/project/repos/repo/pull-requests')
+      .query({
+        state: 'OPEN',
+        limit: 100,
+        withProperties: false,
+        withAttributes: false,
+      })
+      .reply(200, { size: 42 })
+  )
+  .expectJSONTypes(
+    Joi.object().keys({
+      name: 'pull requests',
+      value: isMetricOpenIssues,
+    })
+  )
+
 // tests for Bitbucket Pipelines
 
 function bitbucketApiResponse(status) {