package jp.dip.arimodoki.connector;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;
import java.util.concurrent.TimeUnit;

import org.openqa.selenium.By;
import org.openqa.selenium.Keys;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.edge.EdgeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.ie.InternetExplorerDriver;
import org.openqa.selenium.interactions.Actions;
import org.openqa.selenium.opera.OperaDriver;
import org.openqa.selenium.remote.DesiredCapabilities;

import cucumber.api.java.Before;

/**
 * WebDriver操作クラス
 */
public class WebConnector {

    private final static long DEFAULT_TIMEOUT = 5000;

    // ブラウザ名
    private final String BROWSER_IE = "IE";
    private final String BROWSER_EDGE = "Edge";
    private final String BROWSER_FF = "FireFox";
    private final String BROWSER_OPERA = "Opera";
    private final String BROWSER_CR = "Chrome";
    // ブラウザタイプ
    private final int BROWSER_TYPE_IE = 0;
    private final int BROWSER_TYPE_EDGE = 1;
    private final int BROWSER_TYPE_FF = 2;
    private final int BROWSER_TYPE_OPERA = 3;
    private final int BROWSER_TYPE_CR = 4;

    /** WebDriverクラス */
    private WebDriver driver = null;

    /** 実行中のWebDriverタイプを保持する */
    private int DriverType;
    /** Screen Shot FolderName */
    private String screenShotPath = null;
    private Actions builder = null;
    /**
     * コンストラクタ
     */
    public WebConnector() {}

    /**
     * WebDriverの選択
     * @param browser ブラウザ名
     */
    public void selectWebDriver(String browser) {
        this.DriverType = BROWSER_TYPE_CR;
        if (browser.equals(BROWSER_IE)) {
            this.DriverType = BROWSER_TYPE_IE;
        } else if (browser.equals(BROWSER_EDGE)) {
            this.DriverType = BROWSER_TYPE_EDGE;
        } else if (browser.equals(BROWSER_FF)) {
            this.DriverType = BROWSER_TYPE_FF;
        } else if (browser.equals(BROWSER_OPERA)) {
            this.DriverType = BROWSER_TYPE_OPERA;
        }
        //WebDriverのセット
        this.setWebDriver();
    }

    /**
     * WebDriverインスタンスを生成する
     */
    private void setWebDriver() {
        switch (this.DriverType) {
        case BROWSER_TYPE_IE: // IE
            //Windows10では、 64bit 版だと動かないので32bitを使う
            System.setProperty("webdriver.ie.driver", "./exe32bit/IEDriverServer.exe");

            this.driver = new InternetExplorerDriver();
            break;

        case BROWSER_TYPE_EDGE: // Edge
            System.setProperty("webdriver.edge.driver", "./exe/MicrosoftWebDriver.exe");

            this.driver = new EdgeDriver();
            break;

        case BROWSER_TYPE_FF: // FireFox
            System.setProperty("webdriver.gecko.driver", "./exe64bit/geckodriver.exe");
            this.driver = new FirefoxDriver();
            break;

        case BROWSER_TYPE_OPERA: // Opera
            System.setProperty("webdriver.opera.driver", "./exe64bit/operadriver.exe");
            this.driver = new OperaDriver();
            break;

        default: // Chrome
            System.setProperty("webdriver.chrome.driver", "./exe/chromedriver.exe");
            //ChromeでのWindow最大化はちょっとコツがいる
            DesiredCapabilities cap = DesiredCapabilities.chrome();
            ChromeOptions options = new ChromeOptions();
            options.addArguments("start-maximized");
            cap.setCapability(ChromeOptions.CAPABILITY, options);
            //ChromeでのWindow最大化はちょっとコツがいる
            this.driver = new ChromeDriver(cap);
            break;
        }
        this.builder =  new Actions(this.driver);
        //暗黙wait
        this.driver.manage().timeouts().implicitlyWait(DEFAULT_TIMEOUT, TimeUnit.MILLISECONDS);
    }

