1/*
2 * Copyright (c) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#ifndef FLEX_BISON_SCANNER_H
17#define FLEX_BISON_SCANNER_H
18
19#include <cstring>
20#include "script_utils.h"
21
22/* 重要 */
23#if !defined(yyFlexLexerOnce)
24#undef yyFlexLexer
25#define yyFlexLexer script_FlexLexer // 根据prefix修改
26#include "FlexLexer.h"
27#endif
28#include "parser.hpp"
29
30/* 替换默认的get_next_token定义 */
31#undef YY_DECL
32#define YY_DECL Uscript::Parser::symbol_type Uscript::Scanner::nextToken()
33
34namespace Uscript {
35class ScriptInterpreter;
36
37class Scanner : public yyFlexLexer {
38public:
39    explicit Scanner(ScriptInterpreter* interpreter)
40    {
41        loc = location();
42    }
43    ~Scanner() override {}
44    // 不需要手动实现这个函数，Flex会生成YY_DECL宏定义的代码来实现这个函数
45    virtual Uscript::Parser::symbol_type nextToken();
46    int yywrap() override
47    {
48        return 1;
49    }
50
51    int LexerInput(char* buf, int maxSize) override;
52
53    void SetPkgStream(Hpackage::PkgManager::StreamPtr pkgStream)
54    {
55        pkgStream_ = pkgStream;
56    }
57
58private:
59    Hpackage::PkgManager::StreamPtr pkgStream_ = nullptr;
60    size_t currPos = 0;
61    location loc {};
62};
63} // namespace Uscript
64#endif // FLEX_BISON_SCANNER_H
65