public function getNode(ImprovedClassSource $classSource)
{
//Namespace
$code = $this->builderFactory->namespace($classSource->getNamespace());
//Adds use to header of file
foreach ($classSource->getImports() as $import) {
$code->addStmt($this->builderFactory->use($import->getFullName()));
}
$classCode = $this->classGenerator->generate($classSource);
foreach ($classSource->getMethods() as $method) {
$result = $this->methodFactory->generate($method);
$classCode->addStmt($result);
}
$code->addStmt($classCode);
return $code->getNode();
}
