fancyholograms-v3: Save and load attached traits

Author: OliverSchlueter

plugins/fancyholograms/fh-api/src/main/java/de/oliver/fancyholograms/api/trait/HologramTraitTrait.java

@@ -4,15 +4,18 @@
 import de.oliver.fancyholograms.api.hologram.Hologram;
 import org.bukkit.entity.Player;
 
+import java.io.IOException;
 import java.util.ArrayList;
 import java.util.List;
 
 @HologramTraitClass(traitName = "trait_trait")
 public class HologramTraitTrait extends HologramTrait {
 
     private final List<HologramTrait> traits;
+    private Configuration configuration;
 
     public HologramTraitTrait(Hologram hologram) {
+        this.configuration = new Configuration(new ArrayList<>());
         this.traits = new ArrayList<>();
         attachHologram(hologram);
     }
@@ -24,6 +27,13 @@ public void addTrait(HologramTrait trait) {
 
         trait.attachHologram(hologram);
         this.traits.add(trait);
+        this.configuration.traits().add(trait.getName());
+        try {
+            storage.set(hologram.getData().getName(), configuration);
+        } catch (IOException e) {
+            logger.error("Failed to save configuration for HologramTraitTrait");
+            logger.error(e);
+        }
     }
 
     @Override
@@ -37,19 +47,53 @@ public void onAttach() {
 
             try {
                 HologramTrait trait = ti.clazz().getConstructor().newInstance();
-                if (!new HologramTraitAttachedEvent(hologram, trait, false).callEvent()) {
+                if (!new HologramTraitAttachedEvent(hologram, trait, true).callEvent()) {
                     continue;
                 }
 
                 trait.attachHologram(hologram);
                 this.traits.add(trait);
             } catch (Exception e) {
-                logger.error("Failed to instantiate trait " + ti.name());
+                logger.error("Failed to instantiate default trait " + ti.name());
                 logger.error(e);
             }
 
             logger.debug("Attached default trait " + ti.name() + " to hologram " + hologram.getData().getName());
         }
+
+        // Attach all traits that are already attached to the hologram
+        try {
+            configuration = storage.get(hologram.getData().getName(), Configuration.class);
+        } catch (IOException e) {
+            logger.error("Failed to load configuration for HologramTraitTrait");
+            logger.error(e);
+            return;
+        }
+        if (configuration == null) {
+            return;
+        }
+
+        for (String traitName : configuration.traits()) {
+            HologramTraitRegistry.TraitInfo traitInfo = api.getTraitRegistry().getTrait(traitName);
+            if (traitInfo == null) {
+                logger.warn("Trait " + traitName + " is not registered");
+                continue;
+            }
+
+            try {
+                HologramTrait trait = traitInfo.clazz().getConstructor().newInstance();
+
+                if (!new HologramTraitAttachedEvent(hologram, trait, false).callEvent()) {
+                    return;
+                }
+
+                trait.attachHologram(hologram);
+                this.traits.add(trait);
+            } catch (Exception e) {
+                logger.error("Failed to instantiate trait " + traitName);
+                logger.error(e);
+            }
+        }
     }
 
     @Override
@@ -93,4 +137,10 @@ public void save() {
             trait.save();
         }
     }
+
+    record Configuration(
+            List<String> traits
+    ) {
+
+    }
 }