Permanently protected module

Difference between revisions of "Module:Data"

From Yugipedia
Jump to: navigation, search
(also strip "(video game)" from video game name input)
(From sandbox. New model for the data module.)
(Tag: Replaced)
Line 1: Line 1:
 
-- <pre>
 
-- <pre>
-- @name Data
+
local DATA = require( 'Module:Data/loader' )()
-- @description Interface for [[Module:Data/data]].
 
-- @notes Internal-only, so far.
 
-- @author [[User:Becasita]]
 
-- @contact [[User talk:Becasita]]
 
  
-------------------
+
return setmetatable( {
-- Export variable:
+
getAnimeRelease = DATA.anime.getRelease,
-------------------
+
getAnimeSeries = DATA.anime.getSeries,
local D = {};
 
  
--------------
+
getMangaRelease = DATA.manga.getRelease,
-- Load data:
+
getMangaSeries = DATA.manga.getSeries,
--------------
 
local DATA = mw.loadData( 'Module:Data/data' );
 
local NORM = DATA.normalize -- mw.loadData( 'Module:Data/normalize' );
 
  
-------------
+
getVideoGameRelease = DATA.videoGames.getRelease,
-- Constants:
+
getVideoGame = DATA.videoGames.getName,
-------------
+
}, {
-- @description
+
__index = DATA,
--local WORD_DELIMITERS = "[%s%-_]";
+
__call  = getmetatable( DATA ).__call,
 
+
} )
---------------------
 
-- Utility functions:
 
---------------------
 
-- mw functions:
 
local trim = mw.text.trim;
 
 
 
-- @name normalize
 
-- @description Normalizes the input applying a specific set of rules.
 
-- The rules are applied in case the input is of the type `string`
 
-- @parameter {*} arg Argument to be normalized.
 
-- @parameter {function} rules Normalization rules to apply.
 
-- @return A generally normalized arg to be used to index the normalization table.
 
local function normalize( arg, rules )
 
return type( arg ) == type( 'string' )
 
and rules( trim( arg ):lower() )
 
or nil
 
;
 
end
 
 
 
-- @name normalizeRegionLanguageMedium
 
-- @description Normalizes the input for region, language and medium,
 
-- since there are dependencies between these.
 
-- @parameter {*} arg Argument to be normalized.
 
-- @return A normalized arg to be used to index the normalization table.
 
local function normalizeRegionLanguageMedium( arg )
 
return normalize( arg, function( normalizedArg )
 
return normalizedArg
 
:gsub( "[%s%-_'!]", '' ) -- Remove a bunch of commonly used characters.
 
:gsub(    'north', '' ) -- Remove "north" (for region and language).
 
:gsub(    'yugioh', '' ) -- Remove "yugioh" (because of the medium).
 
:gsub(  'cardgame', '' ) -- Remove "cardgame" (because of the medium).
 
;
 
end );
 
end
 
 
 
-- @name normalizeRarity
 
-- @description Normalizes the input for rarity.
 
-- @parameter {*} arg Argument to be normalized.
 
-- @return A normalized arg to be used to index the normalization table.
 
--[[local function normalizeRarity( arg )
 
return normalize( arg, function( normalizedArg )
 
return NORM.rarity[ normalizedArg
 
:gsub(  ' rare$', '' ) -- Remove " rare" at the end (and with a space before it).
 
:gsub( "[%s%-_'/]", '' ) -- Remove a bunch of commonly used characters.
 
];
 
end );
 
end--]]
 
 
 
-----------------------
 
-- Methods (Interface):
 
-----------------------
 
-- @name getRegion
 
-- @description Gets the region for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {table|nil} A region or nil.
 
function D.getRegion( arg )
 
return DATA.region[
 
NORM.region[
 
normalizeRegionLanguageMedium( arg )
 
]
 
];
 
end
 
 
 
-- @name getLanguage
 
-- @description Gets the language for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {table|nil} A language or nil.
 
function D.getLanguage( arg )
 
return DATA.language[
 
NORM.language[
 
NORM.region[
 
normalizeRegionLanguageMedium( arg )
 
]
 
]
 
];
 
end
 
 
 
-- @name getMedium
 
-- @description Gets the language for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {table|nil} A language or nil.
 
function D.getMedium( arg )
 
local normalizedArg = normalizeRegionLanguageMedium( arg );
 
return DATA.medium[
 
NORM.medium[
 
NORM.region[
 
normalizedArg
 
] or normalizedArg
 
]
 
];
 
end
 
 
 
-- @name getEdition
 
-- @description Gets the edition for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} An edition or nil.
 
function D.getEdition( arg )
 
return DATA.edition[
 
normalize( arg, function( normalizedArg )
 
return NORM.edition[ normalizedArg
 
:gsub( '[%s%-_]', '' ) -- Remove a bunch of commonly used characters.
 
:gsub( 'edition', '' ) -- Redundant.
 
];
 
end )
 
];
 
end
 
 
 
