mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
4bef9fc6cf
- Rename 'checksum' to 'digest'. - Sort the data in the correct order to reduce load on the client (client sorts it to the same order).
130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
#------------------------------------------------------------------------------
|
|
import os
|
|
import sys
|
|
import json
|
|
import hashlib
|
|
|
|
#------------------------------------------------------------------------------
|
|
# Compute the SHA-256 digest of a file
|
|
#------------------------------------------------------------------------------
|
|
def ComputeDigest(filePath, blockSize=65536):
|
|
sha256 = hashlib.sha256()
|
|
with open(filePath, "rb") as file:
|
|
for block in iter(lambda: file.read(blockSize), b""):
|
|
sha256.update(block)
|
|
|
|
digest = sha256.hexdigest()
|
|
print(f"*** computed digest for '{filePath}': {digest}")
|
|
|
|
return digest
|
|
|
|
#------------------------------------------------------------------------------
|
|
# Compute digests for all files in a directory
|
|
#------------------------------------------------------------------------------
|
|
def RecursiveComputeDigest(directoryPath, settings):
|
|
digests = {}
|
|
scriptPath = os.path.abspath(__file__)
|
|
|
|
for root, _, files in os.walk(directoryPath):
|
|
for fileName in files:
|
|
filePath = os.path.join(root, fileName)
|
|
relativePath = os.path.relpath(filePath, directoryPath)
|
|
normalizedPath = relativePath.replace("\\", "/")
|
|
|
|
# Exclude the script itself from the manifest
|
|
if os.path.abspath(filePath) == scriptPath:
|
|
continue
|
|
|
|
digest = ComputeDigest(filePath)
|
|
digests[normalizedPath] = {
|
|
"digest": digest,
|
|
"restart": False
|
|
}
|
|
|
|
# Check if the file should be updated/installed during a restart
|
|
if settings and "restart" in settings:
|
|
restart_files = settings["restart"]
|
|
for file in restart_files:
|
|
if file in digests:
|
|
digests[file]["restart"] = True
|
|
|
|
return digests
|
|
|
|
#------------------------------------------------------------------------------
|
|
# Gets the settings for this depot
|
|
#------------------------------------------------------------------------------
|
|
def GetDepotSettings(depotName):
|
|
settingsPath = os.path.join('devops', f'{depotName}.json')
|
|
print(f"{settingsPath}")
|
|
if not os.path.isfile(settingsPath):
|
|
settingsPath = os.path.join('devops', 'default.json')
|
|
if not os.path.isfile(settingsPath):
|
|
return None
|
|
|
|
f = open(settingsPath)
|
|
if f.closed:
|
|
print(f"{settingsPath} = closed")
|
|
return None
|
|
|
|
return json.load(f)
|
|
|
|
#------------------------------------------------------------------------------
|
|
# build the manifest file
|
|
#------------------------------------------------------------------------------
|
|
def CreateManifest(version, currentDirectory, outManifestFile):
|
|
depotList = [name for name in os.listdir(currentDirectory) if os.path.isdir(os.path.join(currentDirectory, name))]
|
|
|
|
manifest = {
|
|
"version": version,
|
|
"depot": {}
|
|
}
|
|
|
|
i = 0
|
|
for depot in depotList:
|
|
fileName = f"{depot}.zip"
|
|
|
|
# Each depot must have a release package !!!
|
|
zipFilePath = os.path.join(currentDirectory, 'release', fileName)
|
|
|
|
print(f"{zipFilePath}")
|
|
if not os.path.isfile(zipFilePath):
|
|
continue
|
|
|
|
# Each depot must have a configuration !!!
|
|
settings = GetDepotSettings(depot)
|
|
if not settings:
|
|
continue
|
|
|
|
print(f"*** processing depot '{depot}'...")
|
|
|
|
digest = ComputeDigest(zipFilePath)
|
|
size = os.path.getsize(zipFilePath)
|
|
|
|
depotData = {
|
|
"assets": RecursiveComputeDigest(os.path.join(currentDirectory, depot), settings),
|
|
"digest": digest,
|
|
"name": fileName,
|
|
"optional": False,
|
|
"size": size,
|
|
"vendor": settings["vendor"],
|
|
}
|
|
|
|
manifest["depot"][i] = depotData
|
|
i+=1
|
|
|
|
with open(outManifestFile, "w") as jsonFile:
|
|
json.dump(manifest, jsonFile, indent=4)
|
|
|
|
#------------------------------------------------------------------------------
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 2:
|
|
print(f"Usage: bld_man.py <version>")
|
|
sys.exit(1)
|
|
|
|
version = sys.argv[1]
|
|
|
|
currentDirectory = os.path.dirname(os.path.abspath(__file__))
|
|
outManifestFile = "manifest_patch.json"
|
|
|
|
CreateManifest(version, currentDirectory, outManifestFile)
|