com.redhat.darcy.webdriver.internal.WebDriverCookieManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of darcy-webdriver Show documentation
An implementation of darcy and darcy-web that uses Selenium WebDriver as the automation library backend.
The newest version!
/*
 Copyright 2014 Red Hat, Inc. and/or its affiliates.

 This file is part of darcy-webdriver.

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see .
 */

package com.redhat.darcy.webdriver.internal;

import com.redhat.darcy.web.Cookie;
import com.redhat.darcy.web.api.CookieManager;

import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.Date;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;

public class WebDriverCookieManager implements CookieManager {
    private final TargetedWebDriver driver;

    public WebDriverCookieManager(TargetedWebDriver driver) {
        this.driver = driver;
    }

    @Override
    public void add(Cookie cookie) {
        cookie.validate();

        driver.manage().addCookie(transformToSeleniumCookie(cookie));
    }

    @Override
    public void delete(String name) {
        driver.manage().deleteCookieNamed(name);
    }

    @Override
    public void deleteAll() {
        driver.manage().deleteAllCookies();
    }

    @Override
    public Set getAll() {
        return driver.manage().getCookies().stream()
                .map(this::transformToDarcyCookie)
                .collect(Collectors.toSet());
    }

    @Override
    public Optional get(String name) {
        return Optional.ofNullable(driver.manage().getCookieNamed(name))
                .map(this::transformToDarcyCookie);
    }

    private org.openqa.selenium.Cookie transformToSeleniumCookie(Cookie cookie) {
        //From java.time.LocalDateTime to java.util.Date
        Date expiry = Optional.ofNullable(cookie.getExpiry())
                .map(e -> Date.from(e.atZone(ZoneId.systemDefault()).toInstant()))
                .orElse(null);

        return new org.openqa.selenium.Cookie(cookie.getName(), cookie.getValue(),
                cookie.getDomain(), cookie.getPath(), expiry, cookie.isSecure(),
                cookie.isHttpOnly());
    }

    private Cookie transformToDarcyCookie(org.openqa.selenium.Cookie cookie) {
        // From java.util.Date to java.time.LocalDateTime
        LocalDateTime expiry = Optional.ofNullable(cookie.getExpiry())
                .map(Date::toInstant)
                .map(e -> LocalDateTime.ofInstant(e, ZoneId.systemDefault()))
                .orElse(null);

        return new Cookie(cookie.getName(), cookie.getValue(),
                cookie.getDomain(), cookie.getPath(), expiry, cookie.isSecure(),
                cookie.isHttpOnly());
    }
}