include/swift/Parse/PersistentParserState.h

//===--- PersistentParserState.h - Parser State -----------------*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
// Parser state persistent across multiple parses.
//
//===----------------------------------------------------------------------===//

#ifndef SWIFT_PARSE_PERSISTENTPARSERSTATE_H
#define SWIFT_PARSE_PERSISTENTPARSERSTATE_H

#include "swift/Basic/SourceLoc.h"
#include "swift/Parse/LocalContext.h"

namespace swift {

class SourceFile;
class DeclContext;
class IterableDeclContext;

enum class CodeCompletionDelayedDeclKind {
  TopLevelCodeDecl,
  Decl,
  FunctionBody,
};

class CodeCompletionDelayedDeclState {
public:
  CodeCompletionDelayedDeclKind Kind;
  unsigned Flags;
  DeclContext *ParentContext;
  unsigned StartOffset;
  unsigned EndOffset;
  unsigned PrevOffset;

  CodeCompletionDelayedDeclState(CodeCompletionDelayedDeclKind Kind,
                                 unsigned Flags, DeclContext *ParentContext,
                                 unsigned StartOffset, unsigned EndOffset,
                                 unsigned PrevOffset)
      : Kind(Kind), Flags(Flags), ParentContext(ParentContext),
        StartOffset(StartOffset), EndOffset(EndOffset),
        PrevOffset(PrevOffset) {}
};

/// Parser state persistent across multiple parses.
class PersistentParserState {
  std::unique_ptr<CodeCompletionDelayedDeclState> CodeCompletionDelayedDeclStat;

  /// The local context for all top-level code.
  TopLevelContext TopLevelCode;

public:
  PersistentParserState();
  PersistentParserState(ASTContext &ctx) : PersistentParserState() { }
  ~PersistentParserState();

  void setCodeCompletionDelayedDeclState(SourceManager &SM, unsigned BufferID,
                                         CodeCompletionDelayedDeclKind Kind,
                                         unsigned Flags,
                                         DeclContext *ParentContext,
                                         SourceRange BodyRange,
                                         SourceLoc PreviousLoc);
  void restoreCodeCompletionDelayedDeclState(
      const CodeCompletionDelayedDeclState &other);

  bool hasCodeCompletionDelayedDeclState() const {
    return CodeCompletionDelayedDeclStat.get() != nullptr;
  }

  CodeCompletionDelayedDeclState &getCodeCompletionDelayedDeclState() {
    return *CodeCompletionDelayedDeclStat.get();
  }
  const CodeCompletionDelayedDeclState &
  getCodeCompletionDelayedDeclState() const {
    return *CodeCompletionDelayedDeclStat.get();
  }

  std::unique_ptr<CodeCompletionDelayedDeclState>
  takeCodeCompletionDelayedDeclState() {
    assert(hasCodeCompletionDelayedDeclState());
    return std::move(CodeCompletionDelayedDeclStat);
  }

  TopLevelContext &getTopLevelContext() {
    return TopLevelCode;
  }
};

} // end namespace swift

#endif