Regexp.cc

#include "rewriter/Regexp.h"
#include "ast/Helpers.h"
#include "ast/ast.h"
#include "core/Context.h"
#include "core/Names.h"
#include "core/core.h"
#include "rewriter/rewriter.h"

using namespace std;

namespace sorbet::rewriter {

vector<ast::ExpressionPtr> Regexp::run(core::MutableContext ctx, ast::Assign *asgn) {
    auto lhs = ast::cast_tree<ast::UnresolvedConstantLit>(asgn->lhs);
    if (lhs == nullptr) {
        return {};
    }

    auto send = ast::cast_tree<ast::Send>(asgn->rhs);
    if (send == nullptr || send->fun != core::Names::new_()) {
        return {};
    }

    auto recv = ast::cast_tree<ast::ConstantLit>(send->recv);
    if (recv == nullptr || recv->symbol != core::Symbols::Regexp()) {
        return {};
    }

    vector<ast::ExpressionPtr> stats;
    auto type = ast::MK::Constant(send->loc, core::Symbols::Regexp());
    auto newRhs = ast::MK::Let(send->loc, std::move(asgn->rhs), std::move(type));
    stats.emplace_back(ast::MK::Assign(asgn->loc, std::move(asgn->lhs), std::move(newRhs)));
    return stats;
}

}; // namespace sorbet::rewriter