42 lines
1.1 KiB
C++
42 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include "AST/Base.h"
|
|
#include <iomanip>
|
|
#include <memory>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace plsm {
|
|
namespace ast {
|
|
class FnArg : public ASTNode {
|
|
const std::string name;
|
|
const std::unique_ptr<Type> type;
|
|
|
|
public:
|
|
FnArg(LOC_ARG, const std::string &name, std::unique_ptr<Type> &type)
|
|
: ASTNode(location), name(name), type(std::move(type)) {}
|
|
|
|
virtual boost::json::value toJson() override;
|
|
static std::unique_ptr<FnArg> fromJson(boost::json::value json);
|
|
};
|
|
|
|
class FnDef : public Stmt {
|
|
const std::string name;
|
|
const std::vector<std::unique_ptr<FnArg>> args;
|
|
const std::unique_ptr<Type> returnType;
|
|
const std::unique_ptr<Expr> body;
|
|
|
|
public:
|
|
FnDef(LOC_ARG, const std::string &name,
|
|
std::vector<std::unique_ptr<FnArg>> &args,
|
|
std::unique_ptr<Type> &returnType, std::unique_ptr<Expr> &body)
|
|
: Stmt(location), name(name), args(std::move(args)),
|
|
returnType(std::move(returnType)), body(std::move(body)) {}
|
|
|
|
virtual boost::json::value toJson() override;
|
|
static std::unique_ptr<FnDef> fromJson(boost::json::value json);
|
|
};
|
|
} // namespace ast
|
|
} // namespace plsm
|