From c4cfd02b1a3463c741a66630f44fb5d258a4d95f Mon Sep 17 00:00:00 2001
From: fullwall <fullwall25@gmail.com>
Date: Wed, 19 Dec 2012 22:26:46 +0800
Subject: [PATCH] Add Storage type plotters, update version to 2.0.6-SNAPSHOT

---
 pom.xml                                       |  4 ++--
 src/main/java/net/citizensnpcs/Citizens.java  |  1 +
 .../java/net/citizensnpcs/NPCDataStore.java   | 23 +++++++++++++++++++
 3 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/pom.xml b/pom.xml
index a376e5011..7e147c3e2 100644
--- a/pom.xml
+++ b/pom.xml
@@ -6,13 +6,13 @@
 
 	<groupId>net.citizensnpcs</groupId>
 	<artifactId>citizens</artifactId>
-	<version>2.0.5</version>
+	<version>2.0.6-SNAPSHOT</version>
 	<name>Citizens</name>
 
 	<properties>
 		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
 		<craftbukkit.version>LATEST</craftbukkit.version>
-		<citizensapi.version>2.0.5</citizensapi.version>
+		<citizensapi.version>2.0.6-SNAPSHOT</citizensapi.version>
 		<vault.version>1.2.19-SNAPSHOT</vault.version>
 		<powermock.version>1.4.12</powermock.version>
 		<build.number>Unknown</build.number>
diff --git a/src/main/java/net/citizensnpcs/Citizens.java b/src/main/java/net/citizensnpcs/Citizens.java
index 1a69c3703..523056e79 100644
--- a/src/main/java/net/citizensnpcs/Citizens.java
+++ b/src/main/java/net/citizensnpcs/Citizens.java
@@ -314,6 +314,7 @@ public class Citizens extends JavaPlugin implements CitizensPlugin {
             });
 
             traitFactory.addPlotters(metrics.createGraph("traits"));
+            saves.addPlotters(metrics.createGraph("Storage type"));
             metrics.start();
         } catch (IOException e) {
             Messaging.logTr(Messages.METRICS_ERROR_NOTIFICATION, e.getMessage());
diff --git a/src/main/java/net/citizensnpcs/NPCDataStore.java b/src/main/java/net/citizensnpcs/NPCDataStore.java
index 07001b4a4..103ddff43 100644
--- a/src/main/java/net/citizensnpcs/NPCDataStore.java
+++ b/src/main/java/net/citizensnpcs/NPCDataStore.java
@@ -3,6 +3,8 @@ package net.citizensnpcs;
 import java.io.File;
 import java.sql.SQLException;
 
+import net.citizensnpcs.Metrics.Graph;
+import net.citizensnpcs.Metrics.Plotter;
 import net.citizensnpcs.Settings.Setting;
 import net.citizensnpcs.api.npc.NPC;
 import net.citizensnpcs.api.npc.NPCRegistry;
@@ -96,4 +98,25 @@ public class NPCDataStore {
             return null;
         return new NPCDataStore(saves);
     }
+
+    public void addPlotters(Graph graph) {
+        graph.addPlotter(new Plotter("Database") {
+            @Override
+            public int getValue() {
+                return root instanceof DatabaseStorage ? 1 : 0;
+            }
+        });
+        graph.addPlotter(new Plotter("YAML") {
+            @Override
+            public int getValue() {
+                return root instanceof YamlStorage ? 1 : 0;
+            }
+        });
+        graph.addPlotter(new Plotter("NBT") {
+            @Override
+            public int getValue() {
+                return root instanceof NBTStorage ? 1 : 0;
+            }
+        });
+    }
 }