Grasscutter/src/main/java/emu/grasscutter/game/home/HomeBlockItem.java

package emu.grasscutter.game.home;

import dev.morphia.annotations.*;
import emu.grasscutter.data.binout.HomeworldDefaultSaveData;
import emu.grasscutter.net.proto.HomeBlockArrangementInfoOuterClass.HomeBlockArrangementInfo;
import java.util.*;
import java.util.stream.Stream;
import lombok.*;
import lombok.experimental.FieldDefaults;

@Entity
@Data
@Builder(builderMethodName = "of")
@FieldDefaults(level = AccessLevel.PRIVATE)
public class HomeBlockItem {
    @Id int blockId;
    boolean unlocked;
    List<HomeFurnitureItem> deployFurnitureList;
    List<HomeFurnitureItem> persistentFurnitureList;
    List<HomeAnimalItem> deployAnimalList;
    List<HomeNPCItem> deployNPCList;

    public static HomeBlockItem parseFrom(HomeworldDefaultSaveData.HomeBlock homeBlock) {
        // create from default setting
        return HomeBlockItem.of()
                .blockId(homeBlock.getBlockId())
                .unlocked(homeBlock.getFurnitures() != null)
                .deployFurnitureList(
                        homeBlock.getFurnitures() == null
                                ? List.of()
                                : homeBlock.getFurnitures().stream().map(HomeFurnitureItem::parseFrom).toList())
                .persistentFurnitureList(
                        homeBlock.getPersistentFurnitures() == null
                                ? List.of()
                                : homeBlock.getPersistentFurnitures().stream()
                                        .map(HomeFurnitureItem::parseFrom)
                                        .toList())
                .deployAnimalList(List.of())
                .deployNPCList(List.of())
                .build();
    }

    public void update(HomeBlockArrangementInfo homeBlockArrangementInfo) {
        this.blockId = homeBlockArrangementInfo.getBlockId();

        this.deployFurnitureList =
                homeBlockArrangementInfo.getDeployFurniureListList().stream()
                        .map(HomeFurnitureItem::parseFrom)
                        .toList();

        this.persistentFurnitureList =
                homeBlockArrangementInfo.getPersistentFurnitureListList().stream()
                        .map(HomeFurnitureItem::parseFrom)
                        .toList();

        this.deployAnimalList =
                homeBlockArrangementInfo.getDeployAnimalListList().stream()
                        .map(HomeAnimalItem::parseFrom)
                        .toList();

        this.deployNPCList =
                homeBlockArrangementInfo.getDeployNpcListList().stream()
                        .map(HomeNPCItem::parseFrom)
                        .toList();
    }

    public int calComfort() {
        return this.deployFurnitureList.stream().mapToInt(HomeFurnitureItem::getComfort).sum();
    }

    public HomeBlockArrangementInfo toProto() {
        var proto =
                HomeBlockArrangementInfo.newBuilder()
                        .setBlockId(blockId)
                        .setIsUnlocked(unlocked)
                        .setComfortValue(calComfort());

        this.reassignIfNull();

        this.deployFurnitureList.forEach(f -> proto.addDeployFurniureList(f.toProto()));
        this.persistentFurnitureList.forEach(f -> proto.addPersistentFurnitureList(f.toProto()));
        this.deployAnimalList.forEach(f -> proto.addDeployAnimalList(f.toProto()));
        this.deployNPCList.forEach(f -> proto.addDeployNpcList(f.toProto()));

        return proto.build();
    }

    // TODO add more types (farm field and suite)
    public List<? extends HomeMarkPointProtoFactory> getMarkPointProtoFactories() {
        this.reassignIfNull();

        return Stream.of(this.deployFurnitureList, this.persistentFurnitureList, this.deployNPCList)
                .flatMap(Collection::stream)
                .toList();
    }

    public void reassignIfNull() {
        if (this.deployFurnitureList == null) {
            this.deployFurnitureList = List.of();
        }
        if (this.persistentFurnitureList == null) {
            this.persistentFurnitureList = List.of();
        }
        if (this.deployAnimalList == null) {
            this.deployAnimalList = List.of();
        }
        if (this.deployNPCList == null) {
            this.deployNPCList = List.of();
        }
    }
}