java计算公司加班时间

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;

import java.io.*;
import java.net.*;
import java.util.ArrayList;
import java.util.Scanner;

/**
 * @author: zhaoxu
 * @date: 2020/9/12 12:55
 */
public class CalculateTime {
    public ArrayList<String> xiabanlist = new ArrayList<String>();
    public ArrayList<String> shangbanlist = new ArrayList<String>();
    public ArrayList<String> yueri = new ArrayList<String>();
    String bc_date = "";
    String bc_date_end = "";
    int sum = 0;
    public ArrayList<Integer> index = new ArrayList<Integer>();
    byte[] buffer;
    String all;
    int length;
    String lt = "";

    public void login(String username, String password) {
        final String loginURL = "登录验证地址";
        String form = "portal_username=" + username + "&password=" + password + "&lt=xxx&_eventId=submit&username=" + username;
        final String headerAgent = "User-Agent";
        final String headerAgentArg = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.135 Safari/537.36";
        try {
            CookieManager manager = new CookieManager();
            manager.setCookiePolicy(CookiePolicy.ACCEPT_ORIGINAL_SERVER);
            CookieHandler.setDefault(manager);
            HttpURLConnection httpURLConnection = (HttpURLConnection) (new URL(loginURL).openConnection());
            httpURLConnection.setRequestMethod("GET");
            httpURLConnection.setRequestProperty(headerAgent, headerAgentArg);
            httpURLConnection.connect();
            if (httpURLConnection.getResponseCode() == 200) {
                InputStream inputStream = httpURLConnection.getInputStream();
                buffer = new byte[1024];
                while ((length = inputStream.read(buffer)) != -1) {
                    all = all + new String(buffer, 0, length, "UTF-8");
                }
                httpURLConnection.disconnect();
                // 通过第一次打开的页面获取lt
                lt = all.split("<input type=\"hidden\" name=\"lt\" value=\"")[1].split("\" />")[0];
                all = null;
                httpURLConnection = (HttpURLConnection) (new URL(loginURL).openConnection());
                httpURLConnection.setRequestMethod("POST");
                httpURLConnection.setRequestProperty(headerAgent, headerAgentArg);
                httpURLConnection.setDoOutput(true);
                httpURLConnection.connect();
                httpURLConnection.getOutputStream().write(form.replace("xxx", lt).getBytes("UTF-8"));
                httpURLConnection.getInputStream();
                httpURLConnection.disconnect();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    public String calculate(String year1, String month1) throws IOException {
        xiabanlist.clear();
        shangbanlist.clear();
        yueri.clear();
        if (month1.equals("01") || month1.equals("03") || month1.equals("05") || month1.equals("07") || month1.equals("08") || month1.equals("10") || month1.equals("12")) {
            bc_date = year1 + "-" + month1 + "-01";
            bc_date_end = year1 + "-" + month1 + "-31";
        } else if (month1.equals("04") || month1.equals("06") || month1.equals("09") || month1.equals("11")) {
            bc_date = year1 + "-" + month1 + "-01";
            bc_date_end = year1 + "-" + month1 + "-30";
        } else if (Integer.valueOf(year1) % 4 == 0 && Integer.valueOf(year1) % 100 != 0 || Integer.valueOf(year1) % 400 == 0) {
            bc_date = year1 + "-" + month1 + "-01";
            bc_date_end = year1 + "-" + month1 + "-29";
        } else {
            bc_date = year1 + "-" + month1 + "-01";
            bc_date_end = year1 + "-" + month1 + "-28";
        }

        String url = "获取打卡数据地址";
        String dateJson = getJson(url);
        JSONObject json = JSONObject.parseObject(dateJson);
        JSONArray list = JSONArray.parseArray(json.get("jsonList").toString());
        String sblist = "";
        String xblist = "";
        for (int i = 0; i < list.size(); i++) {
            String times = list.get(i).toString();
            JSONObject times1 = JSONObject.parseObject(times);
            String status = times1.get("status").toString();
            String riqi = "";
            String xbtime = "";
            String sbtime = "";
            if (status == "1" && i == 0) {
                continue;
            }
            //如果当前节点是上班节点
            else if (status.equals("1")) {
                //如果当前日期的上班节点还没有查找成功
                if (xiabanlist.size() == shangbanlist.size() + 1) {
                    sblist = list.get(i).toString();
                    JSONObject sb = JSONObject.parseObject(sblist);
                    sbtime = sb.get("kq_time").toString();
                    //如果今天上班未打卡，忽略这一天的打卡记录
                    if (!sbtime.split(" ")[0].equals(yueri.get(yueri.size() - 1))) {
                        yueri.remove(yueri.size() - 1);
                        xiabanlist.remove(xiabanlist.size() - 1);
                        continue;
                    }
                    //如果当前节点拥有下个节点
                    if (i + 1 < list.size()) {
                        String next = list.get(i + 1).toString();
                        JSONObject ne = JSONObject.parseObject(next);
                        //如果下个节点是下班节点
                        if (!ne.get("status").toString().equals("1") && (sb.get("addr_status") == null || !sb.get("addr_status").toString().equals("1"))) {
                            //判断当前节点是否脱岗打卡
                            System.out.println(sbtime + "上班");
                            shangbanlist.add(sbtime);
                        }
                        //如果下个节点是脱岗打卡
                        else if (ne.get("addr_status").toString().equals("1")) {
                            System.out.println(sbtime + "上班");
                            shangbanlist.add(sbtime);
                        }
                    }
                    //当前节点为最后一个节点
                    else {
                        System.out.println(sbtime + "上班");
                        shangbanlist.add(sbtime);
                    }
                }
            }
            //如果当前节点是下班节点
            else if (status.equals("5")) {
                if (xiabanlist.size() + 1 == shangbanlist.size() + 1) {
                    xblist = list.get(i).toString();
                    JSONObject xb = JSONObject.parseObject(xblist);
                    xbtime = xb.get("kq_time").toString();
                    riqi = xb.get("bc_date").toString();
                    if (i + 1 < list.size()) {
                        String next = list.get(i + 1).toString();
                        JSONObject ne = JSONObject.parseObject(next);
                        String now = xb.get("kq_time").toString().split(" ")[0];
                        String n = ne.get("kq_time").toString().split(" ")[0];
                        //如果下个节点不是当前日期的节点
                        if (!n.equals(now)) {
                            continue;
                        }
                        //如果当前节点为非脱岗节点
                        else if (xb.get("addr_status") == null || !xb.get("addr_status").toString().equals("1")) {
                            System.out.println(xbtime + "下班");
                            xiabanlist.add(xbtime);
                            yueri.add(riqi);
                        }
                    }
                }
            }
            //如果当前节点是周六/周日的打卡记录节点
            else if (status.equals("6")) {
                if (xiabanlist.size() == shangbanlist.size() + 1) {
                    //当前节点拥有下一个节点
                    sblist = list.get(i).toString();
                    JSONObject sb = JSONObject.parseObject(sblist);
                    sbtime = sb.get("kq_time").toString();
                    //如果今天上班未打卡，忽略这一天的打卡记录
                    if (!sbtime.split(" ")[0].equals(yueri.get(yueri.size() - 1))) {
                        yueri.remove(yueri.size() - 1);
                        xiabanlist.remove(xiabanlist.size() - 1);
                        continue;
                    }
                    if (i + 1 < list.size()) {
                        String next = list.get(i + 1).toString();
                        JSONObject ne = JSONObject.parseObject(next);
                        //如果下一个节点为脱岗打卡
                        if (ne.get("address_status") != null && ne.get("addr_status").toString().equals("1")) {
                            System.out.println(sbtime + "上班");
                            shangbanlist.add(sbtime);
                        } else {
                            String now = sb.get("kq_time").toString().split(" ")[0];
                            String n = ne.get("kq_time").toString().split(" ")[0];
                            //如果下个节点不是当前日期的节点
                            if (!n.equals(now)) {
                                sbtime = sb.get("kq_time").toString();
                                System.out.println(sbtime + "上班");
                                shangbanlist.add(sbtime);
                            }
                        }
                    } else {
                        System.out.println(sbtime + "上班");
                        shangbanlist.add(sbtime);
                    }
                }
                //如果当前节点为第一个下班节点
                else if (xiabanlist.size() + 1 == shangbanlist.size() + 1) {
                    xblist = list.get(i).toString();
                    JSONObject xb = JSONObject.parseObject(xblist);
                    xbtime = xb.get("kq_time").toString();
                    riqi = xb.get("bc_date").toString();

                    if (i + 1 < list.size()) {
                        String next = list.get(i + 1).toString();
                        JSONObject ne = JSONObject.parseObject(next);
                        String now = xb.get("kq_time").toString().split(" ")[0];
                        String n = ne.get("kq_time").toString().split(" ")[0];
                        //如果下个节点不是当前日期的节点
                        if (!n.equals(now)) {
                            continue;
                        }
                        //如果当前节点为非脱岗节点
                        else if (xb.get("addr_status") == null || !xb.get("addr_status").toString().equals("1")) {
                            System.out.println(xbtime + "下班");
                            xiabanlist.add(xbtime);
                            yueri.add(riqi);
                            index.add(yueri.size() - 1);
                        }
                    }
                }
            }
        }
        int index2 = 0;
        for (int i = 0; i < yueri.size(); i++) {
            Integer hour = Integer.valueOf(shangbanlist.get(i).split(" ")[1].split(":")[0]);
            Integer minute = Integer.valueOf(shangbanlist.get(i).split(" ")[1].split(":")[1]);
            Integer second = Integer.valueOf(shangbanlist.get(i).split(" ")[1].split(":")[2]);
            Integer hour1 = Integer.valueOf(xiabanlist.get(i).split(" ")[1].split(":")[0]);
            Integer minute1 = Integer.valueOf(xiabanlist.get(i).split(" ")[1].split(":")[1]);
            Integer second1 = Integer.valueOf(xiabanlist.get(i).split(" ")[1].split(":")[2]);
            //将所有的秒加起来
            Integer sum1 = hour1 * 60 * 60 + minute1 * 60 + second1 - hour * 60 * 60 - minute * 60 - second - 9 * 60 * 60 - 30 * 60;
            //如果当前节点为周六/周日节点
            if (index2 < index.size() && i == index.get(index2)) {
                sum = sum + hour1 * 60 * 60 + minute1 * 60 + second1 - hour * 60 * 60 - minute * 60 - second;
                index2++;
            } else if (sum1 >= 30 * 60) {
                sum = sum + sum1;
            }
        }
        //计算小时/分钟/秒
        int xiaoshi = sum / 3600;
        int fenzhong = sum % 3600 / 60;
        int miao = sum % 3600 % 60;

        System.out.println("您的加班时间为: " + xiaoshi + "小时" + fenzhong + "分钟" + miao + "秒");

        return "您在" + year1 + "年" + month1 + "月" + "的加班时间为: " + xiaoshi + "小时" + fenzhong + "分钟" + miao + "秒";
    }

    private String getJson(String url) throws IOException {
        Document doc = Jsoup.connect(url)
                .ignoreContentType(true)
                .get();
        return doc.select("body").text();
    }

    public static void main(String[] args) throws IOException {
        File file = new File("1.txt");
        Scanner in = new Scanner(System.in);
        if(file.exists()){
            BufferedReader bufferedReader = new BufferedReader(new FileReader(file));
            new CalculateTime().login(bufferedReader.readLine(), bufferedReader.readLine());
            new CalculateTime().calculate(bufferedReader.readLine(), bufferedReader.readLine());
        }
        else{
            System.out.println("请输入您的用户名：");
            String username = in.nextLine();
            System.out.println("请输入您的密码：");
            String password = in.nextLine();
            System.out.println("请输入要查询的年份：");
            String year = in.nextLine();
            System.out.println("请输入要查询的月份：");
            String month = in.nextLine();
            new CalculateTime().login(username, password);
            new CalculateTime().calculate(year, month);
        }

        System.out.println("按任意键退出。。。");
        in.nextLine();
        System.exit(0);
    }
}