Merge pull request #109 from Nibba2018/library_c

Extracts libraries used in C source files

Author: peti2001

src/language/C.py

@@ -0,0 +1,30 @@
+import re
+
+def extract_libraries(files):
+    """Extracts a list of imports that were used in the files
+
+    Parameters
+    ----------
+    files : []string
+        Full paths to files that need to be analysed
+
+    Returns
+    -------
+    dict
+        imports that were used in the provided files, mapped against the language
+    """
+
+    res = []
+
+    # regex to find imports like #include<stdio.h> or #include "foo.h"
+    regex = re.compile(r"#include\s?[<\"]([/a-zA-Z0-9.-]+)[\">]", re.IGNORECASE)
+
+    for f in files:
+        with open(file=f, mode='r', errors='ignore') as fr:
+
+            contents = ' '.join(fr.readlines())
+            matches = regex.findall(contents)
+
+        if matches:
+            res.extend(matches)
+    return {"C": res}

test/fixtures/C.c

@@ -0,0 +1,15 @@
+#include<assert.h>
+#include "complex.h"
+#include <ctype.h>
+#include "stdio.h"
+#include<float.h>
+#include"string.h"
+#include "hey/sup/iomanip.h"
+#include<Hey/ssup/math.h>
+#include"hello/how/stdlib.h"
+#include <great/wchar.h>
+#include"stdbool.h"
+#include<stdint.h>
+#include "hello12"
+#include<WsSup34>
+#include <heyYo3-lol>

test/test_C.py

@@ -0,0 +1,27 @@
+import os
+from language.C import extract_libraries
+
+def test_extract_libraries():
+    dir_path = os.path.dirname(os.path.realpath(__file__))
+    files = ['fixtures/C.c']
+    fq_files = [os.path.join(dir_path, f) for f in files]
+    libs = extract_libraries(fq_files)
+    assert len(libs) == 1
+
+    assert libs['C'] == [
+        "assert.h",
+        "complex.h",
+        "ctype.h",
+        "stdio.h",
+        "float.h",
+        "string.h",
+        "hey/sup/iomanip.h",
+        "Hey/ssup/math.h",
+        "hello/how/stdlib.h",
+        "great/wchar.h",
+        "stdbool.h",
+        "stdint.h",
+        "hello12",
+        "WsSup34",
+        "heyYo3-lol"
+    ]