//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
///
/// \file
/// LFI assembly parser.
///
//===----------------------------------------------------------------------===//

#include "llvm/MC/MCLFIRewriter.h"
#include "llvm/MC/MCParser/MCAsmParserExtension.h"

using namespace llvm;

namespace {
class LFIAsmParser : public MCAsmParserExtension {
  MCLFIRewriter *Rewriter;
  template <bool (LFIAsmParser::*HandlerMethod)(StringRef, SMLoc)>
  void addDirectiveHandler(StringRef Directive) {
    MCAsmParser::ExtensionDirectiveHandler Handler =
        std::make_pair(this, HandleDirective<LFIAsmParser, HandlerMethod>);

    getParser().addDirectiveHandler(Directive, Handler);
  }

public:
  LFIAsmParser(MCLFIRewriter *Exp) : Rewriter(Exp) {}
  void Initialize(MCAsmParser &Parser) override {
    // Call the base implementation.
    MCAsmParserExtension::Initialize(Parser);
    addDirectiveHandler<&LFIAsmParser::parseRewriteDisable>(
        ".lfi_rewrite_disable");
    addDirectiveHandler<&LFIAsmParser::parseRewriteEnable>(
        ".lfi_rewrite_enable");
  }

  /// ::= {.lfi_rewrite_disable}
  bool parseRewriteDisable(StringRef Directive, SMLoc Loc) {
    getParser().checkForValidSection();
    if (getLexer().isNot(AsmToken::EndOfStatement))
      return TokError("unexpected token");
    Lex();

    Rewriter->disable();

    return false;
  }

  /// ::= {.lfi_rewrite_enable}
  bool parseRewriteEnable(StringRef Directive, SMLoc Loc) {
    getParser().checkForValidSection();
    if (getLexer().isNot(AsmToken::EndOfStatement))
      return TokError("unexpected token");
    Lex();

    Rewriter->enable();

    return false;
  }
};
} // namespace

MCAsmParserExtension *llvm::createLFIAsmParser(MCLFIRewriter *Exp) {
  return new LFIAsmParser(Exp);
}