CAT_TOM/pyasic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added the scan page to scan for miners on a subnet

Browse Source
This commit is contained in:
UpstreamData
2022-02-25 15:58:01 -07:00
parent da5a784214
commit cf3163dccf
5 changed files with 147 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
tools/web_monitor/app.py
View File

@@ -3,9 +3,11 @@ import asyncio
import uvicorn
from fastapi import FastAPI
from fastapi import Request
from fastapi import WebSocket
from fastapi import WebSocket, WebSocketDisconnect
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
from tools.web_monitor.miner_factory import miner_factory
from network import MinerNetwork
app = FastAPI()
@@ -13,10 +15,47 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")
@app.route("/")
@app.get("/")
def dashboard(request: Request):
return templates.TemplateResponse("index.html", {"request": request})
@app.get("/scan")
def scan(request: Request):
return templates.TemplateResponse("scan.html", {"request": request})
@app.websocket("/scan/ws")
async def websocket_scan(websocket: WebSocket):
await websocket.accept()
try:
while True:
network_ip = await websocket.receive_text()
if "/" in network_ip:
network_ip, network_subnet = network_ip.split("/")
network = MinerNetwork(network_ip, mask=network_subnet)
else:
network = MinerNetwork(network_ip)
miner_generator = network.scan_network_generator()
miners = []
async for miner in miner_generator:
if miner:
miners.append(miner)
get_miner_genenerator = miner_factory.get_miner_generator(miners)
all_miners = []
async for found_miner in get_miner_genenerator:
all_miners.append({"ip": found_miner.ip, "model": await found_miner.get_model()})
all_miners.sort(key=lambda x: x["ip"])
send_miners = []
for miner in all_miners:
send_miners.append({"ip": str(miner["ip"]), "model": miner["model"]})
await websocket.send_json(send_miners)
await websocket.send_text("Done")
except WebSocketDisconnect:
pass
if __name__ == "__main__":
uvicorn.run("app:app", host="127.0.0.1", port=80, debug=True)
uvicorn.run("app:app", host="127.0.0.1", port=80)

7
tools/web_monitor/index.py
View File

@@ -1,7 +0,0 @@
from tools.web_monitor.app import sio
@sio.event
async def connect(sid, _environ) -> None:
"""Event for connection"""
await sio.emit('init', "hello")

7
tools/web_monitor/miner_factory.py Normal file
View File

@@ -0,0 +1,7 @@
"""
This file stores the MinerFactory instance used by the WebMonitor for use in other files.
"""
from miners.miner_factory import MinerFactory
miner_factory = MinerFactory()

4
tools/web_monitor/templates/navbar.html
View File

@@ -107,11 +107,11 @@
</nav>
<main class="col-md-9 ms-sm-auto col-lg-10 px-md-4"><div class="chartjs-size-monitor"><div class="chartjs-size-monitor-expand"><div class=""></div></div><div class="chartjs-size-monitor-shrink"><div class=""></div></div></div>
<div class="col-md-9 ms-sm-auto col-lg-10 px-md-4 ps-4">
{% block content %}
{% endblock content %}
</main>
</div>
</div>
</div>
</body>

96
tools/web_monitor/templates/scan.html Normal file
View File

@@ -0,0 +1,96 @@
{% extends 'navbar.html'%}
{% block content %}
<div class="row w-100 my-4">
<form action="" onsubmit="sendMessage(event)">
<div class="input-group mb-3">
<span class="input-group-text" id="scan-ip">Subnet</span>
<input type="text" class="form-control" id="messageText" placeholder="192.168.1.1/24" aria-describedby="scan-ip">
<span class="input-group-text"><button>Scan</button></span>
</div>
</form>
</div>
<div class="row w-100">
<table class="table table-striped table-responsive" style="max-height:300px;">
<thead class="table-dark">
<tr>
<th class="active col-1">
<input type="checkbox" class="select-all checkbox" name="select-all" id="selectAllCheckbox"/>
</th>
<th>IP</th>
<th>Model</th>
<th id="scanStatus" class="col-2"></th>
</tr>
</thead>
<tbody id="minerTable">
</tbody>
</table>
</div>
<script src="https://code.jquery.com/jquery-3.3.1.slim.min.js" integrity="sha384-q8i/X+965DzO0rT7abK41JStQIAqVgRVzpbzo5smXKp4YfRvH+8abtTE1Pi6jizo" crossorigin="anonymous"></script>
<script>
$(function(){
//column checkbox select all or cancel
$("input.select-all").click(function () {
var checked = this.checked;
$("input.select-item").each(function (index,item) {
item.checked = checked;
});
});
//check selected items
$("input.select-item").click(function () {
var checked = this.checked;
var all = $("input.select-all")[0];
var total = $("input.select-item").length;
var len = $("input.select-item:checked:checked").length;
all.checked = len===total;
});
});
</script>
<script>
var ws = new WebSocket("ws://localhost:80/scan/ws");
ws.onmessage = function(event) {
if (event.data == "Done") {
document.getElementById("selectAllCheckbox").disabled = false
document.getElementById("scanStatus").innerHTML = document.getElementById('minerTable').rows.length + " Miners"
} else {
document.getElementById("selectAllCheckbox").disabled = true
document.getElementById("scanStatus").innerHTML = "Scanning..."
};
var miner_data = JSON.parse(event.data)
var miners = document.getElementById('minerTable')
miners.innerHTML = ""
miner_data.forEach(function(miner) {
console.log(miner)
var tr = document.createElement('tr')
var checkbox_td = document.createElement('td')
checkbox_td.innerHTML = '<input type="checkbox" class="select-item checkbox" name="select-item" value="' + miner["ip"] + '" />'
checkbox_td.className = "active"
var ip_td = document.createElement('td')
ip_td.innerHTML = miner["ip"]
var model_td = document.createElement('td')
model_td.innerHTML = miner["model"]
var empty_td = document.createElement('td')
tr.append(checkbox_td)
tr.append(ip_td)
tr.append(model_td)
tr.append(empty_td)
miners.append(tr)
});
};
function sendMessage(event) {
var input = document.getElementById("messageText")
document.getElementById("scanStatus").innerHTML = "Scanning..."
ws.send(input.value)
input.value = ''
event.preventDefault()
}
</script>
{% endblock content %}