Skip to content

Feat/mode activation #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
SOORAJTS2001 merged 5 commits into from
Aug 16, 2025
Merged

Feat/mode activation #3

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
43a4a5a
feat: Introduce web sockets as the communication mode, since it is fa…
SOORAJTS2001 Aug 15, 2025
bb5cca8
fix: Avoided hardcoding browser height and width and now sending only…
SOORAJTS2001 Aug 15, 2025
6674152
interim commit
SOORAJTS2001 Aug 15, 2025
e28ba51
fix: Update poetry dependency also add emoji
SOORAJTS2001 Aug 15, 2025
28fd727
feat: Add mode activation for Rage, Wander and Shadow from extension …
SOORAJTS2001 Aug 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
102 changes: 102 additions & 0 deletions app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,102 @@
# ruff : noqa
import socket
from contextlib import asynccontextmanager

import qrcode
import uvicorn
from fastapi import FastAPI, WebSocket, WebSocketDisconnect
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
from rich.columns import Columns
from rich.console import Console
from rich.panel import Panel

PORT = 8000
HOST = "0.0.0.0"
RELOAD = False
console = Console()


def get_wifi_ip():
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
s.connect(("8.8.8.8", 80))
ip = s.getsockname()[0]
finally:
s.close()
return ip


def get_server_url():
ip = get_wifi_ip()
return f"http://{ip}:{PORT}"


def generate_qr_ascii(url: str) -> str:
qr = qrcode.QRCode(border=1)
qr.add_data(url)
qr.make()
# Capture ASCII QR into a string
import io, sys

buf = io.StringIO()
sys_stdout = sys.stdout
sys.stdout = buf
qr.print_ascii(invert=True)
sys.stdout = sys_stdout
return buf.getvalue()


@asynccontextmanager
async def lifespan(app: FastAPI):
url = get_server_url()
mobile_page_url = f"{url}/mobile_page"
qr_ascii = generate_qr_ascii(mobile_page_url)

qr_panel = Panel.fit(qr_ascii, title="Scan to Open", border_style="green")
steps_panel = Panel.fit(
"\n".join(
[
f"[bold cyan]1.[/bold cyan] Connect to the same Wi-Fi network",
f"[bold cyan]2.[/bold cyan] Scan the QR code",
f"[bold cyan] [/bold cyan] Or [yellow]{mobile_page_url}[/yellow]",
f"[bold cyan]3.[/bold cyan] Go to the scroll area and enjoy!",
]
),
title="Steps",
border_style="blue",
)

console.print(Columns([qr_panel, steps_panel]))
yield


app = FastAPI(lifespan=lifespan)
app.mount("/static", StaticFiles(directory="mobile_page"), name="static")

# Store connected clients
connected_clients: list[WebSocket] = []


@app.get("/mobile_page")
async def get_page2():
with open("mobile_page/index.html") as f:
return HTMLResponse(f.read())


@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
await websocket.accept()
connected_clients.append(websocket)
try:
while True:
data = await websocket.receive_text()
for client in connected_clients:
if client != websocket:
await client.send_text(data)
except WebSocketDisconnect:
connected_clients.remove(websocket)


if __name__ == "__main__":
uvicorn.run("app:app", host=HOST, port=PORT, reload=RELOAD)
26 changes: 26 additions & 0 deletions browser_extension/content.js
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
function injectWhenReady() {
if (!document.head || !document.body) {
// Retry until DOM is ready
requestAnimationFrame(injectWhenReady);
return;
}
const css = document.createElement('link');
css.rel = 'stylesheet';
css.href = chrome.runtime.getURL('runtime/pyscript.css');
document.head.appendChild(css);

const pyscriptJs = document.createElement('script');
pyscriptJs.src = chrome.runtime.getURL('runtime/pyscript.js');
pyscriptJs.defer = true;
document.head.appendChild(pyscriptJs);

pyscriptJs.onload = () => {
fetch(chrome.runtime.getURL('main.py'))
.then(res => res.text())
.then(code => {
const pyTag = document.createElement('py-script');
pyTag.textContent = code;
document.body.appendChild(pyTag);
});
}}
injectWhenReady()
Loading