-- @name getRelease
 
-- @description Gets the release for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} A release or `nil`.
 
function D.getRelease( arg )
 
return DATA.release[
 
normalize( arg, function( normalizedArg )
 
return NORM.release[ normalizedArg
 
:gsub(  '[%s%-_]', '' ) -- Remove a bunch of commonly used characters.
 
:gsub(    'case', '' ) -- for "Case Topper".
 
:gsub(    'card', '' ) -- for "Giant Card".
 
:gsub( 'official', '' ) -- For "Official Proxy".
 
];
 
end )
 
];
 
end
 
 
 
-- @name getRarity
 
-- @description Gets the rarity for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} A rarity or `nil`.
 
function D.getRarity( arg )
 
return DATA.rarity[
 
normalize( arg, function( normalizedArg )
 
return NORM.rarity[ normalizedArg
 
:gsub(  ' rare$', '' ) -- Remove " rare" at the end (and with a space before it).
 
:gsub( "[%s%-_'/]", '' ) -- Remove a bunch of commonly used characters.
 
];
 
end )
 
];
 
end
 
 
 
-- @name getCardType
 
function D.getCardType( arg )
 
return DATA.cardType[
 
normalize( arg, function( normalizedArg )
 
return normalizedArg
 
:gsub( "[%s%-_']", '' ) -- Remove a bunch of commonly used characters.
 
:gsub(    'card', '' )
 
;
 
end )
 
];
 
end
 
 
 
-- @name getAttribute
 
function D.getAttribute( arg )
 
return DATA.attribute[
 
normalize( arg, function( normalizedArg )
 
return normalizedArg
 
:gsub( "[%s%-_']", '' ) -- Remove a bunch of commonly used characters.
 
;
 
end )
 
];
 
end
 
 
 
-- @name getProperty
 
function D.getProperty( arg )
 
return DATA.property[
 
normalize( arg, function( normalizedArg )
 
return normalizedArg
 
:gsub( "[%s%-_']", '' ) -- Remove a bunch of commonly used characters.
 
;
 
end )
 
];
 
end
 
 
 
-- @name getEffectType
 
function D.getEffectType( arg, like )
 
local normalizedArg = normalize( arg, function( normalizedArg )
 
return normalizedArg;
 
end );
 
 
 
return normalizedArg and DATA.effectType[
 
(like or normalizedArg:match( 'like' )) and 'like' or 'regular'
 
][ normalizedArg
 
:gsub( "[%s%-_']", '' ) -- Remove a bunch of commonly used characters.
 
:gsub(  "effect", '' )
 
:gsub(    "like", '' )
 
];
 
end
 
 
 
-- @name getLinkArrow
 
function D.getLinkArrow( arg )
 
return DATA.linkArrow[
 
normalize( arg, function( normalizedArg )
 
return normalizedArg
 
:gsub( "[%s%-_']", '' ) -- Remove a bunch of commonly used characters.
 
;
 
end )
 
];
 
end
 
---------------
 
-- Anime stuff:
 
---------------
 
-- @name getAnimeRelease
 
-- @description Gets the anime release for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} Anime release or `nil`.
 
function D.getAnimeRelease( arg )
 
return DATA.anime.release[
 
normalize( arg, function( normalizedArg )
 
return NORM.anime.release[ normalizedArg
 
:gsub( '[%s%-_]', '' ) -- Remove a bunch of commonly used characters.
 
];
 
end )
 
];
 
end
 
 
 
-- @name getAnimeSeries
 
-- @description Gets the anime series for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} Anime series or `nil`.
 
function D.getAnimeSeries( arg )
 
return DATA.anime.series[
 
normalize( arg, function( normalizedArg )
 
return NORM.anime.series[ normalizedArg
 
:gsub( "[%s%-_'/:!]", '' ) -- Remove a bunch of commonly used characters.
 
:gsub(        'the', '' )
 
:gsub(        'ygo', '' )
 
:gsub(      'yugioh', '' )
 
];
 
end )
 
];
 
end
 
 
 
---------------
 
-- Manga stuff:
 
---------------
 
-- @name getMangaRelease
 
-- @description Gets the manga release for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} Manga release or `nil`.
 
function D.getMangaRelease( arg )
 
return DATA.manga.release[
 
normalize( arg, function( normalizedArg )
 
return NORM.manga.release[ normalizedArg
 
:gsub( '[%s%-_]', '' ) -- Remove a bunch of commonly used characters.
 
];
 
end )
 
];
 
end
 
 
 
-- @name getMangaSeries
 
-- @description Gets the manga series for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} Manga series or `nil`.
 
function D.getMangaSeries( arg )
 
return DATA.manga.series[
 
normalize( arg, function( normalizedArg )
 
return NORM.manga.series[ normalizedArg
 
:gsub( "[%s%-_'/:!]", '' ) -- Remove a bunch of commonly used characters.
 
:gsub(        'the', '' )
 
:gsub(  'strongest', '' )
 
:gsub(        'ygo', '' )
 
:gsub(      'yugioh', '' )
 
];
 
end )
 
];
 
