80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
import os
|
|
import yaml
|
|
from collections import OrderedDict
|
|
from flask import Flask, jsonify, send_from_directory, request, redirect, url_for, session, render_template
|
|
from werkzeug.security import check_password_hash
|
|
import database
|
|
|
|
# --- Path Configurations ---
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
FRONTEND_DIR = os.path.join(BASE_DIR, '..', 'frontend')
|
|
|
|
YML_PATH = os.path.join(BASE_DIR, '..', 'mxpic', 'PDKs', 'Silterra', 'directories.yaml')
|
|
COMPS_ROOT = os.path.join(BASE_DIR, '..', 'mxpic', 'PDKs', 'Silterra')
|
|
# Define where your new icons folder is located (adjust if it's placed elsewhere)
|
|
ICONS_DIR = os.path.join(BASE_DIR, 'icons')
|
|
|
|
app = Flask(__name__, template_folder=FRONTEND_DIR, static_folder=FRONTEND_DIR)
|
|
app.secret_key = 'super_secret_mxpic_key'
|
|
app.json.sort_keys = False
|
|
|
|
database.init_db()
|
|
|
|
# ... [Keep countSpaces and buildTree exactly as they are] ...
|
|
|
|
def findComps(baseDir):
|
|
"""Scan component folders, return map of paths -> component info."""
|
|
compMap = {}
|
|
refDir = baseDir
|
|
for root, dirs, files in os.walk(baseDir):
|
|
ymlFiles = [f for f in files if f.endswith('.yml')]
|
|
if ymlFiles:
|
|
parentDir = os.path.dirname(root)
|
|
relPath = os.path.relpath(parentDir, refDir)
|
|
parts = () if relPath == '.' else tuple(relPath.split(os.sep))
|
|
compName = os.path.basename(root)
|
|
|
|
# Extract the category (the mother folder's name)
|
|
category = os.path.basename(parentDir)
|
|
|
|
compMap[parts] = {
|
|
'folder': compName,
|
|
'yml': ymlFiles[0],
|
|
'category': category # Save the category to the map
|
|
}
|
|
dirs.clear()
|
|
return compMap
|
|
|
|
def addCompsToTree(compMap):
|
|
"""Build a completely fresh tree from scratch and insert component nodes."""
|
|
fresh_tree = OrderedDict()
|
|
|
|
for pathSeg, compItem in compMap.items():
|
|
compName = compItem['folder']
|
|
curNode = fresh_tree
|
|
|
|
for seg in pathSeg:
|
|
if seg not in curNode:
|
|
curNode[seg] = OrderedDict()
|
|
curNode = curNode[seg]
|
|
|
|
curNode[compName] = OrderedDict({
|
|
"__type__": "component",
|
|
"__name__": compName,
|
|
"__yml__": compItem['yml'],
|
|
"__category__": compItem['category'] # Inject category into the tree
|
|
})
|
|
|
|
return fresh_tree
|
|
|
|
|
|
if os.path.isdir(COMPS_ROOT):
|
|
compMap = findComps(COMPS_ROOT)
|
|
fresh_tree = addCompsToTree(compMap)
|
|
|
|
print(compMap)
|
|
print(fresh_tree)
|
|
|
|
|
|
|