package d2asset import ( "strings" "github.com/OpenDiablo2/OpenDiablo2/d2common" "github.com/OpenDiablo2/OpenDiablo2/d2common/d2interface" "github.com/OpenDiablo2/OpenDiablo2/d2common/d2resource" "github.com/OpenDiablo2/OpenDiablo2/d2core/d2config" ) const ( fileBudget = 1024 * 1024 * 32 ) // Static checks to confirm struct conforms to interface var _ d2interface.ArchivedFileManager = &fileManager{} var _ d2interface.Cacher = &fileManager{} type fileManager struct { cache d2interface.Cache archiveManager d2interface.ArchiveManager config *d2config.Configuration } func createFileManager(config *d2config.Configuration, archiveManager d2interface.ArchiveManager) d2interface.ArchivedFileManager { return &fileManager{ d2common.CreateCache(fileBudget), archiveManager, config, } } // LoadFileStream loads a file as a stream automatically from an archive func (fm *fileManager) LoadFileStream(filePath string) (d2interface.ArchiveDataStream, error) { filePath = fm.fixupFilePath(filePath) archive, err := fm.archiveManager.LoadArchiveForFile(filePath) if err != nil { return nil, err } return archive.ReadFileStream(filePath) } // LoadFile loads a file automatically from a managed archive func (fm *fileManager) LoadFile(filePath string) ([]byte, error) { filePath = fm.fixupFilePath(filePath) if value, found := fm.cache.Retrieve(filePath); found { return value.([]byte), nil } archive, err := fm.archiveManager.LoadArchiveForFile(filePath) if err != nil { return nil, err } data, err := archive.ReadFile(filePath) if err != nil { return nil, err } if err := fm.cache.Insert(filePath, data, len(data)); err != nil { return nil, err } return data, nil } // FileExists checks if a file exists in an archive func (fm *fileManager) FileExists(filePath string) (bool, error) { filePath = fm.fixupFilePath(filePath) return fm.archiveManager.FileExistsInArchive(filePath) } func (fm *fileManager) ClearCache() { fm.cache.Clear() } func (fm *fileManager) GetCache() d2interface.Cache { return fm.cache } func (fm *fileManager) fixupFilePath(filePath string) string { filePath = fm.removeLocaleTokens(filePath) filePath = strings.ToLower(filePath) filePath = strings.ReplaceAll(filePath, `/`, "\\") filePath = strings.TrimPrefix(filePath, "\\") return filePath } func (fm *fileManager) removeLocaleTokens(filePath string) string { tableToken := d2resource.LanguageTableToken fontToken := d2resource.LanguageFontToken filePath = strings.ReplaceAll(filePath, tableToken, fm.config.Language) // fixme: not all languages==latin filePath = strings.ReplaceAll(filePath, fontToken, "latin") return filePath }