100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
|
import os
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
def recursiveSearchForDockerComposeFiles(watchFolder: Path):
|
||
|
|
dockerComposeList = []
|
||
|
|
for root, _, files in os.walk(watchFolder):
|
||
|
|
for file in files:
|
||
|
|
if file == "docker-compose.yaml":
|
||
|
|
dockerComposeList.append(Path(root) / file)
|
||
|
|
return dockerComposeList
|
||
|
|
|
||
|
|
def readDockerComposeFile(dockerComposeFile: Path) -> str:
|
||
|
|
with open(dockerComposeFile, "r") as file:
|
||
|
|
data = file.read()
|
||
|
|
return data
|
||
|
|
|
||
|
|
def getTraefikLabelsFromDockerComposeFile(dockerComposeFile: str):
|
||
|
|
traefikLabels = []
|
||
|
|
# Find all traefik labels
|
||
|
|
d = readDockerComposeFile(dockerComposeFile)
|
||
|
|
lines = d.split("\n")
|
||
|
|
for line in lines:
|
||
|
|
if "traefik." in line:
|
||
|
|
traefikLabels.append(line)
|
||
|
|
return traefikLabels
|
||
|
|
|
||
|
|
def parseTraefikLabels(traefikLabels: list):
|
||
|
|
s = {
|
||
|
|
"routers": [],
|
||
|
|
"services": [],
|
||
|
|
}
|
||
|
|
for label in traefikLabels:
|
||
|
|
if "traefik.http.routers" in label:
|
||
|
|
routerName = label.split(".")[3]
|
||
|
|
s["routers"].append(routerName)
|
||
|
|
elif "traefik.http.services" in label:
|
||
|
|
serviceName = label.split(".")[3]
|
||
|
|
s["services"].append(serviceName)
|
||
|
|
# Make unique
|
||
|
|
s["routers"] = list(set(s["routers"]))
|
||
|
|
s["services"] = list(set(s["services"]))
|
||
|
|
return s
|
||
|
|
|
||
|
|
def checkLabelUnique(labels: dict) -> bool:
|
||
|
|
routerNames = []
|
||
|
|
wrong = False
|
||
|
|
for project in labels:
|
||
|
|
for router in labels[project]["routers"]:
|
||
|
|
if router in routerNames:
|
||
|
|
if not wrong:
|
||
|
|
print("\n\n Seems like some routers are not unique 🐖\n")
|
||
|
|
print(f"🔴 ERROR : Router *{router}* is not unique")
|
||
|
|
wrong = True
|
||
|
|
for project2 in labels:
|
||
|
|
if router in labels[project2]["routers"]:
|
||
|
|
print(f"\t{project} and {project2}")
|
||
|
|
routerNames.append(router)
|
||
|
|
return wrong
|
||
|
|
def beautifulDisplayLabels(labels: dict, projectName: str):
|
||
|
|
print(f"\n======\n🧰 Project: {projectName}")
|
||
|
|
print("🎛️ Routers:")
|
||
|
|
for router in labels["routers"]:
|
||
|
|
print(f"\t{router}")
|
||
|
|
if len(labels["services"]) != 0:
|
||
|
|
print("🎛️ Services:")
|
||
|
|
for service in labels["services"]:
|
||
|
|
print(f"\t{service}")
|
||
|
|
|
||
|
|
def main(watchFolder: str):
|
||
|
|
# For each folder, read docker-compose.yaml and look for traefik labels
|
||
|
|
dockerComposeList = recursiveSearchForDockerComposeFiles(watchFolder)
|
||
|
|
labels = {}
|
||
|
|
for dockerComposeFile in dockerComposeList:
|
||
|
|
traefikLabels = getTraefikLabelsFromDockerComposeFile(dockerComposeFile)
|
||
|
|
if len(traefikLabels) == 0:
|
||
|
|
#print(f"No traefik labels found in {dockerComposeFile}")
|
||
|
|
continue
|
||
|
|
parsedTraefikLabels = parseTraefikLabels(traefikLabels)
|
||
|
|
projectName = dockerComposeFile.parent.name
|
||
|
|
labels[projectName] = parsedTraefikLabels
|
||
|
|
beautifulDisplayLabels(parsedTraefikLabels, projectName)
|
||
|
|
# Check if all routers are unique
|
||
|
|
wrong = checkLabelUnique(labels)
|
||
|
|
|
||
|
|
if wrong:
|
||
|
|
exit(1)
|
||
|
|
else:
|
||
|
|
print("\n\n✅ All routers are unique\n\n")
|
||
|
|
exit(0)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
print("Traefik label watcher CI/CD")
|
||
|
|
# Read folder from env
|
||
|
|
watchFolder = os.getenv("WATCH_FOLDER")
|
||
|
|
if watchFolder == None:
|
||
|
|
watchFolder = "."
|
||
|
|
|
||
|
|
watchFolder = Path(watchFolder)
|
||
|
|
|
||
|
|
main(watchFolder)
|