net.minecraft.server.PersistentCollection Maven / Gradle / Ivy

Go to download
package net.minecraft.server;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

import java.io.*;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

public class PersistentCollection {

  public List c = Lists.newArrayList(); // Spigot
  protected Map a = Maps.newHashMap();
  private final IDataManager b;
  private final Map d = Maps.newHashMap();

  public PersistentCollection(IDataManager idatamanager) {
    this.b = idatamanager;
    this.b();
  }

  public PersistentBase get(Class oclass, String s) {
    PersistentBase persistentbase = this.a.get(s);

    if (persistentbase == null) {
      if (this.b != null) {
        try {
          File file = this.b.getDataFile(s);

          if (file != null && file.exists()) {
            try {
              persistentbase = oclass.getConstructor(new Class[]{String.class}).newInstance(new Object[]{s});
            } catch (Exception exception) {
              throw new RuntimeException("Failed to instantiate " + oclass.toString(), exception);
            }

            FileInputStream fileinputstream = new FileInputStream(file);
            NBTTagCompound nbttagcompound = NBTCompressedStreamTools.a(fileinputstream);

            fileinputstream.close();
            persistentbase.a(nbttagcompound.getCompound("data"));
          }
        } catch (Exception exception1) {
          exception1.printStackTrace();
        }
      }

      if (persistentbase != null) {
        this.a.put(s, persistentbase);
        this.c.add(persistentbase);
      }

    }
    return persistentbase;
  }

  public void a(String s, PersistentBase persistentbase) {
    if (this.a.containsKey(s)) {
      this.c.remove(this.a.remove(s));
    }

    this.a.put(s, persistentbase);
    this.c.add(persistentbase);
  }

  public void a() {
    for (int i = 0; i < this.c.size(); ++i) {
      PersistentBase persistentbase = this.c.get(i);

      if (persistentbase.d()) {
        this.a(persistentbase);
        persistentbase.a(false);
      }
    }

  }

  private void a(PersistentBase persistentbase) {
    if (this.b != null) {
      try {
        File file = this.b.getDataFile(persistentbase.id);

        if (file != null) {
          NBTTagCompound nbttagcompound = new NBTTagCompound();

          persistentbase.b(nbttagcompound);
          NBTTagCompound nbttagcompound1 = new NBTTagCompound();

          nbttagcompound1.set("data", nbttagcompound);
          FileOutputStream fileoutputstream = new FileOutputStream(file);

          NBTCompressedStreamTools.a(nbttagcompound1, fileoutputstream);
          fileoutputstream.close();
        }
      } catch (Exception exception) {
        exception.printStackTrace();
      }

    }
  }

  private void b() {
    try {
      this.d.clear();
      if (this.b == null) {
        return;
      }

      File file = this.b.getDataFile("idcounts");

      if (file != null && file.exists()) {
        DataInputStream datainputstream = new DataInputStream(new FileInputStream(file));
        NBTTagCompound nbttagcompound = NBTCompressedStreamTools.a(datainputstream);

        datainputstream.close();
        Iterator iterator = nbttagcompound.c().iterator();

        while (iterator.hasNext()) {
          String s = (String) iterator.next();
          NBTBase nbtbase = nbttagcompound.get(s);

          if (nbtbase instanceof NBTTagShort) {
            NBTTagShort nbttagshort = (NBTTagShort) nbtbase;
            short short0 = nbttagshort.e();

            this.d.put(s, Short.valueOf(short0));
          }
        }
      }
    } catch (Exception exception) {
      exception.printStackTrace();
    }

  }

  public int a(String s) {
    Short oshort = this.d.get(s);

    if (oshort == null) {
      oshort = Short.valueOf((short) 0);
    } else {
      oshort = Short.valueOf((short) (oshort.shortValue() + 1));
    }

    this.d.put(s, oshort);
    if (this.b == null) {
      return oshort.shortValue();
    } else {
      try {
        File file = this.b.getDataFile("idcounts");

        if (file != null) {
          NBTTagCompound nbttagcompound = new NBTTagCompound();
          Iterator iterator = this.d.keySet().iterator();

          while (iterator.hasNext()) {
            String s1 = (String) iterator.next();
            short short0 = this.d.get(s1).shortValue();

            nbttagcompound.setShort(s1, short0);
          }

          DataOutputStream dataoutputstream = new DataOutputStream(new FileOutputStream(file));

          NBTCompressedStreamTools.a(nbttagcompound, (DataOutput) dataoutputstream);
          dataoutputstream.close();
        }
      } catch (Exception exception) {
        exception.printStackTrace();
      }

      return oshort.shortValue();
    }
  }
}