end
 
 
 
--------------------
 
-- Video game stuff:
 
--------------------
 
-- @name getVideoGameRelease
 
-- @description Gets the video game release for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} Video game release or `nil`.
 
function D.getVideoGameRelease( arg )
 
return DATA.videoGame.release[
 
normalize( arg, function( normalizedArg )
 
return NORM.videoGame.release[ normalizedArg
 
:gsub( '[%s%-_]', '' ) -- Remove a bunch of commonly used characters.
 
];
 
end )
 
];
 
end
 
 
 
-- @name getVideoGame
 
-- @description Gets the video game for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} Manga series or `nil`.
 
function D.getVideoGame( arg )
 
return DATA.videoGame.game[
 
normalize( arg, function( normalizedArg )
 
-- Add a space at the end, to replace roman numerals more easily:
 
normalizedArg = table.concat( { normalizedArg, ' ' } )
 
-- Replace roman numerals:
 
:gsub(  ' viii[!: ]', '8' )
 
:gsub(  ' vii[!: ]', '7' )
 
:gsub(    ' vi[!: ]', '6' )
 
:gsub(    ' v[!: ]', '5' )
 
:gsub(    ' iv[!: ]', '4' )
 
:gsub(  ' iii[!: ]', '3' )
 
:gsub(    ' ii[!: ]', '2' )
 
:gsub(    ' i[!: ]', '1' )
 
-- Remove a bunch of commonly used characters:
 
:gsub( "[%s%-_'/:!]",  '' )
 
-- Remove series names:
 
:gsub(      'yugioh',  '' )
 
:gsub(        '5ds',  '' )
 
:gsub(      'zexal',  '' )
 
:gsub(       'arcv',  '' )
 
:gsub(      'vrains',  '' )
 
-- Remove some redundant words:
 
:gsub(        'the',  '' )
 
:gsub(    'gameboy',  '' )
 
-- Normalize some titles:
 
:gsub(    'expert', 'ex' )
 
:gsub( 'worldchampionshiptournament', 'worldchampionship' )
 
-- Remove "(video game)"
 
:gsub( '(videogame)',  '' )
 
;
 
 
 
-- Remove "gx"/"ygo", if it's large enough (preserve "gx01", "gx3", "ygoo", "ygo" ):
 
if normalizedArg:len() > 4 then
 
normalizedArg
 
:gsub(  'gx', '' )
 
:gsub( 'ygo', '' )
 
;
 
-- Remove "duelmonsters", if it's large enough (preserve cases like /duelmonsters\d/):
 
if normalizedArg:len() > 13 then
 
normalizedArg:gsub( 'duelmonsters', '' );
 
-- Remove "worldchampionship", if it's large enough (preserve cases like /worldchampionship20\d\d/):
 
if normalizedArg:len() > 21 then
 
normalizedArg:gsub( 'worldchampionship', '' );
 
end
 
end
 
end
 
 
 
-- Finally:
 
return NORM.videoGame.game[ normalizedArg ];
 
end )
 
];
 
end
 
 
 
-------------------
 
-- Templates stuff:
 
-------------------
 
-- @name getCardGalleryType
 
-- @description Gets the `{{Card gallery}}` type for `arg`. `nil` if not found.
 
-- @parameter {string} arg
 
-- @return {string|nil} `{{Card gallery}}` possible `{{{type}}}`s.
 
function D.getCardGalleryType( arg )
 
return DATA.templates[ 'Card gallery' ].types[
 
normalize( arg, function( normalizedArg )
 
return normalizedArg
 
:gsub( "[%s%-s]", '' ) -- Remove a bunch of commonly used characters.
 
:gsub(  'video', '' ) -- Remove "video".
 
;
 
end )
 
];
 
end
 
 
 
----------
 
-- Return:
 
----------
 
-- @exports `D`: Interface to interact with [[Module:Data/data]].
 
return setmetatable( D, {
 
__call = function( t, ... )
 
assert( t == D );
 
 
 
local arguments = {
 
size = select( '#', ... ),
 
...
 
};
 
 
 
local data = DATA;
 
 
for i = 1, arguments.size do
 
data = (data or {})[ arguments[ i ] ];
 
end
 
 
 
return data;
 
end
 
} );
 
 
-- </pre>
 
-- </pre>

Revision as of 00:19, 4 January 2020

-- <pre>
local DATA = require( 'Module:Data/loader' )()

return setmetatable( {
	getAnimeRelease = DATA.anime.getRelease,
	getAnimeSeries = DATA.anime.getSeries,

	getMangaRelease = DATA.manga.getRelease,
	getMangaSeries = DATA.manga.getSeries,

	getVideoGameRelease = DATA.videoGames.getRelease,
	getVideoGame = DATA.videoGames.getName,
}, {
	__index = DATA,
	__call  = getmetatable( DATA ).__call,
} )
-- </pre>