Difference between revisions of "Module:Dts"

Revision as of 06:33, 1 July 2015

local getArgs = require('Module:Arguments').getArgs

--------------------------------------------------------------------------------
-- Dts class
--------------------------------------------------------------------------------

local Dts = {}
Dts.__index = Dts

Dts.monthsSearch = { "jan", "feb", "mar", "apr", "may", "jun", "jul", "aug", "sep", "oct", "nov", "dec" }
Dts.monthsAbr = { "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec" }
Dts.months = { "January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December" }

function Dts.new(args)
	local self = setmetatable({}, Dts)
	self.fmt = "mdy"
	self.abbr = false --default
	if args[1] and (not args[2]) then
		for _, val in pairs(mw.text.split(args[1],"[%s/-]")) do
			self:annonval(val, true)
		end
	else
		for key, val in pairs(args) do
			if tonumber(key) then
				self:annonval(val)
			end
		end
	end
	if args.format then
		self.fmt = args.format
	end
	if args.abbr then
		if args.abbr == "on" then
			self.abbr=true
		else
			self.abbr=false
		end
	end
	if (self.year==0) then --not valid. placeholder for no-year
		self.year = nil
	end
	return self
end

function Dts:setmonth(raw)
	if not raw then
		self.month = nil
		return false
	end
	local numbermonth = tonumber(raw)
	if numbermonth and numbermonth > 0 and numbermonth < 13 then
		self.month = numbermonth
		return true
	end
	for i, mon in pairs(self.monthsSearch) do
		if string.find(string.lower(raw),mon) then
			self.month=i
			if string.find(string.lower(raw),string.lower(self.months[i])) then
				self.abbr=false
			else
				self.abbr=true
			end
			return true
		end
	end
	return false
end

function Dts:annonval(val, dayfirst)
	local numberval
	if val then
		numberval = tonumber(mw.text.trim(val,"%s%t,"))
	end
	if (not val) or (type(val)=="table") or (mw.text.trim(val)=="") then
		numberval = 0
	end
	if not numberval then
		if mw.text.trim(string.lower(val)) == "bc" then
			if (not self.year) then
				self.year = self.day
				self.day = nil
			end
			if self.year then
				self.year = 0 - self.year
			end
		else
			if self:setmonth(val,dayfirst) and dayfirst and self.year and (not self.day) and (self.year > 0) and (self.year<31) then
				self.day = self.year
				self.year = nil
				self.fmt = "dmy"
			end
		end
		return
	end
	if self.month and (not self.day) and (numberval < 32) and (numberval > 0) then
		self.day = numberval
		return
	end
	if self.year and (not self.month) then
		self.month = numberval
		return
	end
	if (not self.year) then
		self.year = numberval
		return
	end
end

function Dts:monthName()
	if (not self.month) or (self.month < 0) or (self.month > 12) then
		return ""
	end
	if self.abbr then
		return self.monthsAbr[self.month]
	else
		return self.months[self.month]
	end
end

function Dts:sortkey()
	local root = mw.html.create()

	-- Sort span
	local sortSpan = root:tag('span')
	sortSpan
		:addClass('sortkey')
		:css('display', 'none')
		:css('speak', 'none')
	local current = os.date("*t")
	local year = self.year or current.year
	year = year > 0 and year or -10000 - year
	sortSpan:wikitext(string.format(
		"%05d-%02d-%02d-%02d%02d",
		year,
		self.month or 1,
		self.day or 1,
		0,
		0
	))

	return root
end

function Dts:__tostring()
	local root = self:sortkey()
	local displaySpan = root:tag('span')
	displaySpan:css('white-space', 'nowrap')
	if self.day then
		if self.fmt == "mdy" then
			displaySpan:wikitext(self:monthName())
			displaySpan:wikitext(' ')
			displaySpan:wikitext(self.day)
			if self.year then
				displaySpan:wikitext(',')
			end
		else
			displaySpan:wikitext(self.day)
			displaySpan:wikitext(' ')
			displaySpan:wikitext(self:monthName())
		end
	else
		if self.month then
			displaySpan:wikitext(self:monthName())
		end
	end
	if self.year then
		if self.month then
			displaySpan:wikitext(' ')
		end
		displaySpan:wikitext(math.abs(self.year))
		if self.year < 0 then
			displaySpan:wikitext(' BC')
		end
	end

	return tostring(root)
end

--------------------------------------------------------------------------------
-- Exports
--------------------------------------------------------------------------------

local p = {}

function p._main(args)
	local dts = Dts.new(args)
	return tostring(dts)
end

function p.main(frame)
	local args = getArgs(frame, {
		wrappers = 'Template:Dts',
		removeBlanks = false
	})
	return p._main(args)
end

return p