Module:AeroWikiListTools: Difference between revisions

← Older edit

@@ Line 1: / Line 1: @@
+-- Created by @Sqwishyish on discord
+-- Contact me for any issues
 local getArgs = require('Module:Arguments').getArgs
@@ Line 12: / Line 15: @@
 p.split_list = makeInvokeFunc('_split_list')
 p._split_list = function (args)
-	local input_string = args[1]
+	-- Get arguments passed from function call
-	local template = args[2]
+	local input_string = args[1] -- List string to modify
+	local template = args[2] -- String to put at each {split}
+	local missing_string = args[3] -- String to return when either the template or the input is nil
      local new_string = ''
-     if type(input_string) == table then
+    -- Check if input string exists
-		input_string = table.concat(input_string, '')
+     if input_string == nil or template == nil then
+		return missing_string
+    end
+    -- Check if input string is not 'None'
+    if input_string == 'None' then
+		return 'None'
+    end
+	-- Split string by commas and append the template at each {split} point
+    for split in string.gmatch(input_string , '([^,]+)') do
+        new_string = new_string .. template:gsub('{split}', split:match('^%s*(.-)%s*$')) .. ' '
      end
-     for split in string.gmatch(input_string, '([^, ]+)') do
+    -- Remove comma on end of list
-        new_string = new_string .. template:gsub('{split}', split) .. ', '
+     if template:sub(-1) == ',' then
+    	new_string = new_string:sub(1, -3)
+    elseif template:sub(-2) == ', ' then
+    	new_string = new_string:sub(1, -4)
      end
-     return new_string:sub(1, -3)
+    -- Return the new string to the list
+     return new_string
 end
 return p