public function handle(CodeGeneratorRequest $request)
{
$files = array();
foreach ((array) $request->getProtoFile() as $file) {
/** @var FileDescriptorProto $file */
$this->collectFile($file, array());
$specClassName = "Skrz\\Meta\\MetaSpec";
$customSpecClassName = false;
foreach ($file->getSourceCodeInfo()->getLocation() as $location) {
if ($location->getPath() === array(FileDescriptorProtoMeta::PACKAGE_PROTOBUF_FIELD) && $location->getLeadingComments() && preg_match("/@spec\\s+([a-zA-Z0-9_\\\\]+)/", $location->getLeadingComments(), $m)) {
$specClassName = $m[1];
$customSpecClassName = true;
}
}
uksort($this->messages, function ($a, $b) {
if (strlen($a) === strlen($b)) {
return strcmp($b, $a);
}
return strlen($b) - strlen($a);
});
$tmpFiles = array();
foreach ($this->messages as $className => $message) {
// compile message file
$codeFile = new CodeGeneratorResponse\File();
$codeFile->setName(str_replace("\\", "/", $className) . ".php");
$result = $this->generateMessage($className, $message);
$codeFile->setContent((string) $result->getFile());
$files[$className] = $codeFile;
// compile meta file
if (!class_exists($className)) {
$tmpFiles[] = $tmpFile = tempnam(sys_get_temp_dir(), "protoc-gen-php");
file_put_contents($tmpFile, (string) $result->getFile());
require_once $tmpFile;
}
/** @var AbstractMetaSpec $spec */
$spec = new $specClassName();
if (!$customSpecClassName) {
$spec->match($className)->addModule(new ProtobufModule());
}
$metaResult = $spec->compile($type = Type::fromString($className));
if ($metaResult !== null) {
$metaFile = new CodeGeneratorResponse\File();
$metaFile->setName(str_replace("\\", "/", $spec->createMetaClassName($type)) . ".php");
$metaFile->setContent((string) $metaResult->getFile());
$files[$metaResult->getClass()->getName()] = $metaFile;
}
}
foreach ($tmpFiles as $tmpFile) {
unlink($tmpFile);
}
foreach ($this->enums as $className => $enum) {
$enumFile = new CodeGeneratorResponse\File();
$enumFile->setName(str_replace("\\", "/", $className) . ".php");
$result = $this->generateEnum($className, $enum);
$enumFile->setContent((string) $result->getFile());
$files[$className] = $enumFile;
}
}
$response = new CodeGeneratorResponse();
$response->setFile($files);
return $response;
}
public function setupDictionary(CodeGeneratorRequest $req)
{
/* rough parsing, but works fine */
$result = array();
foreach ($req->getProtoFile() as $file) {
/* @var $file FileDescriptorProto */
$path = $file->getName();
$lines = preg_split("/\r?\n/", file_get_contents($path));
$info = $file->getSourceCodeInfo();
$result[$file->getName()] = array();
$tmp_location = null;
$is_nested = true;
$type = 0;
$stack = array();
$tmp = array();
$prior = 0;
$llvel = 0;
$prior = array(0, 0, 0);
foreach ($info->getLocation() as $location) {
/* @var $location \google\protobuf\SourceCodeInfo\Location */
$path = $location->getPath();
$span = $location->getSpan();
if (empty($path)) {
// whole entry
continue;
}
$level = count($path);
if (count($path) % 2 == 0 && $path[0] == 4) {
if ($prior[0] >= $level) {
array_pop($stack);
$prior[0] = $level - 1;
}
// whole
$name = substr($lines[$span[0]], $span[1], $span[2] - $span[1]);
if (preg_match("/^message/i", $name)) {
$stack[] = array("name" => $name, "info" => $location);
$prior[0] = $level;
} else {
if (preg_match("/^enum/i", $name)) {
$stack[] = array("name" => $name, "info" => $location);
$prior[0] = $level;
} else {
$tmp[] = $location;
}
}
}
if (count($path) % 2 == 1 && $path[0] == 4 && $path[count($path) - 1] == 1) {
// name
$name = substr($lines[$span[0]], $span[1], $span[2] - $span[1]);
// TODO(chobie): fix warnings
@(list($dummy, $n, $dummy) = explode(" ", $stack[count($stack) - 1]['name']));
if ($name == $n) {
// message or enum
$result[$file->getName()][$n]["message"] = $stack[count($stack) - 1]['info'];
} else {
$result[$file->getName()][$n][$name] = @$tmp[count($tmp) - 1];
array_pop($tmp);
}
}
}
// remove source code info
$file->clearSourceCodeInfo();
}
SourceInfoDictionary::register($result);
}
