mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-12-24 19:53:08 +00:00
57312961e8
* Renamed assets->resources; enforcing Manifest build if it does not exist * Rebuild resources from CI * Added Manifest to repo - be sure to rebuild it with `make -C assets` before committing changes! * Actually added Manifest. * Keeping Manifest on assets clean * Spelling fix in Makefile
94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
from enum import Enum
|
|
from collections import OrderedDict
|
|
|
|
|
|
class FsNode:
|
|
class NodeType(Enum):
|
|
File = 0
|
|
Directory = 1
|
|
|
|
def __init__(self, name: str, nodetype: "FsNode.Type", **kwargs):
|
|
self.name = name
|
|
self.nodetype = nodetype
|
|
self.data = kwargs
|
|
self.parent = None
|
|
self.children = OrderedDict()
|
|
|
|
def addChild(self, node: "FsNode"):
|
|
self.children[node.name] = node
|
|
node.parent = self
|
|
|
|
def addDirectory(self, path):
|
|
fragments = path.split("/")
|
|
name = fragments[-1]
|
|
fragments = fragments[:-1]
|
|
parent_node = self.traverse(fragments)
|
|
if not parent_node:
|
|
raise Exception(f"No parent node found for: {path}")
|
|
parent_node.addChild(FsNode(name, FsNode.NodeType.Directory))
|
|
|
|
def addFile(self, path, md5, size):
|
|
fragments = path.split("/")
|
|
name = fragments[-1]
|
|
fragments = fragments[:-1]
|
|
parent_node = self.traverse(fragments)
|
|
if not parent_node:
|
|
raise Exception(f"No parent node found for: {path}")
|
|
parent_node.addChild(FsNode(name, FsNode.NodeType.File, md5=md5, size=size))
|
|
|
|
def getChild(self, name):
|
|
return self.children[name]
|
|
|
|
def traverse(self, fragments):
|
|
current = self
|
|
for fragment in fragments:
|
|
current = current.getChild(fragment)
|
|
if not current:
|
|
break
|
|
return current
|
|
|
|
def getPath(self):
|
|
fragments = []
|
|
current = self
|
|
while current.parent:
|
|
fragments.append(current.name)
|
|
current = current.parent
|
|
return "/".join(reversed(fragments))
|
|
|
|
def dump(self):
|
|
ret = {}
|
|
ret["name"] = (self.name,)
|
|
ret["type"] = (self.nodetype,)
|
|
ret["path"] = (self.getPath(),)
|
|
if len(self.children):
|
|
ret["children"] = [node.dump() for node in self.children.values()]
|
|
return ret
|
|
|
|
|
|
def walk_nodes(node: FsNode):
|
|
yield node
|
|
for child in node.children.values():
|
|
yield from walk_nodes(child)
|
|
|
|
|
|
# Returns filenames: [only_in_left], [changed], [only_in_right]
|
|
def compare_fs_trees(left: FsNode, right: FsNode):
|
|
# import pprint
|
|
# pprint.pprint(left.dump())
|
|
# pprint.pprint(right.dump())
|
|
left_dict = dict((node.getPath(), node) for node in walk_nodes(left))
|
|
right_dict = dict((node.getPath(), node) for node in walk_nodes(right))
|
|
|
|
left_names = set(left_dict.keys())
|
|
right_names = set(right_dict.keys())
|
|
common_names = left_names.intersection(right_names)
|
|
|
|
return (
|
|
list(left_names - right_names),
|
|
list(
|
|
name
|
|
for name in common_names
|
|
if left_dict[name].data != right_dict[name].data
|
|
),
|
|
list(right_names - left_names),
|
|
)
|