Add Python Requirement Scan (#40)

* add python requirement

* clean up

Co-authored-by: anotherminh <[email protected]>
Co-authored-by: jwahjudi <[email protected]>

Author: 3 people

src/plugins/constants.ts

@@ -1,11 +1,16 @@
 import { SiloDiscoveryConfig } from './types';
-import { javascriptPackageJson, cocoapods } from './integrations';
+import {
+  javascriptPackageJson,
+  cocoapods,
+  pythonRequirementsTxt,
+} from './integrations';
 
 export const SILO_DISCOVERY_FUNCTIONS: {
   [k in string]: SiloDiscoveryConfig;
 } = {
   javascriptPackageJson,
   cocoapods,
+  pythonRequirementsTxt,
 };
 
 /**

src/plugins/integrations/index.ts

@@ -1,2 +1,3 @@
 export * from './cocoapods';
-export * from './packageJson';
+export * from './javascriptPackageJson';
+export * from './pythonRequirementsTxt';

src/plugins/integrations/packageJson.ts renamed to src/plugins/integrations/javascriptPackageJson.ts

@@ -0,0 +1,27 @@
+import { readFileSync } from 'fs';
+import { SiloDiscoveryConfig } from '../types';
+
+export const pythonRequirementsTxt: SiloDiscoveryConfig = {
+  supportedFiles: ['requirements.txt'],
+  ignoreDirs: ['build', 'lib', 'lib64'],
+  scanFunction: (filePath) => {
+    const lines = readFileSync(filePath)
+      .toString()
+      // split on new line character
+      .split(String.fromCharCode(10));
+
+    const deps: string[] = [];
+
+    lines.map((line) => {
+      if (line.includes('==')) {
+        const dep = line.split('==')[0];
+        if (!dep.includes('#')) {
+          deps.push(dep);
+        }
+      }
+      return deps;
+    });
+
+    return deps;
+  },
+};