96 lines
2.2 KiB
Python
96 lines
2.2 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
|
||
|
|
# Set variables
|
||
|
|
USER_LIB_PATH = sys.argv[1]
|
||
|
|
USER_LIBS = [] ;
|
||
|
|
|
||
|
|
includeRegex = re.compile("(?<=^\#include\s[\<\"])(.*)(?=\.h[\>\"])", re.DOTALL|re.M)
|
||
|
|
|
||
|
|
MAIN_SRCS = [] ;
|
||
|
|
MAIN_LIBS = [] ;
|
||
|
|
|
||
|
|
LIBS_DEPS = [] ;
|
||
|
|
LIBS_DEPS_STACK = [] ;
|
||
|
|
|
||
|
|
# Find local sources .ino, .c or .cpp
|
||
|
|
for file in os.listdir(os.curdir):
|
||
|
|
if file.endswith((".c", ".cpp", ".ino")):
|
||
|
|
MAIN_SRCS.append(file)
|
||
|
|
|
||
|
|
# Find all USER_LIBS
|
||
|
|
for path, dirs, files in os.walk(USER_LIB_PATH):
|
||
|
|
for d in dirs:
|
||
|
|
USER_LIBS.append(d)
|
||
|
|
|
||
|
|
# Find MAIN_LIBS included in MAIN_SRCS
|
||
|
|
for src in MAIN_SRCS:
|
||
|
|
currentFile = open(src)
|
||
|
|
includes = []
|
||
|
|
|
||
|
|
for line in currentFile:
|
||
|
|
match = includeRegex.search(line)
|
||
|
|
if match:
|
||
|
|
if match.group(1) in USER_LIBS:
|
||
|
|
MAIN_LIBS.append(match.group(1))
|
||
|
|
|
||
|
|
MAIN_LIBS = list(sorted(MAIN_LIBS))
|
||
|
|
|
||
|
|
# Find LIBS_DEPS includes in MAIN_LIBS
|
||
|
|
for lib in MAIN_LIBS:
|
||
|
|
if lib in USER_LIBS:
|
||
|
|
currentFile = open(USER_LIB_PATH + "/" + lib + "/" + lib + ".h")
|
||
|
|
|
||
|
|
for line in currentFile:
|
||
|
|
match = includeRegex.search(line)
|
||
|
|
if match:
|
||
|
|
if match.group(1) in USER_LIBS and match.group(1) not in MAIN_LIBS:
|
||
|
|
LIBS_DEPS_STACK.append(match.group(1))
|
||
|
|
|
||
|
|
LIBS_DEPS_STACK = sorted(set(LIBS_DEPS_STACK))
|
||
|
|
|
||
|
|
# Recursively find all dependencies of every libraries in USER_LIB_PATH
|
||
|
|
while LIBS_DEPS_STACK:
|
||
|
|
for lib in LIBS_DEPS_STACK:
|
||
|
|
if lib in USER_LIBS:
|
||
|
|
currentFile = open(USER_LIB_PATH + "/" + lib + "/" + lib + ".h")
|
||
|
|
|
||
|
|
for line in currentFile:
|
||
|
|
match = includeRegex.search(line)
|
||
|
|
if match:
|
||
|
|
if match.group(1) in USER_LIBS and match.group(1) not in LIBS_DEPS_STACK or match.group(1) in LIBS_DEPS and match.group(1) not in MAIN_LIBS:
|
||
|
|
LIBS_DEPS_STACK.append(match.group(1))
|
||
|
|
|
||
|
|
else:
|
||
|
|
LIBS_DEPS.append(lib)
|
||
|
|
if lib in LIBS_DEPS_STACK:
|
||
|
|
LIBS_DEPS_STACK.remove(lib)
|
||
|
|
|
||
|
|
LIBS_DEPS_STACK = sorted(set(LIBS_DEPS_STACK))
|
||
|
|
# print(LIBS_DEPS_STACK)
|
||
|
|
|
||
|
|
LIBS_DEPS = sorted(set(LIBS_DEPS))
|
||
|
|
|
||
|
|
# print("Main libraries: ")
|
||
|
|
# print(MAIN_LIBS);
|
||
|
|
# print("")
|
||
|
|
# print("Dependencies stack: ")
|
||
|
|
# print(LIBS_DEPS_STACK)
|
||
|
|
# print("")
|
||
|
|
# print("Libraries dependencies: ")
|
||
|
|
# print(LIBS_DEPS);
|
||
|
|
|
||
|
|
def outputLibs(libArray):
|
||
|
|
for lib in libArray:
|
||
|
|
print(lib),
|
||
|
|
print("")
|
||
|
|
|
||
|
|
print("MAIN_LIBS"),
|
||
|
|
outputLibs(MAIN_LIBS)
|
||
|
|
|
||
|
|
print("LIBS_DEPS"),
|
||
|
|
outputLibs(LIBS_DEPS)
|