Module:File link

From annadreambrush.com/wiki
Revision as of 02:49, 30 May 2014 by imported>Mr. Stradivarius (add type checks and write the render function; read-only code not working yet)
Jump to navigation Jump to search

Documentation for this module may be created at Module:File link/doc

-- This module provides a library for formatting image wikilinks.

local libraryUtil = require('libraryUtil')

local image = {}

function image.new()
	local obj, data = {}, {}
	
	local checkSelf = libraryUtil.makeCheckSelfFunction('image', 'image', obj, 'image object')
	local checkType = libraryUtil.checkType
	
	function data:name(s)
		checkSelf(self, 'name')
		checkType('name', 1, s, 'string')
		self.theName = s
	end
	
	function data:format(s, filename)
		checkSelf(self, 'format')
		checkType('format', 1, s, 'string')
		checkType('format', 2, format, 'string', true)
		local validFormats = {
			thumb = true,
			thumbnail = true,
			frame = true,
			framed = true,
			frameless = true
		}
		if validFormats[s] then
			self.theFormat = s
			self.theFormatFilename = filename
		else
			error('invalid format')
		end
	end
	
	function data:width(px)
		checkSelf(self, 'width')
		checkType('width', 1, px, 'number')
		self.theWidth = px
	end
	
	function data:height(px)
		checkSelf(self, 'height')
		checkType('height', 1, px, 'number')
		self.theHeight = px
	end
	
	function data:upright(factor)
		checkSelf(self, 'upright')
		checkType('upright', 1, factor, 'number', true)
		self.isUpright = true
		self.uprightFactor = factor
	end
	
	function data:resetSize()
		checkSelf(self, 'resetSize')
		for i, field in ipairs{'theWidth', 'theHeight', 'isUpright', 'uprightFactor'} do
			self[field] = nil
		end
	end
	
	function data:location(s)
		checkSelf(self, 'location')
		checkType('location', 1, s, 'string')
		local validLocations = {
			right = true,
			left = true,
			center = true,
			none = true
		}
		if validLocations[s] then
			self.theLocation = s
		else
			error(string.format(
				"bad argument #1 to 'image:location' ('%s' is not a valid location)",
				s
			))
		end
	end
	
	function data:alignment(s)
		checkSelf(self, 'alignment')
		checkType('alignment', 1, s, 'string')
		local validAlignments = {
			baseline = true,
			middle = true,
			sub = true,
			super = true,
			['text-top'] = true,
			['text-bottom'] = true,
			top = true,
			bottom = true
		}
		if validAlignments[s] then
			self.theAlignment = s
		else
			error(string.format(
				"bad argument #1 to 'image:alignment' ('%s' is not a valid alignment)"
			))
		end
	end
	
	function data:border()
		checkSelf(self, 'border')
		self.hasBorder = true
	end
	
	function data:link(s)
		checkSelf(self, 'link')
		checkType('link', 1, s, 'string')
		self.theLink = s
	end
	
	function data:alt(s)
		checkSelf(self, 'alt')
		checkType('alt', 1, s, 'string')
		self.theAlt = s
	end
	
	function data:caption(s)
		checkSelf(self, 'caption')
		checkType('caption', 1, s, 'string')
		self.theCaption = s
	end
	
	function data:render()
		checkSelf(self, 'render')
		
		local ret = {}
		
		-- Image name.
		if not self.theName then
			error('image:render: no image name was found')
		end
		ret[#ret + 1] = 'File:' .. self.theName
		
		-- Image format
		if self.theFormat and self.theFormatFilename then
			ret[#ret + 1] = self.theFormat .. '=' .. self.theFormatFilename
		elseif self.theFormat then
			ret[#ret + 1] = self.theFormat
		end
		
		-- Border
		if self.hasBorder then
			ret[#ret + 1] = 'border'
		end
		
		-- Location
		ret[#ret + 1] = self.theLocation

		-- Alignment
		ret[#ret + 1] = self.theAlignment
		
		-- Size
		if self.isUpright and (self.theWidth or self.theHeight) then
			error("duplicate size value detected in 'render' (height/width cannot be used at the same time as 'upright')")
		elseif self.isUpright and self.uprightFactor then
			ret[#ret + 1] = 'upright=' .. tostring(self.uprightFactor)
		elseif self.isUpright then
			ret[#ret + 1] = 'upright'
		elseif self.theWidth and self.theHeight then
			ret[#ret + 1] = string.format('%dx%dpx', self.theWidth, self.theHeight)
		elseif self.theWidth then
			ret[#ret + 1] = tostring(self.theWidth) .. 'px'
		elseif self.theHeight then
			ret[#ret + 1] = string.format('x%dpx', self.theHeight)
		end
		
		-- Link
		if self.theLink then
			ret[#ret + 1] = 'link=' .. self.theLink
		end
		
		-- Alt
		if self.theAlt then
			ret[#ret + 1] = 'alt=' .. self.theAlt
		end
		
		-- Caption
		ret[#ret + 1] = self.theCaption
		
		return string.format('[[%s]]', table.concat(ret, '|'))
	end
	
	local readOnlyFields = {
		theName = true,
		theFormat = true,
		theFormatFilename = true,
		theWidth = true,
		theHeight = true,
		isUpright = true,
		uprightFactor = true,
		theLocation = true,
		theAlignment = true,
		hasBorder = true,
		theLink = true,
		theAlt = true,
		theCaption = true
	}
	for field in pairs(data) do
		readOnlyFields[field] = true
	end
	
	setmetatable(obj, {
		__index = data,
		__newindex = function (t, key, value)
			if readOnlyFields[key] then
				error(string.format(
					"field '%s' is read-only",
					tostring(key)
				), 2)
			else
				data[key] = value
			end
		end,
		__tostring = function (t)
			return t:render()
		end
	})
	
	return obj
end

-- return image

local p = {}

function p.test()
	local myImage = image.new()
	myImage:name('Foo')
	return myImage:render()
end

return p