import org.openqa.selenium.*;

import java.io.File;

/**
 * 自定义截图
 *
 * @author erichary
 */
public class CustomScreenshot {

    /**
     * 向下滑动次数
     */
    private int scrollTimes = 10;

    /**
     * 上次高度
     */
    private int lastHeight = 0;

    /**
     * 重置窗口大小(调整至可以正常截图)
     *
     * @param driver 驱动对象
     */
    private void resetWindowSizeToScreenshot(WebDriver driver) {
        // 窗口最大化
        resetWindowSize(driver);
        // 向下滑动页面：到指定次数 || 高度不再变化，退出
        JavascriptExecutor javascriptExecutor = (JavascriptExecutor) driver;
        for (int i = 0; i < this.scrollTimes; i++) {
            // 获取当前高度
            Object thisHeightObject = javascriptExecutor.executeScript("return document.body.scrollHeight;");
            int thisHeight = Integer.parseInt(String.valueOf(thisHeightObject));
            // 判断高度
            if (this.lastHeight != thisHeight) {
                // 向下滑动
                javascriptExecutor.executeScript("window.scrollBy(0,10000)");
                // 滑动后赋值
                this.lastHeight = thisHeight;
            } else {
                // 高度相同，跳出
                break;
            }
        }
        // 设置窗口高度
        Dimension size = driver.manage().window().getSize();
        driver.manage().window().setSize(new Dimension(size.width, this.lastHeight));
    }

    /**
     * 重置窗口大小(最大化窗口)
     *
     * @param driver 驱动对象
     */
    private void resetWindowSize(WebDriver driver) {
        driver.manage().window().maximize();
    }

    /**
     * 全屏截图（当前可视范围）
     *
     * @param driver 驱动对象
     * @return 截图内容
     */
    public File fullScreenshot(WebDriver driver) {
        // 调整窗口
        resetWindowSize(driver);
        // 截图
        File file = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        // 调整窗口
        resetWindowSize(driver);
        return file;
    }

    /**
     * 全屏截图（长图）
     *
     * @param driver 驱动对象
     * @return 截图内容
     */
    public File fullScreenshotLong(WebDriver driver) {
        // 调整窗口
        resetWindowSizeToScreenshot(driver);
        // 截图
        File file = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        // 调整窗口
        resetWindowSize(driver);
        return file;
    }

    /**
     * 局域截图（单图）
     *
     * @param driver  驱动对象
     * @param element 要截图的文档节点对象
     * @return 截图内容
     */
    public File localScreenshot(WebDriver driver, WebElement element) {
        // 调整窗口
        resetWindowSizeToScreenshot(driver);
        // 截图
        File file = element.getScreenshotAs(OutputType.FILE);
        // 调整窗口
        resetWindowSize(driver);
        return file;
    }

    public int getScrollTimes() {
        return scrollTimes;
    }

    public void setScrollTimes(int scrollTimes) {
        this.scrollTimes = scrollTimes;
    }

    public int getLastHeight() {
        return lastHeight;
    }

    public void setLastHeight(int lastHeight) {
        this.lastHeight = lastHeight;
    }
}

/**
 * 首字母转大写、驼峰命名
 *
 * @param s 待转换的字符串
 * @return 转换结果
 */
public static String initialToCapital(String s) {
    StringBuilder sb = new StringBuilder();
    if (s == null || s.trim().isEmpty()) {
        return sb.toString();
    }
    if (s.length() <= 1) {
        return sb.append(s).toString().toUpperCase();
    }
    String[] split = s.split("_");
    for (String string : split) {
        sb.append(string.substring(0, 1).toUpperCase());
        sb.append(string.substring(1).toLowerCase());
    }
    return sb.toString();
}

/**
 * 首字母转小写、驼峰转下划线
 *
 * @param s 待转换的字符串
 * @return 转换结果
 */
public static String capitalToInitial(String s) {
    StringBuilder sb = new StringBuilder();
    if (s == null || s.trim().isEmpty()) {
        return sb.toString();
    }
    if (s.length() <= 1) {
        return sb.append(s).toString().toLowerCase();
    }
    for (int i = 0; i < s.length(); i++) {
        char charAt = s.charAt(i);
        if (Character.isUpperCase(charAt)) {
            sb.append("_");
            sb.append(Character.toLowerCase(charAt));
        } else {
            sb.append(charAt);
        }
    }
    return sb.toString();
}

下载安装jdk

百度搜索jdk+版本，以1.8版本为例，百度搜索“jdk1.8”，一般是第一个。

百度搜索jdk1.8

打开jdk下载页面，这里我们下载jdk1.8 x64版本下载地址：https://www.oracle.com/java/technologies/javase/javase-jdk8-downloads.html

jdk1.8下载页面

下载历史版本的jdk自动跳转登录，需要登录你的Oracle账号才能下载，用邮箱注册登录就可以，登录成功自动下载。

输入账号密码登录

双击安装包，打开安装程序，下一步

安装引导

选择安装选项，一般默认即可。可以自定义安装位置，这里我们选择默认，下一步

安装选项、安装位置

jdk安装中……

jdk安装中……

jdk安装完成后，自动打开安装jre窗口，可以自定义安装位置，这里我们保持默认，下一步

配置jre安装位置

jre安装中……

jre安装中……

安装成功，点击关闭

安装成功

配置jdk环境变量

右键我的电脑 > 属性 > 高级系统设置

找到高级系统设置

选择环境变量

选择环境变量

在系统变量内，加上jdk环境变量系统变量 -> 新建JAVA_HOME变量系统变量 -> 查找path变量 -> 变量后输入%JAVA_HOME%\bin;%JAVA_HOME%\jre\bin;（如果变量结尾没有`;`，先加上`;`，然后输入变量）系统变量 -> 新建CLASSPATH变量 -> 变量值.;%JAVA_HOME%\lib;%JAVA_HOME%\lib\tools.jar（注意前面有个点）Win+R运行cmd，输入java -version （java 和 -version 之间有空格），正确输出版本号，安装配置成功。

验证jdk是否安装成功

咖啡

这会是一个美好的开端吧！