    /**
     * Windowを最大化する
     * ただし、ChromeDriver以外
     */
    public void setWindowMax() {
        //ただし、Chromeではorg.openqa.selenium.NoSuchSessionExceptionで動かない）
        if(this.DriverType != BROWSER_TYPE_CR) {
            //Chrome ドライバ以外では、最大化する ⇒ Chrome はドライバインスタンス生成時に最大化オプション指定
            try {
                this.driver.manage().window().maximize();
            } catch (Exception e) {
                this.driver.quit();
            }
        }
    }

    /**
     * 指定時間待つ
     * @param sec waitする秒数
     */
    public void sleep(int sec) {
        long msec = sec * 1000;
        try {
            Thread.sleep(msec);
        } catch (Exception e) {
        }
    }

    /**
     * iFrame を切り替える
     * @param sec waitする秒数
     */
    public WebDriver switchFrame(String selector) {
        return this.driver.switchTo().frame(selector);
    }

    /**
     * フレームが表示されるまで待つ
     * DEFAULT_TIMEOUTで表示されなかったら Exceptionをスローする
     * @param iFrameのIDセレクタ
     * @param フレームが表示された後で、フレーム内で探すセレクタ(ID->name->class順）
     * 　　　　　targetClass iFrame内の class名
     * FireFoxだとダメみたいです。
     */
    public void sleepFrame(String selector, String targetClass) {
        String parentHandle = this.driver.getWindowHandle();
        logger.log_info(this, "WindowHandle["+ parentHandle+"]");
        //iFrame エレメントを取得
        WebElement iframe = this.driver.findElement(By.id(selector));
        logger.log_info(this, "iframe["+ iframe.getLocation()+"]");
        //iFrame Switch
        this.driver.switchTo().frame(iframe);
        WebElement element =this.driver.findElement(By.className(targetClass));
        //親画面に戻る
        this.driver.switchTo().window(parentHandle);
    }

    public void parentWindow() {
        String parentHandle = this.driver.getWindowHandle();
        //親画面に戻る
        this.driver.switchTo().window(parentHandle);
    }

    /**
     * スクリーンショット画像格納フォルダを指定する
     * @param path フォルダ名
     */
    public void setScreenShotPath(String path) {
        this.screenShotPath = path;
    }

    @Before
    public void initSelenium() throws Exception {
    }

    /**
     * 終了処理
     */
    public void destroySelenium() {
        //WebDriver プロセスを終了し、ブラウザを閉じる
        this.driver.quit();
    }

    /**
     * 指定されたURLでブラウザを開く
     * @param location 表示するURL
     */
    public void openAndWait(String location) {
        //location(URL)のタイプミスとかでもエラーにはならない
        this.driver.navigate().to(location);
    }

    /**
     * ブラウザのスクリーンショットを実行する
     * @param filename 保存するファイル名（パスは付けない）
     */
    public void getScreenShot(String filename) {
        String path = this.screenShotPath + "/" + filename;
        /**
         * Chromeではスクショ　できません、2.9では
         * 色々やってみたけど、Dhromeでは Exceptionで動作しないため、無効化
         */
        if(this.DriverType != BROWSER_TYPE_CR) {
            TakesScreenshot screen = (TakesScreenshot)this.driver;
            Path capture = Paths.get(path);
            try {
                Files.write(capture, screen.getScreenshotAs(OutputType.BYTES));
            } catch(Exception e) {
                this.driver.quit();
            }
        }
    }

