From 01bbf00d2c838e8eb7145d15d0bb90fc1d6f6a63 Mon Sep 17 00:00:00 2001
From: ZomgPonies <lorden@zomgponies.com>
Date: Sun, 27 Apr 2014 18:20:59 -0400
Subject: [PATCH] Ability to upgrade powercells in energy weapons

---
 code/modules/projectiles/guns/energy.dm | 29 +++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/code/modules/projectiles/guns/energy.dm b/code/modules/projectiles/guns/energy.dm
index dfc41f945c8..a02eb3f921d 100644
--- a/code/modules/projectiles/guns/energy.dm
+++ b/code/modules/projectiles/guns/energy.dm
@@ -42,3 +42,32 @@
 			icon_state = "[modifystate][ratio]"
 		else
 			icon_state = "[initial(icon_state)][ratio]"
+
+
+	attackby(obj/item/weapon/W, mob/user)
+		if(istype(W, /obj/item/weapon/cell))
+			if(!power_supply)
+				user.drop_item()
+				W.loc = src
+				power_supply = W
+				user << "<span class='notice'>You install a cell in [src].</span>"
+				update_icon()
+			else
+				user << "<span class='notice'>[src] already has a cell.</span>"
+
+		else if(istype(W, /obj/item/weapon/screwdriver))
+			if(power_supply)
+				power_supply.updateicon()
+				power_supply.loc = get_turf(src.loc)
+				power_supply = null
+				user << "<span class='notice'>You remove the cell from the [src].</span>"
+				update_icon()
+				return
+			..()
+		return
+
+	examine()
+		set src in view(1)
+		..()
+		if(!power_supply)
+			usr <<"<span class='warning'>The weapon does not have a power source installed.</span>"
\ No newline at end of file