Merge pull request #12 from jeremiah-k/main

Cleaning the slate, getting ready for a plugin system
pull/1/head
Jeremiah K 2023-04-24 20:12:19 -05:00 zatwierdzone przez GitHub
commit b7447955ef
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
5 zmienionych plików z 5 dodań i 134 usunięć

2
.gitignore vendored
Wyświetl plik

@ -2,4 +2,4 @@
.vscode
config.yaml
meshtastic.sqlite
__pycache__
./plugins/__pycache__

57
main.py
Wyświetl plik

@ -264,61 +264,10 @@ def truncate_message(
# Callback for new messages in Matrix room
async def on_room_message(
room: MatrixRoom, event: Union[RoomMessageText, RoomMessageNotice]
) -> None:
if event.server_timestamp < bot_start_time:
return
body = event.body.strip()
match = re.match(r"^.*: !ping$", body.strip())
if match:
# Respond to the command
await matrix_client.room_send(
room.room_id, "m.room.message", {"msgtype": "m.text", "body": "pong!"}
)
return
pattern = r"^.*:(?: !map(?: zoom=(\d+))?(?: size=(\d+),(\d+))?)?$"
match = re.match(pattern, body.strip())
if match:
zoom = match.group(1)
image_size = match.group(2, 3)
try:
zoom = int(zoom)
except:
zoom = 8
if zoom < 0 or zoom > 30:
zoom = 8
try:
image_size = (int(image_size[0]), int(image_size[1]))
except:
image_size = (1000, 1000)
if image_size[0] > 1000 or image_size[1] > 1000:
image_size = (1000, 1000)
from map import get_map
locations = []
for node, info in meshtastic_interface.nodes.items():
if "position" in info and "latitude" in info["position"]:
locations.append(
{
"lat": info["position"]["latitude"],
"lon": info["position"]["longitude"],
}
)
pillow_image = get_map(locations=locations, zoom=zoom, image_size=image_size)
from matrix import send_image
await send_image(matrix_client, room.room_id, pillow_image)
return
room: MatrixRoom, event: Union[RoomMessageText, RoomMessageNotice]) -> None:
full_display_name = "Unknown user"
if event.sender != bot_user_id:
message_timestamp = event.server_timestamp
@ -429,4 +378,4 @@ async def main():
await asyncio.sleep(60) # Update longnames every 60 seconds
asyncio.run(main())
asyncio.run(main())

41
map.py
Wyświetl plik

@ -1,41 +0,0 @@
import staticmaps # pip install py-staticmaps
import math
import random
def anonymize_location(lat, lon, radius=1000):
# Generate random offsets for latitude and longitude
lat_offset = random.uniform(-radius / 111320, radius / 111320)
lon_offset = random.uniform(
-radius / (111320 * math.cos(lat)), radius / (111320 * math.cos(lat))
)
# Apply the offsets to the location coordinates
new_lat = lat + lat_offset
new_lon = lon + lon_offset
return new_lat, new_lon
def get_map(locations, zoom=None, image_size=None, radius=10000):
"""
Anonymize a location to 10km by default
"""
context = staticmaps.Context()
context.set_tile_provider(staticmaps.tile_provider_OSM)
context.set_zoom(zoom)
for location in locations:
new_location = anonymize_location(
lat=float(location["lat"]),
lon=float(location["lon"]),
radius=radius,
)
radio = staticmaps.create_latlng(new_location[0], new_location[1])
context.add_object(staticmaps.Marker(radio, size=10))
# render non-anti-aliased png
if image_size:
return context.render_pillow(image_size[0], image_size[1])
else:
return context.render_pillow(1000, 1000)

Wyświetl plik

@ -1,36 +0,0 @@
import os
import io
import aiofiles.os
from PIL import Image
from nio import AsyncClient, UploadResponse
import base64
async def upload_image(client: AsyncClient, image: Image.Image) -> UploadResponse:
buffer = io.BytesIO()
image.save(buffer, format="PNG")
image_data = buffer.getvalue()
response, maybe_keys = await client.upload(
io.BytesIO(image_data),
content_type="image/png",
filename="location.png",
filesize=len(image_data),
)
return response
async def send_room_image(
client: AsyncClient, room_id: str, upload_response: UploadResponse
):
response = await client.room_send(
room_id=room_id,
message_type="m.room.message",
content={"msgtype": "m.image", "url": upload_response.content_uri, "body": ""},
)
async def send_image(client: AsyncClient, room_id: str, image: Image.Image):
response = await upload_image(client=client, image=image)
await send_room_image(client, room_id, upload_response=response)

Wyświetl plik

@ -1,3 +1,2 @@
meshtastic==2.1.6
matrix-nio==0.20.2
py-staticmaps==0.4.0
matrix-nio==0.20.2