Display Console
This commit is contained in:
parent
7159bae9f7
commit
0bae45a824
@ -1,6 +1,7 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
@ -9,6 +10,7 @@ import (
|
|||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
func removeBOM(input []byte) []byte {
|
func removeBOM(input []byte) []byte {
|
||||||
@ -99,11 +101,8 @@ func downloadFile(item Item, jobInfo *JobInfo) error {
|
|||||||
|
|
||||||
cmd := exec.Command("bash", "-c", command)
|
cmd := exec.Command("bash", "-c", command)
|
||||||
|
|
||||||
jobsMutex.Lock()
|
var outputBuffer bytes.Buffer
|
||||||
jobInfo.Cmd = cmd
|
cmd.Stdout = io.MultiWriter(os.Stdout, &outputBuffer)
|
||||||
jobsMutex.Unlock()
|
|
||||||
|
|
||||||
cmd.Stdout = os.Stdout
|
|
||||||
cmd.Stderr = os.Stderr
|
cmd.Stderr = os.Stderr
|
||||||
|
|
||||||
err = cmd.Start()
|
err = cmd.Start()
|
||||||
@ -116,6 +115,16 @@ func downloadFile(item Item, jobInfo *JobInfo) error {
|
|||||||
done <- cmd.Wait()
|
done <- cmd.Wait()
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
if outputBuffer.Len() > 0 {
|
||||||
|
broadcast(outputBuffer.Bytes())
|
||||||
|
outputBuffer.Reset()
|
||||||
|
}
|
||||||
|
time.Sleep(1 * time.Second)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
select {
|
select {
|
||||||
case <-jobInfo.AbortChan:
|
case <-jobInfo.AbortChan:
|
||||||
if cmd.Process != nil {
|
if cmd.Process != nil {
|
||||||
|
@ -11,6 +11,7 @@ require (
|
|||||||
require (
|
require (
|
||||||
github.com/asticode/go-astikit v0.20.0 // indirect
|
github.com/asticode/go-astikit v0.20.0 // indirect
|
||||||
github.com/asticode/go-astits v1.8.0 // indirect
|
github.com/asticode/go-astits v1.8.0 // indirect
|
||||||
|
github.com/gorilla/websocket v1.5.3
|
||||||
golang.org/x/net v0.0.0-20200904194848-62affa334b73 // indirect
|
golang.org/x/net v0.0.0-20200904194848-62affa334b73 // indirect
|
||||||
golang.org/x/text v0.3.2 // indirect
|
golang.org/x/text v0.3.2 // indirect
|
||||||
)
|
)
|
||||||
|
@ -10,6 +10,8 @@ github.com/beevik/etree v1.4.1 h1:PmQJDDYahBGNKDcpdX8uPy1xRCwoCGVUiW669MEirVI=
|
|||||||
github.com/beevik/etree v1.4.1/go.mod h1:gPNJNaBGVZ9AwsidazFZyygnd+0pAU38N4D+WemwKNs=
|
github.com/beevik/etree v1.4.1/go.mod h1:gPNJNaBGVZ9AwsidazFZyygnd+0pAU38N4D+WemwKNs=
|
||||||
github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
|
github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
|
||||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
|
||||||
|
github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||||
github.com/pkg/exec v0.0.0-20150614095509-0bd164ad2a5a h1:EN123kAtAAE2pg/+TvBsUBZfHCWNNFyL2ZBPPfNWAc0=
|
github.com/pkg/exec v0.0.0-20150614095509-0bd164ad2a5a h1:EN123kAtAAE2pg/+TvBsUBZfHCWNNFyL2ZBPPfNWAc0=
|
||||||
github.com/pkg/exec v0.0.0-20150614095509-0bd164ad2a5a/go.mod h1:b95YoNrAnScjaWG+asr8lxqlrsPUcT2ZEBcjvVGshMo=
|
github.com/pkg/exec v0.0.0-20150614095509-0bd164ad2a5a/go.mod h1:b95YoNrAnScjaWG+asr8lxqlrsPUcT2ZEBcjvVGshMo=
|
||||||
github.com/pkg/profile v1.4.0/go.mod h1:NWz/XGvpEW1FyYQ7fCx4dqYBLlfTcE+A9FLAkNKqjFE=
|
github.com/pkg/profile v1.4.0/go.mod h1:NWz/XGvpEW1FyYQ7fCx4dqYBLlfTcE+A9FLAkNKqjFE=
|
||||||
|
@ -9,6 +9,9 @@ import (
|
|||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/gorilla/websocket"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ProgressInfo struct {
|
type ProgressInfo struct {
|
||||||
@ -354,3 +357,42 @@ func updateProgress(filename string, value float64, currentFile string) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var upgrader = websocket.Upgrader{}
|
||||||
|
var clients = make(map[*websocket.Conn]bool)
|
||||||
|
var mu sync.Mutex
|
||||||
|
|
||||||
|
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
|
||||||
|
conn, err := upgrader.Upgrade(w, r, nil)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error while upgrading connection:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
clients[conn] = true
|
||||||
|
mu.Unlock()
|
||||||
|
|
||||||
|
for {
|
||||||
|
if _, _, err := conn.NextReader(); err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
delete(clients, conn)
|
||||||
|
mu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func broadcast(message []byte) {
|
||||||
|
mu.Lock()
|
||||||
|
defer mu.Unlock()
|
||||||
|
|
||||||
|
for client := range clients {
|
||||||
|
if err := client.WriteMessage(websocket.TextMessage, message); err != nil {
|
||||||
|
client.Close()
|
||||||
|
delete(clients, client)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -77,6 +77,7 @@ func startWebServer() {
|
|||||||
http.HandleFunc("/pause", handlePause)
|
http.HandleFunc("/pause", handlePause)
|
||||||
http.HandleFunc("/resume", handleResume)
|
http.HandleFunc("/resume", handleResume)
|
||||||
http.HandleFunc("/clear-completed", handleClearCompleted)
|
http.HandleFunc("/clear-completed", handleClearCompleted)
|
||||||
|
http.HandleFunc("/ws", handleWebSocket)
|
||||||
|
|
||||||
fmt.Println("Starting web server on http://0.0.0.0:8080")
|
fmt.Println("Starting web server on http://0.0.0.0:8080")
|
||||||
http.ListenAndServe(":8080", nil)
|
http.ListenAndServe(":8080", nil)
|
||||||
|
@ -68,7 +68,7 @@
|
|||||||
#abort-button:hover {
|
#abort-button:hover {
|
||||||
background-color: #d32f2f;
|
background-color: #d32f2f;
|
||||||
}
|
}
|
||||||
#pause-button, #resume-button {
|
#pause-button, #resume-button, #toggle-console {
|
||||||
background-color: #4CAF50;
|
background-color: #4CAF50;
|
||||||
color: white;
|
color: white;
|
||||||
border: none;
|
border: none;
|
||||||
@ -77,7 +77,7 @@
|
|||||||
border-radius: 4px;
|
border-radius: 4px;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
}
|
}
|
||||||
#pause-button:hover, #resume-button:hover {
|
#pause-button:hover, #resume-button:hover, #toggle-console:hover {
|
||||||
background-color: #45a049;
|
background-color: #45a049;
|
||||||
}
|
}
|
||||||
#resume-button {
|
#resume-button {
|
||||||
@ -96,6 +96,17 @@
|
|||||||
#back-button:hover {
|
#back-button:hover {
|
||||||
background-color: #1976D2;
|
background-color: #1976D2;
|
||||||
}
|
}
|
||||||
|
#console {
|
||||||
|
display: none; /* Initially hidden */
|
||||||
|
background-color: black;
|
||||||
|
color: white;
|
||||||
|
height: 300px; /* Adjust height as needed */
|
||||||
|
overflow-y: scroll;
|
||||||
|
white-space: pre; /* Preserve whitespace */
|
||||||
|
font-family: monospace; /* Use monospace font */
|
||||||
|
margin-top: 10px;
|
||||||
|
border: 1px solid #ccc;
|
||||||
|
}
|
||||||
@media (max-width: 600px) {
|
@media (max-width: 600px) {
|
||||||
body {
|
body {
|
||||||
padding: 10px;
|
padding: 10px;
|
||||||
@ -128,8 +139,10 @@
|
|||||||
<button id="abort-button" onclick="abortDownload()">Abort Download</button>
|
<button id="abort-button" onclick="abortDownload()">Abort Download</button>
|
||||||
<button id="pause-button" onclick="pauseDownload()">Pause Download</button>
|
<button id="pause-button" onclick="pauseDownload()">Pause Download</button>
|
||||||
<button id="resume-button" onclick="resumeDownload()" style="display: none;">Resume Download</button>
|
<button id="resume-button" onclick="resumeDownload()" style="display: none;">Resume Download</button>
|
||||||
|
<button id="toggle-console">Toggle Console View</button>
|
||||||
<button id="back-button" onclick="window.location.href='/'">Back to Index</button>
|
<button id="back-button" onclick="window.location.href='/'">Back to Index</button>
|
||||||
</div>
|
</div>
|
||||||
|
<div id="console"></div>
|
||||||
<script>
|
<script>
|
||||||
let isPaused = false;
|
let isPaused = false;
|
||||||
|
|
||||||
@ -203,6 +216,22 @@
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const consoleDiv = document.getElementById('console');
|
||||||
|
const ws = new WebSocket(`ws://${window.location.host}/ws`);
|
||||||
|
|
||||||
|
ws.onmessage = function(event) {
|
||||||
|
consoleDiv.textContent += event.data;
|
||||||
|
consoleDiv.scrollTop = consoleDiv.scrollHeight;
|
||||||
|
};
|
||||||
|
|
||||||
|
document.getElementById('toggle-console').onclick = function() {
|
||||||
|
if (consoleDiv.style.display === "none") {
|
||||||
|
consoleDiv.style.display = "block";
|
||||||
|
} else {
|
||||||
|
consoleDiv.style.display = "none";
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
updateProgress();
|
updateProgress();
|
||||||
</script>
|
</script>
|
||||||
</body>
|
</body>
|
||||||
|
Loading…
Reference in New Issue
Block a user