com.machinepublishers.jbrowserdriver.Context.java Source code

Java tutorial

Introduction

Here is the source code for com.machinepublishers.jbrowserdriver.Context.java

Source

/*
 * jBrowserDriver (TM)
 * Copyright (C) 2014-2016 Machine Publishers, LLC and the jBrowserDriver contributors
 * https://github.com/MachinePublishers/jBrowserDriver
 *
 * 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
 *
 *     http://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 com.machinepublishers.jbrowserdriver;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.rmi.RemoteException;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.Capabilities;

import com.machinepublishers.jbrowserdriver.AppThread.Sync;

class Context {
    final AtomicBoolean initialized = new AtomicBoolean();
    final AtomicReference<TimeoutsServer> timeouts = new AtomicReference<TimeoutsServer>();
    final AtomicReference<TargetLocatorServer> targetLocator = new AtomicReference<TargetLocatorServer>();
    final AtomicReference<OptionsServer> options = new AtomicReference<OptionsServer>();
    final AtomicReference<KeyboardServer> keyboard = new AtomicReference<KeyboardServer>();
    final AtomicReference<MouseServer> mouse = new AtomicReference<MouseServer>();
    final AtomicReference<Capabilities> capabilities = new AtomicReference<Capabilities>();
    final AtomicReference<NavigationServer> navigation = new AtomicReference<NavigationServer>();
    final AtomicReference<AlertServer> alert = new AtomicReference<AlertServer>();
    final AtomicReference<Robot> robot = new AtomicReference<Robot>();
    final AtomicReference<File> userDataDirectory = new AtomicReference<>();
    private final Map<String, ContextItem> itemMap = new LinkedHashMap<String, ContextItem>();
    private final List<ContextItem> items = new ArrayList<ContextItem>();
    private int current = 0;
    private final AtomicReference<File> autoUserDataDirectory = new AtomicReference<>();;
    private final Object lock = new Object();

    Context() {
        synchronized (lock) {
            //UserData dir must be computed before any ContextItem is initialized.
            setUpUserDataDir();
            ContextItem newContext = new ContextItem();
            items.add(newContext);
            itemMap.put(newContext.itemId.get(), newContext);
            try {
                timeouts.set(new TimeoutsServer());
                alert.set(new AlertServer(timeouts));
            } catch (RemoteException e) {
                Util.handleException(e);
            }
        }
    }

    private void setUpUserDataDir() {
        FileUtils.deleteQuietly(autoUserDataDirectory.get());
        autoUserDataDirectory.set(null);
        if (SettingsManager.settings().userDataDirectory() == null) {
            //Temporary because cookies aren't saved too.
            //App has its own temp dir, so this won't collide with other instances.
            try {
                userDataDirectory.set(Files.createTempDirectory("jbd_userdata_").toFile());
                autoUserDataDirectory.set(userDataDirectory.get());
            } catch (IOException e) {
                Util.handleException(e);
            }
        } else {
            userDataDirectory.set(SettingsManager.settings().userDataDirectory());
            userDataDirectory.get().mkdirs();
        }
    }

    void reset(JBrowserDriverServer driver) {
        removeItems();
        synchronized (lock) {
            setUpUserDataDir();
            ContextItem newContext = new ContextItem();
            newContext.init(driver, this);
            items.add(newContext);
            itemMap.put(newContext.itemId.get(), newContext);
        }
    }

    void init(final JBrowserDriverServer driver) {
        synchronized (lock) {
            if (!items.isEmpty()) {
                items.get(current).init(driver, this);
            }
            if (initialized.compareAndSet(false, true)) {
                robot.set(new Robot(this));
                try {
                    targetLocator.set(new TargetLocatorServer(driver, this));
                    keyboard.set(new KeyboardServer(robot));
                    mouse.set(new MouseServer(robot));
                    navigation.set(new NavigationServer(new AtomicReference<JBrowserDriverServer>(driver), this));
                    options.set(new OptionsServer(this, timeouts));
                } catch (RemoteException e) {
                    Util.handleException(e);
                }
            }
        }
    }

    ContextItem item() {
        synchronized (lock) {
            if (current < items.size()) {
                return items.get(current);
            }
        }
        return null;
    }

    String itemId() {
        return AppThread.exec(new Sync<String>() {
            @Override
            public String perform() {
                synchronized (lock) {
                    return items.get(current).itemId.get();
                }
            }
        });
    }

    Set<String> itemIds() {
        return AppThread.exec(new Sync<Set<String>>() {
            @Override
            public Set<String> perform() {
                synchronized (lock) {
                    return new LinkedHashSet<String>(itemMap.keySet());
                }
            }
        });
    }

    ContextItem spawn(final JBrowserDriverServer driver) {
        final Context thisObj = this;
        return AppThread.exec(new Sync<ContextItem>() {
            @Override
            public ContextItem perform() {
                synchronized (lock) {
                    ContextItem newContext = new ContextItem();
                    newContext.init(driver, thisObj);
                    newContext.stage.get().toBack();
                    items.add(newContext);
                    itemMap.put(newContext.itemId.get(), newContext);
                    return newContext;
                }
            }
        });
    }

    void setCurrent(final String id) {
        AppThread.exec(new Sync<Object>() {
            @Override
            public Object perform() {
                synchronized (lock) {
                    current = items.indexOf(itemMap.get(id));
                    items.get(current).stage.get().toFront();
                    return null;
                }
            }
        });
    }

    void removeItem() {
        AppThread.exec(new Sync<Object>() {
            @Override
            public Object perform() {
                synchronized (lock) {
                    items.get(current).stage.get().close();
                    itemMap.remove(items.remove(current).itemId.get());
                    current = 0;
                    return null;
                }
            }
        });
    }

    void removeItem(final String itemId) {
        AppThread.exec(new Sync<Object>() {
            @Override
            public Object perform() {
                synchronized (lock) {
                    itemMap.remove(itemId).stage.get().close();
                    items.remove(itemId);
                    current = 0;
                    return null;
                }
            }
        });
    }

    void removeItems() {
        AppThread.exec(new Sync<Object>() {
            @Override
            public Object perform() {
                synchronized (lock) {
                    for (ContextItem curItem : items) {
                        curItem.stage.get().close();
                    }
                    items.clear();
                    itemMap.clear();
                    current = 0;
                    return null;
                }
            }
        });
    }
}