import os, glob

from ..streams import EndianBinaryReader

from ..files import File

def get_resource_data(*args):

"""

Input:

Option 1:

0 - path - file path

1 - assets_file - SerializedFile

2 - offset -

3 - size -

Option 2:

0 - reader - EndianBinaryReader

1 - offset -

2 - size -

-> -2 = offset, -1 = size

"""

if len(args) == 4:

reader = search_resource(res_path=args[0], assets_file=args[1])

elif len(args) == 3:

reader = args[0]

else:

raise TypeError(f"3 or 4 arguments required, but only {len(args)} given")

reader.Position = args[-2]

return reader.read_bytes(args[-1])

def search_resource(res_path, assets_file):

# try to find the resource in the Unity packages

base_name = os.path.basename(res_path)

if os.path.splitext(base_name)[1] == ".resource":

base_name2 = base_name.replace('.resource', '.assets.resS')

else:

base_name2 = base_name.replace('.assets.resS', '.resource')

for p in [res_path, base_name, base_name2]:

reader = assets_file.parent.files.get(p)

if reader:

if isinstance(reader, File.File):

# in case the import helper accidentally detected a resource file as something else

reader = reader.reader

return reader

# try to find it in the dir environment

c = assets_file

path = getattr(assets_file, "path", None)

while not path:

c = getattr(c,"parent",None)

if c == None:

raise FileNotFoundError(

f"Can't find the resource file {res_path}"

)

path = getattr(c, "path", None)

current_directory = path

resource_file_path = os.path.join(current_directory, *res_path.split("/"))

if not os.path.isfile(resource_file_path):

resource_file_path = search_resource_file(current_directory, base_name)

if not os.path.isfile(resource_file_path):

resource_file_path = search_resource_file(current_directory, base_name.replace('.assets.resS', '.resource'))

if os.path.isfile(resource_file_path):

return EndianBinaryReader(open(resource_file_path, "rb"))

else:

raise FileNotFoundError(

f"Can't find the resource file {res_path}"

)

def search_resource_file(path, name):

#print("real file", path, name)

files = glob.glob(os.path.join(path, "**", name), recursive=True)

return files[0] if len(files) else ""