mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/Scribunto
synced 2024-12-13 08:48:39 +00:00
76 lines
1.7 KiB
PHP
76 lines
1.7 KiB
PHP
|
<?php
|
||
|
|
||
|
class Scribunto_LuaModule extends ScribuntoModuleBase {
|
||
|
/**
|
||
|
* @var mixed
|
||
|
*/
|
||
|
protected $initChunk;
|
||
|
|
||
|
/**
|
||
|
* @var Scribunto_LuaEngine
|
||
|
*/
|
||
|
protected $engine;
|
||
|
|
||
|
/**
|
||
|
* @param Scribunto_LuaEngine $engine
|
||
|
* @param string $code
|
||
|
* @param string|bool $chunkName
|
||
|
*/
|
||
|
public function __construct( Scribunto_LuaEngine $engine, $code, $chunkName ) {
|
||
|
parent::__construct( $engine, $code, $chunkName );
|
||
|
}
|
||
|
|
||
|
public function validate() {
|
||
|
try {
|
||
|
$this->getInitChunk();
|
||
|
} catch ( ScribuntoException $e ) {
|
||
|
return $e->toStatus();
|
||
|
}
|
||
|
return Status::newGood();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Get the chunk which, when called, will return the export table.
|
||
|
* @return mixed
|
||
|
*/
|
||
|
public function getInitChunk() {
|
||
|
if ( !$this->initChunk ) {
|
||
|
$this->initChunk = $this->engine->getInterpreter()->loadString(
|
||
|
$this->code,
|
||
|
// Prepending an "=" to the chunk name avoids truncation or a "[string" prefix
|
||
|
'=' . $this->chunkName );
|
||
|
}
|
||
|
return $this->initChunk;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Invoke a function within the module. Return the expanded wikitext result.
|
||
|
*
|
||
|
* @param string $name
|
||
|
* @param PPFrame $frame
|
||
|
* @throws ScribuntoException
|
||
|
* @return string|null
|
||
|
*/
|
||
|
public function invoke( $name, $frame ) {
|
||
|
$ret = $this->engine->executeModule( $this->getInitChunk(), $name, $frame );
|
||
|
|
||
|
if ( !isset( $ret ) ) {
|
||
|
throw $this->engine->newException(
|
||
|
'scribunto-common-nosuchfunction', [ 'args' => [ $name ] ]
|
||
|
);
|
||
|
}
|
||
|
if ( !$this->engine->getInterpreter()->isLuaFunction( $ret ) ) {
|
||
|
throw $this->engine->newException(
|
||
|
'scribunto-common-notafunction', [ 'args' => [ $name ] ]
|
||
|
);
|
||
|
}
|
||
|
|
||
|
$result = $this->engine->executeFunctionChunk( $ret, $frame );
|
||
|
if ( isset( $result[0] ) ) {
|
||
|
return $result[0];
|
||
|
} else {
|
||
|
return null;
|
||
|
}
|
||
|
}
|
||
|
}
|