    /**
     * セレクタに値を送信する
     * @param selector id or name属性セレクタ
     * @param value 送信する値
     */
    public void inputAndWait(String selector, String value) {
        try {
            WebElement element = this.driver.findElement(By.id(selector));
            element.sendKeys(value);
            return;
        } catch(Exception e) {
        }
        try {
            WebElement element = this.driver.findElement(By.name(selector));
            element.sendKeys(value);
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * セレクタにEnterを送信する
     * @param selector id or name属性セレクタ
     * @param value 送信する値
     */
    public void inputEnterAndWait(String selector) {
        try {
            WebElement element = this.driver.findElement(By.id(selector));
            element.sendKeys(Keys.RETURN);
            return;
        } catch(Exception e) {
        }
        try {
            WebElement element = this.driver.findElement(By.name(selector));
            element.sendKeys(Keys.RETURN);
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * ボタンセレクタをクリックする
     * @param selector [input type]セレクタ
     */
    public void btnClickAndWait(String value) {
        try {
            WebElement element = this.driver.findElement(
                By.xpath("//input[@type=\"submit\" and @value=\"" + value + "\"]"));
            element.click();
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * 複数のボタンの内、value属性がlabelのボタンをクリックする
     * @param type [input type] "button"  or "submit"
     * @param label value属性値
     */
    public void btnClickAndWait(String type, String label) {
        try {
            List element = this.driver.findElements(By.xpath("//input[@type=\"" + type + "\"]"));
            for(WebElement entity : element) {
                if(entity.getAttribute("value").equals(label)) {
                    entity.click();
                    break;
                }
            }
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * 規則的に並ぶ複数のボタンの内、value属性がvalueのindexで指定された順番のボタンをクリックする
     * 複数のボタン配列の特定のvalueのボタンをクリックする
     * @param selector [input type]セレクタ
     * @param value value属性値
     * @param index 配列の順番(0オリジン)
     */
    public void btnByblockClickAndWait(String selector, String value, int index) {
        int i = 0;
        try {
            List element = this.driver.findElements(
                    By.xpath("//input[@type=\"" + selector + "\" and @value=\"" + value + "\"]"));
            for(WebElement entity : element) {
                if(i == index) {
                    entity.click();
                    break;
                }
                i++;
            }
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * 複数のボタンの内 selector名の name属性のボタンをクリックする
     * @param selector
     */
    public void btnBynameClickAndWait(String selector) {
        try {
            WebElement element = this.driver.findElement(By.name(selector));
            element.click();
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * href リンクをクリックする
     * @param text 検索したいリンクテキスト
     */
    public void clickAndWait(String text) {
        try {
            //エレメントから、a href のリンク文字列を探す
            WebElement element = this.driver.findElement(By.linkText(text));
            element.click();
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * href リンクをクリックする
     * @param href リンクテキスト
     */
    public void clickHrefAndWait(String href) {
        try {
            //エレメントから、a href のリンクを探す
            WebElement element = this.driver.findElement(By.xpath("//a[@href='"+href+"']"));
            element.click();    //見つかったらクリック
        } catch(Exception e) {
        }
    }

    /**
     * 指定されたテキストが、現在のページ内にあるかチェックする
     * @param text 検索対象のテキスト
     * @return テキストが見つかれば true, 見つからなければ false を返す
     */
    public boolean isTextPresent(String text) {
        WebElement content = this.driver.findElement(By.tagName("body"));
        boolean res = content.getText().contains(text);
        return res;
    }

    /**
     * spanタグをクリックする
     * @param tagname "span"を期待
     * @param text タグから検索したい文字列
     */
    public void spanClickAndWait(String tagname, String text) {
        int i = 0;
        try {
            //tagnameの名前のエレメントを探す
            List element = this.driver.findElements(By.tagName(tagname));
            for(WebElement entity : element) {
                //エレメントのテキストが検索したい文字列と合致
                if(entity.getText().equals(text)) {
                    entity.click(); //エレメントをクリックする
                    break;
                }
            }
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    /**
     * aタグをクリックする
     * @param text a タグのテキスト
     */
    public void linkClickAndWait(String text) {
        int i = 0;
        try {
            WebElement element = this.driver.findElement(By.linkText(text));
            element.click();
        } catch(Exception e) {
            this.driver.quit();
        }
    }

    public void assertTable(String className, List expect) {

        WebElement table = this.driver.findElement(By.className(className));

        List rows = table.findElements(By.tagName("tr"));
        int columnCount = rows.get(0).findElements(By.xpath("./*")).size();
        assertThat(rows.size() * columnCount, is(expect.size()));

        int ix = 0;
        for (WebElement row : rows) {

            List cells = row.findElements(By.xpath("./*"));
            for (WebElement cell : cells) {
                assertThat(cell.getText(), is(expect.get(ix)));
                ix++;
            }
        }
    }

}