Yangyangzhong82
125 lines
3.5 KiB
C++
125 lines
3.5 KiB
C++
#include "Cleaner.h"
|
|
#include "mc/world/actor/provider/SynchedActorDataAccess.h"
|
|
#include "gmlib/mc/world/actor/Actor.h"
|
|
namespace Cleaner {
|
|
|
|
bool isMatch(std::string& A, std::string& B) {
|
|
// 如果B是正则表达式
|
|
if (B.find_first_of(".*+?()[]{}|^$") != std::string::npos) {
|
|
try {
|
|
std::regex regex(B);
|
|
return std::regex_match(A, regex);
|
|
} catch (...) {
|
|
return false;
|
|
}
|
|
}
|
|
return (A == B);
|
|
}
|
|
|
|
bool isTrust(Actor* ac) { return SynchedActorDataAccess::getActorFlag(ac->getEntityContext(), ::ActorFlags::Trusting); }
|
|
|
|
bool shouldIgnore(gmlib::GMActor* ac) {
|
|
if (ac->hasCategory(::ActorCategory::Mob) || ac->hasCategory(::ActorCategory::Item)) {
|
|
if (ac->isTame() || isTrust(ac) || ac->getNameTag() != "" || ac->hasTag("cleaner:ignore")) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool ShouldClean(Actor* actor) {
|
|
// Players
|
|
auto& config = Cleaner::Entry::getInstance().getConfig();
|
|
auto en = (gmlib::GMActor*)actor;
|
|
if (en->isPlayer() || shouldIgnore(en)) {
|
|
return false;
|
|
}
|
|
auto type = en->getTypeName();
|
|
for (auto& tag : config.IgnoreTags) {
|
|
if (en->hasTag(tag)) {
|
|
return false;
|
|
}
|
|
}
|
|
// Items
|
|
if (en->hasCategory(::ActorCategory::Item)) {
|
|
if (config.CleanItem.Enabled) {
|
|
auto itac = (ItemActor*)en;
|
|
if (itac->age() <= config.CleanItem.ExistTicks) {
|
|
return false;
|
|
}
|
|
auto itemType = itac->item().getTypeName();
|
|
auto whitelist = config.CleanItem.Whitelist;
|
|
for (auto& key : whitelist) {
|
|
if (isMatch(itemType, key)) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
// Mobs
|
|
else if (en->hasCategory(::ActorCategory::Mob)) {
|
|
if (config.CleanMobs.Enabled) {
|
|
auto blacklist = config.CleanMobs.BlackList;
|
|
for (auto& key : blacklist) {
|
|
if (isMatch(type, key)) {
|
|
return true;
|
|
}
|
|
}
|
|
auto whitelist = config.CleanMobs.Whitelist;
|
|
for (auto& key : whitelist) {
|
|
if (isMatch(type, key)) {
|
|
return false;
|
|
}
|
|
}
|
|
if (config.CleanMobs.CleanMonstors && en->hasCategory(ActorCategory::Monster)) {
|
|
return true;
|
|
}
|
|
if (config.CleanMobs.CleanPeacefulMobs) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
// Others
|
|
else {
|
|
if (config.CleanInanimate.Enabled) {
|
|
auto blacklist = config.CleanInanimate.Blacklist;
|
|
for (auto& key : blacklist) {
|
|
if (isMatch(type, key)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
|
|
int ExecuteClean() {
|
|
auto level = ll::service::getLevel();
|
|
int clean_count = 0;
|
|
auto all_entities = level->getRuntimeActorList();
|
|
for (auto entity : all_entities) {
|
|
if (ShouldClean(entity)) {
|
|
entity->despawn();
|
|
clean_count++;
|
|
}
|
|
}
|
|
return clean_count;
|
|
}
|
|
|
|
int CountEntities() {
|
|
auto level = ll::service::getLevel();
|
|
int clean_count = 0;
|
|
auto all_entities = level->getRuntimeActorList();
|
|
for (auto* entity : all_entities) {
|
|
if (ShouldClean(entity)) {
|
|
clean_count++;
|
|
}
|
|
}
|
|
return clean_count;
|
|
}
|
|
|
|
} // namespace Cleaner
|