/**
 * 控制臺(tái)進(jìn)度條打印工具
 */
public class ProgressBarPrintingUtils {

    /**
     * 使用示例
     */
    public static void main(String[] args) throws InterruptedException {
        // 1.進(jìn)度條打?。ㄖ苯觽鲄ⅲ?
        System.out.println("進(jìn)度條打印1");
        printProgressBar(100, 50, 30, '=', null, null, null);
        System.out.println();

        // 2.進(jìn)度條打印（通過(guò)進(jìn)度條對(duì)象傳參）
        System.out.println("\n進(jìn)度條打印2");
        printProgressBar(new ProgressBarBuilder()
                        .total(100)
                        .completed(80)
                        .length(30)
                        .character('#')
                        .state(State.ERROR)
                        .name("任務(wù)一")
                        .description("")
                        .build());
        System.out.println();

        // 3.動(dòng)態(tài)效果演示
        System.out.println("\n動(dòng)態(tài)效果演示");
        Random r = new Random();
        ProgressBar progressBar = createProgressBarBuilder()
                                .total(100)
                                .completed(0)
                                .length(30)
                                .character('█')
                                .name("解析文件內(nèi)容")
                                .build();
        while (true) {
            printProgressBar(progressBar);
            if (progressBar.completed >= progressBar.total) {
                break;
            }
            Thread.sleep(500);
            progressBar.completed += r.nextInt(10);
            if (progressBar.getCompleted() >= progressBar.getTotal()) {
                progressBar.setCompleted(progressBar.getTotal());
                progressBar.setState(State.SUCCESS);
                progressBar.setDescription("解析完成");
            }
        }
        System.out.println();
    }

    /**
     * 打印進(jìn)度條
     *
     * @param total 任務(wù)總數(shù)
     * @param completed 已完成任務(wù)數(shù)量
     * @param length 進(jìn)度條的長(zhǎng)度，單位為字符
     * @param character 填充進(jìn)度進(jìn)度條字符
     * @param state 進(jìn)度條的狀態(tài)，用于在控制臺(tái)顯示不同的文字顏色
     *
     */
    public static void printProgressBar(int total, int completed, int length, char character,
                                        State state, String name, String description) {
        System.out.print("\r");
        System.out.print(getColorStartTagByState(state));
        if (name != null) {
            System.out.print(name);
            System.out.print(" ");
        }
        System.out.print("[");
        int ratio = completed >= total ? length : (int) Math.floor(completed / (double)total * length);
        for (int i = 1; i <= ratio; i++) {
            System.out.print(character);
        }
        for (int i = 1; i <= length - ratio; i++) {
            System.out.print(" ");
        }
        System.out.print("] ");
        System.out.print(completed);
        System.out.print("/");
        System.out.print(total);
        System.out.print(" ");
        if (description != null) {
            System.out.print(description);
        }
        if (state != null) {
            System.out.print("\033[0m");
        }
    }

    /**
     * 打印進(jìn)度條
     *
     * @param progressBar 進(jìn)度條配置信息
     */
    public static void printProgressBar(ProgressBar progressBar) {
        if (progressBar == null) {
            return;
        }
        printProgressBar(progressBar.total, progressBar.completed, progressBar.length, progressBar.character,
                        progressBar.state, progressBar.name, progressBar.description);
    }

    /**
     * 創(chuàng)建進(jìn)度條構(gòu)造器
     */
    public static ProgressBarBuilder createProgressBarBuilder() {
        return new ProgressBarBuilder();
    }

    /**
     * 通過(guò)狀態(tài)枚舉獲取顏色開(kāi)始標(biāo)簽
     *
     * @param state 狀態(tài)
     */
    private static String getColorStartTagByState(State state) {
        if (state == null) {
            return "";
        }
        switch (state) {
            case ERROR:
                return "\033[31m";
            case SUCCESS:
                return "\033[32m";
            case WARNING:
                return "\033[33m";
            default:
                return "";
        }
    }

    /**
     * 進(jìn)度條狀態(tài)枚舉類(lèi)
     */
    public static enum State {
        // 正?；蚰J(rèn)
        NORMAL,
        // 警告
        WARNING,
        // 錯(cuò)誤
        ERROR,
        // 成功
        SUCCESS
    }

    /**
     * 進(jìn)度條類(lèi)
     */
    public static class ProgressBar {
        private int total;
        private int completed;
        private int length;
        private char character;
        private State state;
        private String name;
        private String description;

        private ProgressBar(ProgressBarBuilder builder) {
            this.total = builder.total;
            this.completed = builder.completed;
            this.length = builder.length;
            this.character = builder.character;
            this.state = builder.state;
            this.name = builder.name;
            this.description = builder.description;
        }

        public int getTotal() {
            return total;
        }

        public void setTotal(int total) {
            this.total = total;
        }

        public int getCompleted() {
            return completed;
        }

        public void setCompleted(int completed) {
            this.completed = completed;
        }

        public int getLength() {
            return length;
        }

        public void setLength(int length) {
            this.length = length;
        }

        public char getCharacter() {
            return character;
        }

        public void setCharacter(char character) {
            this.character = character;
        }

        public State getState() {
            return state;
        }

        public void setState(State state) {
            if (state == null) {
                this.state = State.NORMAL;
                return;
            }
            this.state = state;
        }

        public String getName() {
            return name;
        }

        public void setName(String name) {
            if (name == null) {
                this.name = "";
            }
            this.name = name;
        }

        public String getDescription() {
            return description;
        }

        public void setDescription(String description) {
            if (description == null) {
                this.description = "";
            }
            this.description = description;
        }
    }

    /**
     * 進(jìn)度條構(gòu)造類(lèi)
     */
    public static class ProgressBarBuilder {
        private int total;
        private int completed;
        private int length;
        private char character;
        private State state;
        private String name;
        private String description;

        public ProgressBarBuilder() {
            this.total = 100;
            this.completed = 0;
            this.length = 20;
            this.character = '=';
            this.state = State.NORMAL;
            this.name = "";
            this.description = "";
        }

        public ProgressBarBuilder total(int total) {
            this.total = total;
            return this;
        }

        public ProgressBarBuilder completed(int completed) {
            this.completed = completed;
            return this;
        }

        public ProgressBarBuilder length(int length) {
            this.length = length;
            return this;
        }

        public ProgressBarBuilder character(char character) {
            this.character = character;
            return this;
        }

        public ProgressBarBuilder state(State state) {
            this.state = state;
            return this;
        }

        public ProgressBarBuilder name(String name) {
            if (name == null) {
                name = "";
            }
            this.name = name;
            return this;
        }

        public ProgressBarBuilder description(String description) {
            if (description == null) {
                description = "";
            }
            this.description = description;
            return this;
        }

        public ProgressBar build() {
            return new ProgressBar(this);
        }
    }
}