1 files changed, 6 insertions, 7 deletions
diff --git a/fpdfsdk/javascript/cjs_runtime.h b/fpdfsdk/javascript/cjs_runtime.h
index 2b075afa01..1b004ded89 100644
--- a/fpdfsdk/javascript/cjs_runtime.h
+++ b/fpdfsdk/javascript/cjs_runtime.h
@@ -30,20 +30,19 @@ class CJS_Runtime : public IJS_Runtime,
   static CJS_Runtime* FromContext(const IJS_Context* cc);
   static CJS_Runtime* CurrentRuntimeFromIsolate(v8::Isolate* pIsolate);
 
-  explicit CJS_Runtime(CPDFSDK_FormFillEnvironment* pApp);
+  explicit CJS_Runtime(CPDFSDK_FormFillEnvironment* pFormFillEnv);
   ~CJS_Runtime() override;
 
   // IJS_Runtime
   IJS_Context* NewContext() override;
   void ReleaseContext(IJS_Context* pContext) override;
   IJS_Context* GetCurrentContext() override;
-  void SetReaderDocument(CPDFSDK_Document* pReaderDoc) override;
-  CPDFSDK_Document* GetReaderDocument() override;
+
+  CPDFSDK_FormFillEnvironment* GetFormFillEnv() const override;
+
   int ExecuteScript(const CFX_WideString& script,
                     CFX_WideString* info) override;
 
-  CPDFSDK_FormFillEnvironment* GetReaderEnv() const { return m_pEnv; }
-
   // Returns true if the event isn't already found in the set.
   bool AddEventToSet(const FieldEvent& event);
   void RemoveEventFromSet(const FieldEvent& event);
@@ -61,10 +60,10 @@ class CJS_Runtime : public IJS_Runtime,
 
  private:
   void DefineJSObjects();
+  void SetFormFillEnvToDocument();
 
   std::vector<std::unique_ptr<CJS_Context>> m_ContextArray;
-  CPDFSDK_FormFillEnvironment* const m_pEnv;
-  CPDFSDK_Document* m_pDocument;
+  CPDFSDK_FormFillEnvironment* const m_pFormFillEnv;
   bool m_bBlocking;
   bool m_isolateManaged;
   std::set<FieldEvent> m_FieldEventSet;