feat: Get top parent name and customize UA

snyk-labs · Jul 20, 2020 · 7c66192 · 7c66192
1 parent a72e834
commit 7c66192
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/lib/request/request.ts b/src/lib/request/request.ts
@@ -10,15 +10,29 @@ interface snykRequest {
     requestId?: string
 }
 
+const getTopParentModuleName = (parent: NodeModule | null): string => {
+    if(parent == null){
+        return ''
+    }
+    if(parent?.parent){
+        return getTopParentModuleName(parent.parent)
+    } else {
+        return parent?.paths[0].split('/')[parent.paths[0].split('/').length-2]+'/' as string
+    }
+}
+
 const makeSnykRequest = async (request: snykRequest, snykToken: string = '') => {
     const userConfig = getConfig()
     const token = snykToken == '' ? userConfig.token : snykToken
 
+    let topParentModuleName = getTopParentModuleName(module.parent)
+
     const requestHeaders: Object = {
         'Content-Type': 'application/json',
         'Authorization': 'token '+ token,
-        'User-Agent': 'tech-services/snyk-request-manager/1.0'
+        'User-Agent': `${topParentModuleName}tech-services/snyk-request-manager/1.0`
     }
+
     const apiClient = axios.create({
         baseURL: userConfig.endpoint,
         responseType: 'json',