XiaomiADBFastbootTools/src/main/java/xiaomiadbfastboottools/Command.java at 95139e23a88738682e59ba40c4170a119bacea65 · LouWii/XiaomiADBFastbootTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package xiaomiadbfastboottools;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Scanner;
import javafx.scene.control.TextInputControl;

/**
 *
 * @author Saki
 */
public class Command {
    protected ProcessBuilder pb;
    protected Process proc;
    protected Scanner scan;
    protected String output;
    protected TextInputControl tic;
    protected String prefix;

    public Command(){
        pb = new ProcessBuilder();
        pb.directory(new File(System.getProperty("user.dir") + "/temp"));
        tic = null;
        if (System.getProperty("os.name").toLowerCase().contains("linux"))
        	prefix = "";
        else if (System.getProperty("os.name").toLowerCase().contains("mac os"))
        	prefix = "./";
        else prefix = "temp/";
    }

    public Command(TextInputControl control){
        pb = new ProcessBuilder();
        pb.directory(new File(System.getProperty("user.dir") + "/temp"));
        tic = control;
        if (System.getProperty("os.name").toLowerCase().contains("linux"))
        	prefix = "";
        else if (System.getProperty("os.name").toLowerCase().contains("mac os"))
        	// prefix = "./";
        	prefix = "";
        else prefix = "temp/";
    }

    public String exec(String arg){
        pb.command(Arrays.asList((prefix + arg).split(" ")));
        pb.redirectErrorStream(true);
        output = "";
        if (tic != null)
            tic.setText("");
        try {
            proc = pb.start();
        } catch (IOException ex) {
            System.out.println("ERROR: Couldn't start process");
            System.out.println(ex.getMessage());
        }
        scan = new Scanner(proc.getInputStream());
        String line;
        while (scan.hasNext()){
            line = scan.nextLine() + System.lineSeparator();
            output += line;
            if (tic != null)
             tic.appendText(line);
        }
        scan.close();
        return output;
    }

    public String exec(String arg, boolean err){
        pb.command(Arrays.asList((prefix + arg).split(" ")));
        pb.redirectErrorStream(false);
        output = "";
        if (tic != null)
            tic.setText("");
        try {
            proc = pb.start();
        } catch (IOException ex) {
            System.out.println("ERROR: Couldn't start process");
            System.out.println(ex.getMessage());
        }
        if (err)
            scan = new Scanner(proc.getErrorStream());
                    else
            scan = new Scanner(proc.getInputStream());
        String line;
        while (scan.hasNext()) {
            line = scan.nextLine() + System.lineSeparator();
            output += line;
            if (tic != null) {
                tic.appendText(line);
            }
        }
        scan.close();
        return output;
    }

    public String exec(String... args){
        pb.redirectErrorStream(true);
        output = "";
        if (tic != null)
            tic.setText("");
        for (String s : args){
            pb.command(Arrays.asList((prefix + s).split(" ")));
            try {
                proc = pb.start();
            } catch (IOException ex) {
                System.out.println("ERROR: Couldn't start process");
            }
            scan = new Scanner(proc.getInputStream());
            String line;
            while (scan.hasNext()) {
                line = scan.nextLine() + System.lineSeparator();
                output += line;
                if (tic != null)
                    tic.appendText(line);
            }
            scan.close();
            try {
                proc.waitFor();
            } catch (InterruptedException ex) {
                System.out.println("ERROR: The process was interrupted.");
            }
        }
        return output;
    }

    public void waitFor(){
        try {
            proc.waitFor();
        } catch (InterruptedException ex) {
            System.out.println("ERROR: The process was interrupted.");
        }
    }

    public void kill(){
        proc.destroy();
        if (proc.isAlive())
            proc.destroyForcibly();
    }
}