129 lines
3.7 KiB
C++
129 lines
3.7 KiB
C++
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "MCPHandler.h"
|
|
#include "MCPAssetFinder.h"
|
|
#include "MCPUtils.h"
|
|
#include "Misc/PackageName.h"
|
|
#include "AssetRegistry/AssetRegistryModule.h"
|
|
#include "AssetRegistry/IAssetRegistry.h"
|
|
#include "HAL/FileManager.h"
|
|
#include "UObject/LinkerLoad.h"
|
|
#include "Asset_Delete.generated.h"
|
|
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
|
|
UCLASS()
|
|
class UMCP_Asset_Delete : public UObject, public IMCPHandler
|
|
{
|
|
GENERATED_BODY()
|
|
|
|
public:
|
|
UPROPERTY(meta=(Description="Package path of the asset to delete, e.g. /Game/Foo/Bar"))
|
|
FString AssetPath;
|
|
|
|
UPROPERTY(meta=(Optional, Description="If true, skip reference check and force delete"))
|
|
bool Force = false;
|
|
|
|
virtual FString GetDescription() const override
|
|
{
|
|
return TEXT("Delete a .uasset after verifying no references. "
|
|
"Use force=true to skip the reference check.");
|
|
}
|
|
|
|
virtual void Handle(FStringBuilderBase& Result) override
|
|
{
|
|
// Verify the asset file exists on disk
|
|
FString PackageFilename = FPackageName::LongPackageNameToFilename(
|
|
AssetPath, FPackageName::GetAssetPackageExtension());
|
|
PackageFilename = FPaths::ConvertRelativePathToFull(PackageFilename);
|
|
|
|
if (!IFileManager::Get().FileExists(*PackageFilename))
|
|
{
|
|
Result.Appendf(TEXT("ERROR: Asset file not found on disk: %s\n"), *PackageFilename);
|
|
return;
|
|
}
|
|
|
|
// Check references
|
|
IAssetRegistry& Registry = *IAssetRegistry::Get();
|
|
TArray<FName> Referencers;
|
|
Registry.GetReferencers(FName(*AssetPath), Referencers);
|
|
|
|
// Filter out self-references
|
|
Referencers.RemoveAll([this](const FName& Ref) {
|
|
return Ref.ToString() == AssetPath;
|
|
});
|
|
|
|
if (Referencers.Num() > 0 && !Force)
|
|
{
|
|
Result.Appendf(TEXT("ERROR: Asset is still referenced by %d package(s):\n"), Referencers.Num());
|
|
for (const FName& Ref : Referencers)
|
|
{
|
|
FString RefStr = Ref.ToString();
|
|
UPackage* RefPackage = FindPackage(nullptr, *RefStr);
|
|
Result.Appendf(TEXT(" %s%s\n"), *RefStr,
|
|
RefPackage ? TEXT(" (loaded)") : TEXT(" (on-disk only)"));
|
|
}
|
|
Result.Append(TEXT("Use force=true to skip the reference check.\n"));
|
|
return;
|
|
}
|
|
|
|
// Force delete: unload the package from memory first
|
|
if (Force && Referencers.Num() > 0)
|
|
{
|
|
Result.Appendf(TEXT("WARNING: Force-deleting despite %d referencer(s).\n"), Referencers.Num());
|
|
}
|
|
|
|
if (Force)
|
|
{
|
|
UPackage* Package = FindPackage(nullptr, *AssetPath);
|
|
if (Package)
|
|
{
|
|
// Collect all objects in this package
|
|
TArray<UObject*> ObjectsInPackage;
|
|
GetObjectsWithPackage(Package, ObjectsInPackage);
|
|
|
|
// Clear flags and remove from root to allow GC
|
|
for (UObject* Obj : ObjectsInPackage)
|
|
{
|
|
if (Obj)
|
|
{
|
|
Obj->ClearFlags(RF_Standalone | RF_Public);
|
|
Obj->RemoveFromRoot();
|
|
}
|
|
}
|
|
Package->ClearFlags(RF_Standalone | RF_Public);
|
|
Package->RemoveFromRoot();
|
|
|
|
// Reset loaders to release file handles
|
|
ResetLoaders(Package);
|
|
// Force garbage collection to free the objects
|
|
CollectGarbage(GARBAGE_COLLECTION_KEEPFLAGS);
|
|
}
|
|
}
|
|
|
|
// Delete the file on disk
|
|
bool bDeleted = IFileManager::Get().Delete(*PackageFilename, false, true);
|
|
|
|
if (!bDeleted)
|
|
{
|
|
Result.Appendf(TEXT("ERROR: Failed to delete file from disk: %s\n"), *PackageFilename);
|
|
return;
|
|
}
|
|
|
|
// Trigger an asset registry rescan so it notices the deletion
|
|
FString PackageDir;
|
|
int32 LastSlash;
|
|
if (AssetPath.FindLastChar(TEXT('/'), LastSlash))
|
|
{
|
|
PackageDir = AssetPath.Left(LastSlash);
|
|
Registry.ScanPathsSynchronous({PackageDir}, true);
|
|
}
|
|
|
|
Result.Appendf(TEXT("Deleted %s\n"), *AssetPath);
|
|
}
|
|
};
|