/*
    * Copyright (c) 2002-2025 Gargoyle Software Inc.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    * https://www.apache.org/licenses/LICENSE-2.0
    *
    * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.htmlunit.javascript.host.worker;
  
  import java.io.IOException;
  import java.lang.reflect.Method;
  import java.net.URL;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.htmlunit.BrowserVersion;
  import org.htmlunit.WebClient;
  import org.htmlunit.WebRequest;
  import org.htmlunit.WebResponse;
  import org.htmlunit.corejs.javascript.Context;
  import org.htmlunit.corejs.javascript.ContextAction;
  import org.htmlunit.corejs.javascript.ContextFactory;
  import org.htmlunit.corejs.javascript.Function;
  import org.htmlunit.corejs.javascript.FunctionObject;
  import org.htmlunit.corejs.javascript.Scriptable;
  import org.htmlunit.corejs.javascript.ScriptableObject;
  import org.htmlunit.html.HtmlPage;
  import org.htmlunit.javascript.AbstractJavaScriptEngine;
  import org.htmlunit.javascript.HtmlUnitContextFactory;
  import org.htmlunit.javascript.HtmlUnitScriptable;
  import org.htmlunit.javascript.JavaScriptEngine;
  import org.htmlunit.javascript.RecursiveFunctionObject;
  import org.htmlunit.javascript.background.BasicJavaScriptJob;
  import org.htmlunit.javascript.background.JavaScriptJob;
  import org.htmlunit.javascript.configuration.ClassConfiguration;
  import org.htmlunit.javascript.configuration.JsxClass;
  import org.htmlunit.javascript.configuration.JsxConstructor;
  import org.htmlunit.javascript.configuration.JsxFunction;
  import org.htmlunit.javascript.configuration.JsxGetter;
  import org.htmlunit.javascript.configuration.JsxSetter;
  import org.htmlunit.javascript.configuration.WorkerJavaScriptConfiguration;
  import org.htmlunit.javascript.host.PermissionStatus;
  import org.htmlunit.javascript.host.Permissions;
  import org.htmlunit.javascript.host.PushManager;
  import org.htmlunit.javascript.host.PushSubscription;
  import org.htmlunit.javascript.host.PushSubscriptionOptions;
  import org.htmlunit.javascript.host.Window;
  import org.htmlunit.javascript.host.WindowOrWorkerGlobalScopeMixin;
  import org.htmlunit.javascript.host.event.Event;
  import org.htmlunit.javascript.host.event.MessageEvent;
  import org.htmlunit.javascript.host.event.SecurityPolicyViolationEvent;
  import org.htmlunit.javascript.host.media.MediaSource;
  import org.htmlunit.javascript.host.media.SourceBuffer;
  import org.htmlunit.javascript.host.media.SourceBufferList;
  import org.htmlunit.util.MimeType;
  
  /**
   * The scope for the execution of {@link Worker}s.
   *
   * @author Marc Guillemot
   * @author Ronald Brill
   * @author Rural Hunter
   */
  @JsxClass
  public class DedicatedWorkerGlobalScope extends WorkerGlobalScope {
  
      private static final Log LOG = LogFactory.getLog(DedicatedWorkerGlobalScope.class);
  
      private static final Method GETTER_NAME;
      private static final Method SETTER_NAME;
  
      private Map<Class<? extends Scriptable>, Scriptable> prototypes_ = new HashMap<>();
      private final Window owningWindow_;
      private final String origin_;
      private String name_;
      private final Worker worker_;
      private WorkerLocation workerLocation_;
      private WorkerNavigator workerNavigator_;
  
      static {
          try {
              GETTER_NAME = DedicatedWorkerGlobalScope.class.getDeclaredMethod("jsGetName");
              SETTER_NAME = DedicatedWorkerGlobalScope.class.getDeclaredMethod("jsSetName", Scriptable.class);
          }
          catch (NoSuchMethodException | SecurityException e) {
              throw new RuntimeException(e);
          }
      }
  
     /**
      * For prototype instantiation.
      */
     public DedicatedWorkerGlobalScope() {
         // prototype constructor
         super();
         owningWindow_ = null;
         origin_ = null;
         name_ = null;
         worker_ = null;
         workerLocation_ = null;
     }
 
     /**
      * JavaScript constructor.
      */
     @Override
     @JsxConstructor
     public void jsConstructor() {
         // nothing to do
     }
 
     /**
      * Constructor.
      * @param webClient the WebClient
      * @param worker the started worker
      * @throws Exception in case of problem
      */
     DedicatedWorkerGlobalScope(final Window owningWindow, final Context context, final WebClient webClient,
             final String name, final Worker worker) throws Exception {
         super();
 
         final BrowserVersion browserVersion = webClient.getBrowserVersion();
 
         context.initSafeStandardObjects(this);
         JavaScriptEngine.configureRhino(webClient, browserVersion, this);
 
         final WorkerJavaScriptConfiguration jsConfig = WorkerJavaScriptConfiguration.getInstance(browserVersion);
 
         ClassConfiguration config = jsConfig.getClassConfiguration(
                 DedicatedWorkerGlobalScope.class.getSuperclass().getSimpleName());
         final HtmlUnitScriptable parentPrototype = JavaScriptEngine.configureClass(config, this);
 
         config = jsConfig.getClassConfiguration(DedicatedWorkerGlobalScope.class.getSimpleName());
         final HtmlUnitScriptable prototype = JavaScriptEngine.configureClass(config, this);
         prototype.setPrototype(parentPrototype);
         setPrototype(prototype);
 
         final Map<Class<? extends Scriptable>, Scriptable> prototypes = new HashMap<>();
         final Map<String, Scriptable> prototypesPerJSName = new HashMap<>();
 
         prototypes.put(config.getHostClass(), prototype);
         prototypesPerJSName.put(config.getClassName(), prototype);
 
         final FunctionObject functionObject =
                 new RecursiveFunctionObject(DedicatedWorkerGlobalScope.class.getSimpleName(),
                         config.getJsConstructor().getValue(), this, browserVersion);
         functionObject.addAsConstructor(this, prototype, ScriptableObject.DONTENUM);
 
         JavaScriptEngine.configureScope(this, config, functionObject, jsConfig,
                 browserVersion, prototypes, prototypesPerJSName);
         // remove some aliases
         delete("webkitURL");
         delete("WebKitCSSMatrix");
 
         // hack for the moment
         if (browserVersion.isFirefox()) {
             delete(MediaSource.class.getSimpleName());
             delete(SecurityPolicyViolationEvent.class.getSimpleName());
             delete(SourceBuffer.class.getSimpleName());
             delete(SourceBufferList.class.getSimpleName());
         }
 
         if (browserVersion.isFirefoxESR()) {
             delete(Permissions.class.getSimpleName());
             delete(PermissionStatus.class.getSimpleName());
             delete(PushManager.class.getSimpleName());
             delete(PushSubscription.class.getSimpleName());
             delete(PushSubscriptionOptions.class.getSimpleName());
             delete(ServiceWorkerRegistration.class.getSimpleName());
         }
 
         if (!webClient.getOptions().isWebSocketEnabled()) {
             delete("WebSocket");
         }
 
         setPrototypes(prototypes);
 
         owningWindow_ = owningWindow;
         final URL currentURL = owningWindow.getWebWindow().getEnclosedPage().getUrl();
         origin_ = currentURL.getProtocol() + "://" + currentURL.getHost() + ':' + currentURL.getPort();
 
         name_ = name;
         defineProperty("name", null, GETTER_NAME, SETTER_NAME, ScriptableObject.READONLY);
 
         worker_ = worker;
         workerLocation_ = null;
     }
 
     /**
      * Get the scope itself.
      * @return this
      */
     @JsxGetter
     public Object getSelf() {
         return this;
     }
 
     /**
      * Returns the {@code onmessage} event handler.
      * @return the {@code onmessage} event handler
      */
     @JsxGetter
     public Function getOnmessage() {
         return getEventHandler(Event.TYPE_MESSAGE);
     }
 
     /**
      * Sets the {@code onmessage} event handler.
      * @param onmessage the {@code onmessage} event handler
      */
     @JsxSetter
     public void setOnmessage(final Object onmessage) {
         setEventHandler(Event.TYPE_MESSAGE, onmessage);
     }
 
     /**
      * @return returns the WorkerLocation associated with the worker
      */
     @JsxGetter
     public WorkerLocation getLocation() {
         return workerLocation_;
     }
 
     /**
      * @return returns the WorkerNavigator associated with the worker
      */
     @JsxGetter
     public WorkerNavigator getNavigator() {
         return workerNavigator_;
     }
 
     /**
      * @return the {@code name}
      */
     public String jsGetName() {
         return name_;
     }
 
     /**
      * Sets the {@code name}.
      * @param name the new name
      */
     public void jsSetName(final Scriptable name) {
         name_ = JavaScriptEngine.toString(name);
     }
 
     /**
      * Posts a message to the {@link Worker} in the page's context.
      * @param message the message
      */
     @JsxFunction
     public void postMessage(final Object message) {
         final MessageEvent event = new MessageEvent();
         event.initMessageEvent(Event.TYPE_MESSAGE, false, false, message, origin_, "",
                                     owningWindow_, JavaScriptEngine.UNDEFINED);
         event.setParentScope(owningWindow_);
         event.setPrototype(owningWindow_.getPrototype(event.getClass()));
 
         if (LOG.isDebugEnabled()) {
             LOG.debug("[DedicatedWorker] postMessage: {}" + message);
         }
         final JavaScriptEngine jsEngine =
                 (JavaScriptEngine) owningWindow_.getWebWindow().getWebClient().getJavaScriptEngine();
         final ContextAction<Object> action = cx -> {
             worker_.getEventListenersContainer().executeCapturingListeners(event, null);
             final Object[] args = {event};
             worker_.getEventListenersContainer().executeBubblingListeners(event, args);
             return null;
         };
 
         final HtmlUnitContextFactory cf = jsEngine.getContextFactory();
 
         final JavaScriptJob job = new WorkerJob(cf, action, "postMessage: " + JavaScriptEngine.toString(message));
 
         final HtmlPage page = (HtmlPage) owningWindow_.getDocument().getPage();
         owningWindow_.getWebWindow().getJobManager().addJob(job, page);
     }
 
     void messagePosted(final Object message) {
         final MessageEvent event = new MessageEvent();
         event.initMessageEvent(Event.TYPE_MESSAGE, false, false, message, origin_, "",
                                     owningWindow_, JavaScriptEngine.UNDEFINED);
         event.setParentScope(owningWindow_);
         event.setPrototype(owningWindow_.getPrototype(event.getClass()));
 
         final JavaScriptEngine jsEngine =
                 (JavaScriptEngine) owningWindow_.getWebWindow().getWebClient().getJavaScriptEngine();
         final ContextAction<Object> action = cx -> {
             executeEvent(cx, event);
             return null;
         };
 
         final HtmlUnitContextFactory cf = jsEngine.getContextFactory();
 
         final JavaScriptJob job = new WorkerJob(cf, action, "messagePosted: " + JavaScriptEngine.toString(message));
 
         final HtmlPage page = (HtmlPage) owningWindow_.getDocument().getPage();
         owningWindow_.getWebWindow().getJobManager().addJob(job, page);
     }
 
     void executeEvent(final Context cx, final MessageEvent event) {
         final List<Scriptable> handlers = getEventListenersContainer().getListeners(Event.TYPE_MESSAGE, false);
         if (handlers != null) {
             final Object[] args = {event};
             for (final Scriptable scriptable : handlers) {
                 if (scriptable instanceof Function) {
                     final Function handlerFunction = (Function) scriptable;
                     handlerFunction.call(cx, this, this, args);
                 }
             }
         }
 
         final Function handlerFunction = getEventHandler(Event.TYPE_MESSAGE);
         if (handlerFunction != null) {
             final Object[] args = {event};
             handlerFunction.call(cx, this, this, args);
         }
     }
 
     /**
      * Import external script(s).
      * @param cx the current context
      * @param scope the scope
      * @param thisObj this object
      * @param args the script(s) to import
      * @param funObj the JS function called
      * @throws IOException in case of problem loading/executing the scripts
      */
     @JsxFunction
     public static void importScripts(final Context cx, final Scriptable scope,
             final Scriptable thisObj, final Object[] args, final Function funObj) throws IOException {
         final DedicatedWorkerGlobalScope workerScope = (DedicatedWorkerGlobalScope) thisObj;
 
         final WebClient webClient = workerScope.owningWindow_.getWebWindow().getWebClient();
         for (final Object arg : args) {
             final String url = JavaScriptEngine.toString(arg);
             workerScope.loadAndExecute(webClient, url, cx, true);
         }
     }
 
     void loadAndExecute(final WebClient webClient, final String url,
             final Context context, final boolean checkMimeType) throws IOException {
         final HtmlPage page = (HtmlPage) owningWindow_.getDocument().getPage();
         final URL fullUrl = page.getFullyQualifiedUrl(url);
 
         workerLocation_ = new WorkerLocation(fullUrl, origin_);
         workerLocation_.setParentScope(this);
         workerLocation_.setPrototype(getPrototype(workerLocation_.getClass()));
 
         workerNavigator_ = new WorkerNavigator(webClient.getBrowserVersion());
         workerNavigator_.setParentScope(this);
         workerNavigator_.setPrototype(getPrototype(workerNavigator_.getClass()));
 
         final WebRequest webRequest = new WebRequest(fullUrl);
         final WebResponse response = webClient.loadWebResponse(webRequest);
         if (checkMimeType && !MimeType.isJavascriptMimeType(response.getContentType())) {
             throw JavaScriptEngine.reportRuntimeError(
                     "NetworkError: importScripts response is not a javascript response");
         }
 
         final String scriptCode = response.getContentAsString();
         final AbstractJavaScriptEngine<?> javaScriptEngine = webClient.getJavaScriptEngine();
 
         final DedicatedWorkerGlobalScope thisScope = this;
         final ContextAction<Object> action = cx -> {
             return javaScriptEngine.execute(page, thisScope, scriptCode, fullUrl.toExternalForm(), 1);
         };
 
         final HtmlUnitContextFactory cf = javaScriptEngine.getContextFactory();
 
         if (context != null) {
             action.run(context);
         }
         else {
             final JavaScriptJob job = new WorkerJob(cf, action, "loadAndExecute " + url);
             owningWindow_.getWebWindow().getJobManager().addJob(job, page);
         }
     }
 
     /**
      * Sets a chunk of JavaScript to be invoked at some specified time later.
      * The invocation occurs only if the window is opened after the delay
      * and does not contain another page than the one that originated the setTimeout.
      *
      * @see <a href="https://developer.mozilla.org/en-US/docs/Web/API/WindowOrWorkerGlobalScope/setTimeout">
      * MDN web docs</a>
      *
      * @param context the JavaScript context
      * @param scope the scope
      * @param thisObj the scriptable
      * @param args the arguments passed into the method
      * @param function the function
      * @return the id of the created timer
      */
     @JsxFunction
     public static Object setTimeout(final Context context, final Scriptable scope,
             final Scriptable thisObj, final Object[] args, final Function function) {
         return WindowOrWorkerGlobalScopeMixin.setTimeout(context,
                 ((DedicatedWorkerGlobalScope) thisObj).owningWindow_, args, function);
     }
 
     /**
      * Sets a chunk of JavaScript to be invoked each time a specified number of milliseconds has elapsed.
      *
      * @see <a href="https://developer.mozilla.org/en-US/docs/Web/API/WindowOrWorkerGlobalScope/setInterval">
      * MDN web docs</a>
      * @param context the JavaScript context
      * @param scope the scope
      * @param thisObj the scriptable
      * @param args the arguments passed into the method
      * @param function the function
      * @return the id of the created interval
      */
     @JsxFunction
     public static Object setInterval(final Context context, final Scriptable scope,
             final Scriptable thisObj, final Object[] args, final Function function) {
         return WindowOrWorkerGlobalScopeMixin.setInterval(context,
                 ((DedicatedWorkerGlobalScope) thisObj).owningWindow_, args, function);
     }
 
     /**
      * Returns the prototype object corresponding to the specified HtmlUnit class inside the window scope.
      * @param jsClass the class whose prototype is to be returned
      * @return the prototype object corresponding to the specified class inside the specified scope
      */
     @Override
     public Scriptable getPrototype(final Class<? extends HtmlUnitScriptable> jsClass) {
         return prototypes_.get(jsClass);
     }
 
     /**
      * Sets the prototypes for HtmlUnit host classes.
      * @param map a Map of ({@link Class}, {@link Scriptable})
      */
     public void setPrototypes(final Map<Class<? extends Scriptable>, Scriptable> map) {
         prototypes_ = map;
     }
 }
 
 class WorkerJob extends BasicJavaScriptJob {
     private final ContextFactory contextFactory_;
     private final ContextAction<Object> action_;
     private final String description_;
 
     WorkerJob(final ContextFactory contextFactory, final ContextAction<Object> action, final String description) {
         super();
         contextFactory_ = contextFactory;
         action_ = action;
         description_ = description;
     }
 
     @Override
     public void run() {
         contextFactory_.call(action_);
     }
 
     @Override
     public String toString() {
         return "WorkerJob(" + description_ + ")";
     }
 }