AddCSLuaFile()

if CLIENT then
	SWEP.DrawAmmo = false
	SWEP.DrawCrosshair = false
end

DEFINE_BASECLASS("weapon_cs_base2")

SWEP.Author = "DarkRP Developers"
SWEP.Contact = ""
SWEP.Purpose = ""
SWEP.IconLetter = ""

SWEP.ViewModelFOV = 62
SWEP.ViewModelFlip = false
SWEP.AnimPrefix = "stunstick"

SWEP.AdminOnly = true

-- SWEP.StickColor = Color(255, 255, 255)

SWEP.ViewModel = Model("models/weapons/v_stunbaton.mdl")
SWEP.WorldModel = Model("models/weapons/w_stunbaton.mdl")

SWEP.Sound = Sound("weapons/stunstick/stunstick_swing1.wav")

SWEP.Primary.ClipSize = -1
SWEP.Primary.DefaultClip = 0
SWEP.Primary.Automatic = false
SWEP.Primary.Ammo = ""

SWEP.Secondary.ClipSize = -1
SWEP.Secondary.DefaultClip = 0
SWEP.Secondary.Automatic = false
SWEP.Secondary.Ammo = ""

function SWEP:Initialize()
	self:SetHoldType("normal")

	if SERVER then return end

	if self.StickColor then
		CreateMaterial("darkrp/" .. self:GetClass(), "VertexLitGeneric", {
			["$basetexture"] = "models/debug/debugwhite",
			["$surfaceprop"] = "metal",
			["$envmap"] = "env_cubemap",
			["$envmaptint"] = "[ .5 .5 .5 ]",
			["$selfillum"] = 0,
			["$model"] = 1
		}):SetVector("$color2", self.StickColor:ToVector())
	end
end

function SWEP:SetupDataTables()
	BaseClass.SetupDataTables(self)
	-- Bool 0 = Ironsights
	-- Bool 1 = Reloading
	self:NetworkVar("Bool", 2, "SeqIdling")
	-- Float 0 = LastPrimaryAttack
	-- Float 1 = ReloadEndTime
	-- Float 2 = BurstTime
	-- Float 3 = LastNonBurst
	self:NetworkVar("Float", 4, "SeqIdleTime")
	self:NetworkVar("Float", 5, "HoldTypeChangeTime")
end

function SWEP:Deploy()
	BaseClass.Deploy(self)
	if SERVER and self.StickColor then
		self:SetMaterial("!darkrp/" .. self:GetClass())
	end

	local vm = self:GetOwner():GetViewModel()
	if not IsValid(vm) then return true end
	self:PreDrawViewModel()
	vm:SendViewModelMatchingSequence(vm:LookupSequence("idle01"))
	return true
end

function SWEP:PreDrawViewModel(vm)
	if not IsValid(vm) then return end
	if self.StickColor then
		for i = 9, 15 do
			vm:SetSubMaterial(i, "!darkrp/" .. self:GetClass())
		end
	end
end

function SWEP:ViewModelDrawn(vm)
	if not IsValid(vm) then return end
	vm:SetSubMaterial() -- clear sub-materials
end

function SWEP:ResetStick(force)
	if game.SinglePlayer() then force = true end
	if not IsValid(self:GetOwner()) or (not force and (not IsValid(self:GetOwner():GetActiveWeapon()) or self:GetOwner():GetActiveWeapon():GetClass() ~= self:GetClass())) then return end
	if SERVER then
		self:SetMaterial() -- clear material
	end
	self:SetSeqIdling(false)
	self:SetSeqIdleTime(0)
	self:SetHoldTypeChangeTime(0)
end

function SWEP:Holster()
	BaseClass.Holster(self)
	self:ResetStick(false)
	return true
end

function SWEP:OnRemove()
	BaseClass.OnRemove(self)
	self:ResetStick(true)
end

function SWEP:Think()
	if self:GetSeqIdling() then
		self:SetSeqIdling(false)

		if not IsValid(self:GetOwner()) then return end
		self:GetOwner():SetAnimation(PLAYER_ATTACK1)
		self:EmitSound(self.Sound)

		local vm = self:GetOwner():GetViewModel()
		if not IsValid(vm) then return end
		vm:SendViewModelMatchingSequence(vm:LookupSequence("attackch"))
		vm:SetPlaybackRate(1 + 1 / 3)
		local duration = vm:SequenceDuration() / vm:GetPlaybackRate()
		local time = CurTime() + duration
		self:SetSeqIdleTime(time)
		self:SetNextPrimaryFire(time)
	end
	if self:GetSeqIdleTime() ~= 0 and CurTime() >= self:GetSeqIdleTime() then
		self:SetSeqIdleTime(0)

		if not IsValid(self:GetOwner()) then return end
		local vm = self:GetOwner():GetViewModel()
		if not IsValid(vm) then return end
		vm:SendViewModelMatchingSequence(vm:LookupSequence("idle01"))
	end
	if self:GetHoldTypeChangeTime() ~= 0 and CurTime() >= self:GetHoldTypeChangeTime() then
		self:SetHoldTypeChangeTime(0)
		self:SetHoldType("normal")
	end
end

function SWEP:PrimaryAttack()
	self:SetHoldType("melee")
	self:SetHoldTypeChangeTime(CurTime() + 0.3)

	self:SetNextPrimaryFire(CurTime() + 0.51) -- Actual delay is set later.

	local vm = self:GetOwner():GetViewModel()
	if IsValid(vm) then
		vm:SendViewModelMatchingSequence(vm:LookupSequence("idle01"))
		self:SetSeqIdling(true)
	end
end

function SWEP:SecondaryAttack()
	-- Do nothing
end

function SWEP:Reload()
	-- Do nothing
end