Create custom got clients to set user agent

geodatagouv · Oct 11, 2018 · 7f6088a · 7f6088a
1 parent f40d150
commit 7f6088a
Show file tree

Hide file tree

Showing 8 changed files with 24 additions and 6 deletions.
diff --git a/lib/api/middlewares/thumbnail-proxy.js b/lib/api/middlewares/thumbnail-proxy.js
@@ -1,9 +1,10 @@
 const {join} = require('path')
-const got = require('got')
 const fileType = require('file-type')
 const fresh = require('fresh')
 const cacheControl = require('@tusbar/cache-control')
 
+const got = require('../../got')
+
 const defaultThumbnailPath = join(__dirname, '../../../france/img/unavailable-thumbnail.png')
 
 const ACCEPTED_MIME_TYPES = [

diff --git a/lib/api/routes/links.js b/lib/api/routes/links.js
@@ -2,9 +2,9 @@ const {createWriteStream} = require('fs')
 const {join} = require('path')
 const {Router} = require('express')
 const unzipper = require('unzipper')
-const got = require('got')
 const pump = require('pump')
 
+const got = require('../../got')
 const {getLink} = require('../../services/link-proxy')
 const {createTempDirectory} = require('../../utils/tempfile')
 const ogr2ogr = require('../../extract/ogr2ogr')

diff --git a/lib/got.js b/lib/got.js
@@ -0,0 +1,9 @@
+const got = require('got')
+
+const pkg = require('../package.json')
+
+module.exports = got.extend({
+  headers: {
+    'user-agent': `geoplatform/${pkg.version} (+https://geo.data.gouv.fr)`
+  }
+})
diff --git a/lib/services/link-proxy.js b/lib/services/link-proxy.js
@@ -1,6 +1,6 @@
 'use strict'
 
-const got = require('got')
+const got = require('../got')
 
 const {LINK_PROXY_URL} = process.env
 

diff --git a/plugins/publish-to-udata/geogw.js b/plugins/publish-to-udata/geogw.js
@@ -1,6 +1,6 @@
 'use strict'
 
-const got = require('got')
+const got = require('./got')
 
 const ROOT_URL = process.env.GEOGW_URL + '/api/geogw'
 const TOKEN = process.env.GEOGW_TOKEN

diff --git a/plugins/publish-to-udata/got.js b/plugins/publish-to-udata/got.js
@@ -0,0 +1,7 @@
+const got = require('got')
+
+module.exports = got.extend({
+  headers: {
+    'user-agent': 'geoplatform/udata-publisher/v1.0.0 (+https://geo.data.gouv.fr)'
+  }
+})
diff --git a/plugins/publish-to-udata/udata-proxy.js b/plugins/publish-to-udata/udata-proxy.js
@@ -1,5 +1,6 @@
 const {Router} = require('express')
-const got = require('got')
+
+const got = require('./got')
 
 const ALLOWED_METHODS = ['GET', 'HEAD', 'POST', 'PUT', 'DELETE']
 const baseUrl = process.env.DATAGOUV_URL + '/api'

diff --git a/plugins/publish-to-udata/udata.js b/plugins/publish-to-udata/udata.js
@@ -1,4 +1,4 @@
-const got = require('got')
+const got = require('./got')
 
 const {DATAGOUV_URL, UDATA_PUBLICATION_USER_API_KEY} = process.env