Module:Library citation

local p = {}

require("Module:Auxiliary functions")

function getDataFromTitle(title)
	local rawString = mw.getCurrentFrame():callParserFunction("#dpl","namespace=Library","title=" .. title,"mode=userformat","include={Library work}:title:author:published_date:source")
	local title,author,published_date,source = rawString:match("^(.-)%s|(.-)%s|(.-)%s|(.-)$")
	local year = ""
	if is_set(published_date) then
		year = published_date:match("%d%d%d%d")
	end
	return title,author,year,source
end

function getDataFromLink(link)
	-- In case it's in URL format
	local str = mw.uri.decode(link)
    local result = str:match("https?://[^/]+/wiki/(.*)")
    if result then
        str = result  -- Use the extracted part after /wiki/
    end
    -- Remove "Library:" prefix, if present
    str = str:gsub("^Library:", "")

    -- Replace underscores with spaces
    str = str:gsub("_", " ")

    -- Return data using the other function
    return getDataFromTitle(str)
end

function wikilinkFromTitle(title)
	local pagename = mw.getCurrentFrame():callParserFunction("#dpl","namespace=Library","titleregexp=" .. title .. "$","format=,%TITLE%")
	if is_set(pagename) then
		return "[[Library:" .. pagename .. "|" .. title .. "]]"
	end
	return title
end

local function wikilinkFromAuthor(author)
	-- if page exists
	if mw.getCurrentFrame():callParserFunction('#ifexist',author,true) == "1" then
		return "[[" .. author .. "]]"
	end
	return author
end

function p.cite(frame)
	-- Parent frame to get parameters from template calls instead of from {{#invoke: ...
	local parentFrame = frame:getParent()
	-- get arguments from "args" table pre-defined by MediaWiki/Scribunto
	local link = parentFrame.args.link
	local author = nil
	local quote = parentFrame.args.quote
	if is_set(link) then
		title,author,year,source = getDataFromLink(link)
	end
	
	local render = ""
	
	if is_set(author) then
		render = render .. wikilinkFromAuthor(author)
		if not is_set(year) then
			render = render .. "."
		end
	end
	if is_set(year) then
		render = render .. " (" .. year .. ")."
	end
	if is_set(title) then
		render = render .. " ''" .. wikilinkFromTitle(title) .. "''."
	end
	if is_set(source) then
		render = render .. " " .. source .. "."
	end
	if is_set (quote) then
		quote = string.gsub(quote,"\n","<br>")
		quote = "“" .. quote .. "”"
		local div = mw.html.create ('div')
		div
		:attr("style", "width:80%; margin-left:10%;")
		:wikitext(quote)
		local br = mw.html.create ('br', selfClosing)
		render = italics(tostring(div)) .. tostring(br) .. render
	end
	return render
end

return p