Compare commits
32 Commits
b9b31a494a
...
f0e68cb31e
| Author | SHA1 | Date | |
|---|---|---|---|
| f0e68cb31e | |||
| 8fe372532b | |||
| 214ac33049 | |||
| 7b245869d5 | |||
| f5a2928f1b | |||
| f74146bb11 | |||
| f6157bf879 | |||
| 704adf6fe8 | |||
| f2575e04fb | |||
| ba7bbbfc62 | |||
| 1db9947267 | |||
| 38d05da272 | |||
| c740de95f9 | |||
| 483a63ede9 | |||
| a817b9b61d | |||
| 0c6da9845c | |||
| 586c653e4f | |||
| c8cc411aa5 | |||
| 71d967114d | |||
| 659c30f376 | |||
| 70d3b3b019 | |||
| 0b8c324944 | |||
| cbbaa22f56 | |||
| fb754cad92 | |||
| 0518cdce0b | |||
| 1af739cac2 | |||
| b28ba3c47d | |||
| fadc57a7c7 | |||
| cca3946a35 | |||
| 18be3fdc19 | |||
| 939e63f244 | |||
| b4c267d584 |
@ -1,3 +1,7 @@
|
|||||||
|
import logging
|
||||||
|
|
||||||
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
|
||||||
import pathlib
|
import pathlib
|
||||||
import shutil
|
import shutil
|
||||||
import sqlite3
|
import sqlite3
|
||||||
|
|||||||
@ -14,6 +14,9 @@ from snek.view.threads import ThreadsView
|
|||||||
logging.basicConfig(level=logging.DEBUG)
|
logging.basicConfig(level=logging.DEBUG)
|
||||||
from concurrent.futures import ThreadPoolExecutor
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
from ipaddress import ip_address
|
from ipaddress import ip_address
|
||||||
|
import time
|
||||||
|
import uuid
|
||||||
|
|
||||||
|
|
||||||
import IP2Location
|
import IP2Location
|
||||||
from aiohttp import web
|
from aiohttp import web
|
||||||
@ -32,7 +35,6 @@ from snek.sgit import GitApplication
|
|||||||
from snek.sssh import start_ssh_server
|
from snek.sssh import start_ssh_server
|
||||||
from snek.system import http
|
from snek.system import http
|
||||||
from snek.system.cache import Cache
|
from snek.system.cache import Cache
|
||||||
from snek.system.stats import middleware as stats_middleware, create_stats_structure, stats_handler
|
|
||||||
from snek.system.markdown import MarkdownExtension
|
from snek.system.markdown import MarkdownExtension
|
||||||
from snek.system.middleware import auth_middleware, cors_middleware, csp_middleware
|
from snek.system.middleware import auth_middleware, cors_middleware, csp_middleware
|
||||||
from snek.system.profiler import profiler_handler
|
from snek.system.profiler import profiler_handler
|
||||||
@ -42,6 +44,7 @@ from snek.system.template import (
|
|||||||
PythonExtension,
|
PythonExtension,
|
||||||
sanitize_html,
|
sanitize_html,
|
||||||
)
|
)
|
||||||
|
from snek.view.new import NewView
|
||||||
from snek.view.about import AboutHTMLView, AboutMDView
|
from snek.view.about import AboutHTMLView, AboutMDView
|
||||||
from snek.view.avatar import AvatarView
|
from snek.view.avatar import AvatarView
|
||||||
from snek.view.channel import ChannelAttachmentView,ChannelAttachmentUploadView, ChannelView
|
from snek.view.channel import ChannelAttachmentView,ChannelAttachmentUploadView, ChannelView
|
||||||
@ -129,10 +132,7 @@ async def trailing_slash_middleware(request, handler):
|
|||||||
class Application(BaseApplication):
|
class Application(BaseApplication):
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
middlewares = [
|
middlewares = [
|
||||||
stats_middleware,
|
|
||||||
cors_middleware,
|
cors_middleware,
|
||||||
web.normalize_path_middleware(merge_slashes=True),
|
|
||||||
ip2location_middleware,
|
|
||||||
csp_middleware,
|
csp_middleware,
|
||||||
]
|
]
|
||||||
self.template_path = pathlib.Path(__file__).parent.joinpath("templates")
|
self.template_path = pathlib.Path(__file__).parent.joinpath("templates")
|
||||||
@ -171,18 +171,11 @@ class Application(BaseApplication):
|
|||||||
self.ip2location = IP2Location.IP2Location(
|
self.ip2location = IP2Location.IP2Location(
|
||||||
base_path.joinpath("IP2LOCATION-LITE-DB11.BIN")
|
base_path.joinpath("IP2LOCATION-LITE-DB11.BIN")
|
||||||
)
|
)
|
||||||
self.on_startup.append(self.prepare_stats)
|
|
||||||
self.on_startup.append(self.prepare_asyncio)
|
self.on_startup.append(self.prepare_asyncio)
|
||||||
self.on_startup.append(self.start_user_availability_service)
|
self.on_startup.append(self.start_user_availability_service)
|
||||||
self.on_startup.append(self.start_ssh_server)
|
self.on_startup.append(self.start_ssh_server)
|
||||||
self.on_startup.append(self.prepare_database)
|
self.on_startup.append(self.prepare_database)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def prepare_stats(self, app):
|
|
||||||
app['stats'] = create_stats_structure()
|
|
||||||
print("Stats prepared", flush=True)
|
|
||||||
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def uptime_seconds(self):
|
def uptime_seconds(self):
|
||||||
@ -272,6 +265,8 @@ class Application(BaseApplication):
|
|||||||
name="static",
|
name="static",
|
||||||
show_index=True,
|
show_index=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self.router.add_view("/new.html", NewView)
|
||||||
self.router.add_view("/profiler.html", profiler_handler)
|
self.router.add_view("/profiler.html", profiler_handler)
|
||||||
self.router.add_view("/container/sock/{channel_uid}.json", ContainerView)
|
self.router.add_view("/container/sock/{channel_uid}.json", ContainerView)
|
||||||
self.router.add_view("/about.html", AboutHTMLView)
|
self.router.add_view("/about.html", AboutHTMLView)
|
||||||
@ -319,7 +314,6 @@ class Application(BaseApplication):
|
|||||||
#self.router.add_view("/drive.json", DriveApiView)
|
#self.router.add_view("/drive.json", DriveApiView)
|
||||||
#self.router.add_view("/drive.html", DriveView)
|
#self.router.add_view("/drive.html", DriveView)
|
||||||
#self.router.add_view("/drive/{drive}.json", DriveView)
|
#self.router.add_view("/drive/{drive}.json", DriveView)
|
||||||
self.router.add_get("/stats.html", stats_handler)
|
|
||||||
self.router.add_view("/stats.json", StatsView)
|
self.router.add_view("/stats.json", StatsView)
|
||||||
self.router.add_view("/user/{user}.html", UserView)
|
self.router.add_view("/user/{user}.html", UserView)
|
||||||
self.router.add_view("/repository/{username}/{repository}", RepositoryView)
|
self.router.add_view("/repository/{username}/{repository}", RepositoryView)
|
||||||
@ -374,6 +368,8 @@ class Application(BaseApplication):
|
|||||||
|
|
||||||
# @time_cache_async(60)
|
# @time_cache_async(60)
|
||||||
async def render_template(self, template, request, context=None):
|
async def render_template(self, template, request, context=None):
|
||||||
|
start_time = time.perf_counter()
|
||||||
|
|
||||||
channels = []
|
channels = []
|
||||||
if not context:
|
if not context:
|
||||||
context = {}
|
context = {}
|
||||||
@ -434,10 +430,12 @@ class Application(BaseApplication):
|
|||||||
|
|
||||||
self.jinja2_env.loader = self.original_loader
|
self.jinja2_env.loader = self.original_loader
|
||||||
|
|
||||||
|
end_time = time.perf_counter()
|
||||||
|
print(f"render_template took {end_time - start_time:.4f} seconds")
|
||||||
|
|
||||||
# rendered.text = whitelist_attributes(rendered.text)
|
# rendered.text = whitelist_attributes(rendered.text)
|
||||||
# rendered.headers['Content-Lenght'] = len(rendered.text)
|
# rendered.headers['Content-Lenght'] = len(rendered.text)
|
||||||
return rendered
|
return rendered
|
||||||
|
|
||||||
async def static_handler(self, request):
|
async def static_handler(self, request):
|
||||||
file_name = request.match_info.get("filename", "")
|
file_name = request.match_info.get("filename", "")
|
||||||
|
|
||||||
|
|||||||
@ -2,9 +2,19 @@ from snek.system.service import BaseService
|
|||||||
from snek.system.template import sanitize_html
|
from snek.system.template import sanitize_html
|
||||||
import time
|
import time
|
||||||
import asyncio
|
import asyncio
|
||||||
from concurrent.futures import ThreadPoolExecutor
|
from concurrent.futures import ProcessPoolExecutor
|
||||||
|
import json
|
||||||
|
|
||||||
|
from jinja2 import Environment, FileSystemLoader
|
||||||
|
global jinja2_env
|
||||||
|
import pathlib
|
||||||
|
template_path = pathlib.Path(__file__).parent.parent.joinpath("templates")
|
||||||
|
|
||||||
|
|
||||||
|
def render(context):
|
||||||
|
template =jinja2_env.get_template("message.html")
|
||||||
|
return sanitize_html(template.render(**context))
|
||||||
|
|
||||||
executor = ThreadPoolExecutor(max_workers=50)
|
|
||||||
|
|
||||||
|
|
||||||
class ChannelMessageService(BaseService):
|
class ChannelMessageService(BaseService):
|
||||||
@ -13,9 +23,26 @@ class ChannelMessageService(BaseService):
|
|||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
self._configured_indexes = False
|
self._configured_indexes = False
|
||||||
|
self._executor_pools = {}
|
||||||
|
global jinja2_env
|
||||||
|
jinja2_env = self.app.jinja2_env
|
||||||
|
self._max_workers = 1
|
||||||
|
def get_or_create_executor(self, uid):
|
||||||
|
if not uid in self._executor_pools:
|
||||||
|
self._executor_pools[uid] = ProcessPoolExecutor(max_workers=self._max_workers)
|
||||||
|
print("Executors available", len(self._executor_pools))
|
||||||
|
return self._executor_pools[uid]
|
||||||
|
|
||||||
|
def delete_executor(self, uid):
|
||||||
|
if uid in self._executor_pools:
|
||||||
|
self._executor_pools[uid].shutdown()
|
||||||
|
del self._executor_pools[uid]
|
||||||
|
|
||||||
async def maintenance(self):
|
async def maintenance(self):
|
||||||
args = {}
|
args = {}
|
||||||
|
|
||||||
|
|
||||||
|
return
|
||||||
for message in self.mapper.db["channel_message"].find():
|
for message in self.mapper.db["channel_message"].find():
|
||||||
print(message)
|
print(message)
|
||||||
try:
|
try:
|
||||||
@ -76,9 +103,12 @@ class ChannelMessageService(BaseService):
|
|||||||
)
|
)
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
try:
|
try:
|
||||||
template = self.app.jinja2_env.get_template("message.html")
|
|
||||||
model["html"] = await loop.run_in_executor(executor, lambda: template.render(**context))
|
context = json.loads(json.dumps(context, default=str))
|
||||||
model['html'] = await loop.run_in_executor(executor, lambda:sanitize_html(model['html']))
|
|
||||||
|
|
||||||
|
model["html"] = await loop.run_in_executor(self.get_or_create_executor(model["uid"]), render,context)
|
||||||
|
#model['html'] = await loop.run_in_executor(self.get_or_create_executor(user["uid"]), sanitize_html,model['html'])
|
||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
print(ex, flush=True)
|
print(ex, flush=True)
|
||||||
|
|
||||||
@ -97,6 +127,8 @@ class ChannelMessageService(BaseService):
|
|||||||
["deleted_at"], unique=False
|
["deleted_at"], unique=False
|
||||||
)
|
)
|
||||||
self._configured_indexes = True
|
self._configured_indexes = True
|
||||||
|
if model['is_final']:
|
||||||
|
self.delete_executor(model['uid'])
|
||||||
return model
|
return model
|
||||||
raise Exception(f"Failed to create channel message: {model.errors}.")
|
raise Exception(f"Failed to create channel message: {model.errors}.")
|
||||||
|
|
||||||
@ -133,12 +165,15 @@ class ChannelMessageService(BaseService):
|
|||||||
"color": user["color"],
|
"color": user["color"],
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
template = self.app.jinja2_env.get_template("message.html")
|
context = json.loads(json.dumps(context, default=str))
|
||||||
|
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
model["html"] = await loop.run_in_executor(executor, lambda: template.render(**context))
|
model["html"] = await loop.run_in_executor(self.get_or_create_executor(model["uid"]), render, context)
|
||||||
model['html'] = await loop.run_in_executor(executor, lambda: sanitize_html(model['html']))
|
#model['html'] = await loop.run_in_executor(self.get_or_create_executor(user["uid"]), sanitize_html,model['html'])
|
||||||
return await super().save(model)
|
|
||||||
|
result = await super().save(model)
|
||||||
|
if model['is_final']:
|
||||||
|
self.delete_executor(model['uid'])
|
||||||
|
return result
|
||||||
|
|
||||||
async def offset(self, channel_uid, page=0, timestamp=None, page_size=30):
|
async def offset(self, channel_uid, page=0, timestamp=None, page_size=30):
|
||||||
channel = await self.services.channel.get(uid=channel_uid)
|
channel = await self.services.channel.get(uid=channel_uid)
|
||||||
|
|||||||
@ -65,7 +65,7 @@ export class Chat extends EventHandler {
|
|||||||
}
|
}
|
||||||
return new Promise((resolve) => {
|
return new Promise((resolve) => {
|
||||||
this._waitConnect = resolve;
|
this._waitConnect = resolve;
|
||||||
console.debug("Connecting..");
|
//console.debug("Connecting..");
|
||||||
|
|
||||||
try {
|
try {
|
||||||
this._socket = new WebSocket(this._url);
|
this._socket = new WebSocket(this._url);
|
||||||
@ -142,7 +142,7 @@ export class NotificationAudio {
|
|||||||
new Audio(this.sounds[soundIndex])
|
new Audio(this.sounds[soundIndex])
|
||||||
.play()
|
.play()
|
||||||
.then(() => {
|
.then(() => {
|
||||||
console.debug("Gave sound notification");
|
//console.debug("Gave sound notification");
|
||||||
})
|
})
|
||||||
.catch((error) => {
|
.catch((error) => {
|
||||||
console.error("Notification failed:", error);
|
console.error("Notification failed:", error);
|
||||||
|
|||||||
@ -81,7 +81,152 @@ class ChatInputComponent extends NjetComponent {
|
|||||||
return Array.from(text.matchAll(/@([a-zA-Z0-9_-]+)/g), m => m[1]);
|
return Array.from(text.matchAll(/@([a-zA-Z0-9_-]+)/g), m => m[1]);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
matchMentionsToAuthors(mentions, authors) {
|
matchMentionsToAuthors(mentions, authors) {
|
||||||
|
return mentions.map((mention) => {
|
||||||
|
const lowerMention = mention.toLowerCase();
|
||||||
|
let bestMatch = null;
|
||||||
|
let bestScore = 0;
|
||||||
|
|
||||||
|
for (const author of authors) {
|
||||||
|
const lowerAuthor = author.toLowerCase();
|
||||||
|
let score = 0;
|
||||||
|
|
||||||
|
if (lowerMention === lowerAuthor) {
|
||||||
|
score = 100;
|
||||||
|
} else if (lowerAuthor.startsWith(lowerMention)) {
|
||||||
|
score = 90 + (5 * (lowerMention.length / lowerAuthor.length));
|
||||||
|
} else if (lowerAuthor.includes(lowerMention) && lowerMention.length >= 2) {
|
||||||
|
const position = lowerAuthor.indexOf(lowerMention);
|
||||||
|
score = 80 - (10 * (position / lowerAuthor.length));
|
||||||
|
} else if (this.isFuzzyMatch(lowerMention, lowerAuthor)) {
|
||||||
|
const ratio = lowerMention.length / lowerAuthor.length;
|
||||||
|
score = 40 + (20 * ratio);
|
||||||
|
} else if (this.isCloseMatch(lowerMention, lowerAuthor)) {
|
||||||
|
score = 30 + (10 * (lowerMention.length / lowerAuthor.length));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (score > bestScore) {
|
||||||
|
bestScore = score;
|
||||||
|
bestMatch = author;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const minScore = 40;
|
||||||
|
return {
|
||||||
|
mention,
|
||||||
|
closestAuthor: bestScore >= minScore ? bestMatch : null,
|
||||||
|
score: bestScore,
|
||||||
|
};
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
isFuzzyMatch(needle, haystack) {
|
||||||
|
if (needle.length < 2) return false;
|
||||||
|
|
||||||
|
let needleIndex = 0;
|
||||||
|
for (let i = 0; i < haystack.length && needleIndex < needle.length; i++) {
|
||||||
|
if (haystack[i] === needle[needleIndex]) {
|
||||||
|
needleIndex++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return needleIndex === needle.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
isCloseMatch(str1, str2) {
|
||||||
|
if (Math.abs(str1.length - str2.length) > 2) return false;
|
||||||
|
|
||||||
|
const shorter = str1.length <= str2.length ? str1 : str2;
|
||||||
|
const longer = str1.length > str2.length ? str1 : str2;
|
||||||
|
|
||||||
|
let differences = 0;
|
||||||
|
let j = 0;
|
||||||
|
|
||||||
|
for (let i = 0; i < shorter.length && j < longer.length; i++) {
|
||||||
|
if (shorter[i] !== longer[j]) {
|
||||||
|
differences++;
|
||||||
|
if (j + 1 < longer.length && shorter[i] === longer[j + 1]) {
|
||||||
|
j++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
j++;
|
||||||
|
}
|
||||||
|
|
||||||
|
differences += Math.abs(longer.length - j);
|
||||||
|
return differences <= 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
matchMentions4ToAuthors(mentions, authors) {
|
||||||
|
return mentions.map((mention) => {
|
||||||
|
let closestAuthor = null;
|
||||||
|
let minDistance = Infinity;
|
||||||
|
const lowerMention = mention.toLowerCase();
|
||||||
|
|
||||||
|
authors.forEach((author) => {
|
||||||
|
const lowerAuthor = author.toLowerCase();
|
||||||
|
let distance = this.levenshteinDistance(lowerMention, lowerAuthor);
|
||||||
|
|
||||||
|
if (!this.isSubsequence(lowerMention, lowerAuthor)) {
|
||||||
|
distance += 10;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (distance < minDistance) {
|
||||||
|
minDistance = distance;
|
||||||
|
closestAuthor = author;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
if (minDistance < 5) {
|
||||||
|
closestAuthor = 0;
|
||||||
|
}
|
||||||
|
return { mention, closestAuthor, distance: minDistance };
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
levenshteinDistance(a, b) {
|
||||||
|
const matrix = [];
|
||||||
|
|
||||||
|
for (let i = 0; i <= b.length; i++) {
|
||||||
|
matrix[i] = [i];
|
||||||
|
}
|
||||||
|
for (let j = 0; j <= a.length; j++) {
|
||||||
|
matrix[0][j] = j;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (let i = 1; i <= b.length; i++) {
|
||||||
|
for (let j = 1; j <= a.length; j++) {
|
||||||
|
if (b.charAt(i - 1) === a.charAt(j - 1)) {
|
||||||
|
matrix[i][j] = matrix[i - 1][j - 1];
|
||||||
|
} else {
|
||||||
|
matrix[i][j] = Math.min(
|
||||||
|
matrix[i - 1][j] + 1,
|
||||||
|
matrix[i][j - 1] + 1,
|
||||||
|
matrix[i - 1][j - 1] + 1
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return matrix[b.length][a.length];
|
||||||
|
}
|
||||||
|
|
||||||
|
replaceMentionsWithAuthors(text) {
|
||||||
|
const authors = this.getAuthors();
|
||||||
|
const mentions = this.extractMentions(text);
|
||||||
|
|
||||||
|
const matches = this.matchMentionsToAuthors(mentions, authors);
|
||||||
|
let updatedText = text;
|
||||||
|
matches.forEach(({ mention, closestAuthor }) => {
|
||||||
|
if(closestAuthor){
|
||||||
|
const mentionRegex = new RegExp(`@${mention}`, 'g');
|
||||||
|
updatedText = updatedText.replace(mentionRegex, `@${closestAuthor}`);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return updatedText;
|
||||||
|
}
|
||||||
|
|
||||||
|
matchMentions2ToAuthors(mentions, authors) {
|
||||||
return mentions.map(mention => {
|
return mentions.map(mention => {
|
||||||
let closestAuthor = null;
|
let closestAuthor = null;
|
||||||
let minDistance = Infinity;
|
let minDistance = Infinity;
|
||||||
@ -100,53 +245,14 @@ class ChatInputComponent extends NjetComponent {
|
|||||||
closestAuthor = author;
|
closestAuthor = author;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
if (minDistance < 5){
|
||||||
|
closestAuthor = 0;
|
||||||
|
}
|
||||||
return { mention, closestAuthor, distance: minDistance };
|
return { mention, closestAuthor, distance: minDistance };
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
levenshteinDistance(a, b) {
|
|
||||||
const matrix = [];
|
|
||||||
|
|
||||||
// Initialize the first row and column
|
|
||||||
for (let i = 0; i <= b.length; i++) {
|
|
||||||
matrix[i] = [i];
|
|
||||||
}
|
|
||||||
for (let j = 0; j <= a.length; j++) {
|
|
||||||
matrix[0][j] = j;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Fill in the matrix
|
|
||||||
for (let i = 1; i <= b.length; i++) {
|
|
||||||
for (let j = 1; j <= a.length; j++) {
|
|
||||||
if (b.charAt(i - 1) === a.charAt(j - 1)) {
|
|
||||||
matrix[i][j] = matrix[i - 1][j - 1];
|
|
||||||
} else {
|
|
||||||
matrix[i][j] = Math.min(
|
|
||||||
matrix[i - 1][j] + 1, // Deletion
|
|
||||||
matrix[i][j - 1] + 1, // Insertion
|
|
||||||
matrix[i - 1][j - 1] + 1 // Substitution
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return matrix[b.length][a.length];
|
|
||||||
}
|
|
||||||
|
|
||||||
replaceMentionsWithAuthors(text) {
|
|
||||||
const authors = this.getAuthors();
|
|
||||||
const mentions = this.extractMentions(text);
|
|
||||||
|
|
||||||
const matches = this.matchMentionsToAuthors(mentions, authors);
|
|
||||||
let updatedText = text;
|
|
||||||
matches.forEach(({ mention, closestAuthor }) => {
|
|
||||||
const mentionRegex = new RegExp(`@${mention}`, 'g');
|
|
||||||
updatedText = updatedText.replace(mentionRegex, `@${closestAuthor}`);
|
|
||||||
});
|
|
||||||
|
|
||||||
return updatedText;
|
|
||||||
}
|
|
||||||
textToLeet(text) {
|
textToLeet(text) {
|
||||||
// L33t speak character mapping
|
// L33t speak character mapping
|
||||||
const leetMap = {
|
const leetMap = {
|
||||||
@ -392,7 +498,7 @@ textToLeetAdvanced(text) {
|
|||||||
}
|
}
|
||||||
j++;
|
j++;
|
||||||
}
|
}
|
||||||
return i === s.length;
|
return i === s.length && s.length > 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
flagTyping() {
|
flagTyping() {
|
||||||
|
|||||||
@ -256,7 +256,9 @@ class MessageList extends HTMLElement {
|
|||||||
this.querySelectorAll('.avatar').forEach((el) => {
|
this.querySelectorAll('.avatar').forEach((el) => {
|
||||||
const anchor = el.closest('a');
|
const anchor = el.closest('a');
|
||||||
if (anchor && typeof anchor.href === 'string' && anchor.href.includes(uid)) {
|
if (anchor && typeof anchor.href === 'string' && anchor.href.includes(uid)) {
|
||||||
lastElement = el;
|
if(!lastElement)
|
||||||
|
lastElement = el;
|
||||||
|
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
if (lastElement) {
|
if (lastElement) {
|
||||||
@ -278,11 +280,15 @@ class MessageList extends HTMLElement {
|
|||||||
upsertMessage(data) {
|
upsertMessage(data) {
|
||||||
let message = this.messageMap.get(data.uid);
|
let message = this.messageMap.get(data.uid);
|
||||||
if (message && (data.is_final || !data.message)) {
|
if (message && (data.is_final || !data.message)) {
|
||||||
message.parentElement?.removeChild(message);
|
//message.parentElement?.removeChild(message);
|
||||||
// TO force insert
|
// TO force insert
|
||||||
message = null;
|
//message = null;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
if(message && !data.message){
|
||||||
|
message.parentElement?.removeChild(message);
|
||||||
|
message = null;
|
||||||
|
}
|
||||||
if (!data.message) return;
|
if (!data.message) return;
|
||||||
|
|
||||||
const wrapper = document.createElement("div");
|
const wrapper = document.createElement("div");
|
||||||
|
|||||||
@ -52,7 +52,7 @@ self.addEventListener("push", async (event) => {
|
|||||||
data,
|
data,
|
||||||
}).then(e => console.log("Showing notification", e)).catch(console.error);
|
}).then(e => console.log("Showing notification", e)).catch(console.error);
|
||||||
|
|
||||||
event.waitUntil(reg);
|
// event.waitUntil(reg);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -86,6 +86,7 @@ export class Socket extends EventHandler {
|
|||||||
}
|
}
|
||||||
this.emit("data", data.data);
|
this.emit("data", data.data);
|
||||||
if (data["event"]) {
|
if (data["event"]) {
|
||||||
|
console.info([data.event,data.data])
|
||||||
this.emit(data.event, data.data);
|
this.emit(data.event, data.data);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -99,7 +100,7 @@ export class Socket extends EventHandler {
|
|||||||
console.log("Reconnecting");
|
console.log("Reconnecting");
|
||||||
this.emit("reconnecting");
|
this.emit("reconnecting");
|
||||||
return this.connect();
|
return this.connect();
|
||||||
}, 0);
|
}, 4000);
|
||||||
}
|
}
|
||||||
|
|
||||||
_camelToSnake(str) {
|
_camelToSnake(str) {
|
||||||
|
|||||||
@ -1,75 +1,137 @@
|
|||||||
|
import asyncio
|
||||||
import functools
|
import functools
|
||||||
import json
|
import json
|
||||||
|
from collections import OrderedDict
|
||||||
from snek.system import security
|
from snek.system import security
|
||||||
|
|
||||||
cache = functools.cache
|
|
||||||
|
|
||||||
CACHE_MAX_ITEMS_DEFAULT = 5000
|
CACHE_MAX_ITEMS_DEFAULT = 5000
|
||||||
|
|
||||||
|
|
||||||
class Cache:
|
class Cache:
|
||||||
|
"""
|
||||||
|
An asynchronous, thread-safe, in-memory LRU (Least Recently Used) cache.
|
||||||
|
|
||||||
|
This implementation uses an OrderedDict for efficient O(1) time complexity
|
||||||
|
for its core get, set, and delete operations.
|
||||||
|
"""
|
||||||
|
|
||||||
def __init__(self, app, max_items=CACHE_MAX_ITEMS_DEFAULT):
|
def __init__(self, app, max_items=CACHE_MAX_ITEMS_DEFAULT):
|
||||||
self.app = app
|
self.app = app
|
||||||
self.cache = {}
|
# OrderedDict is the core of the LRU logic. It remembers the order
|
||||||
|
# in which items were inserted.
|
||||||
|
self.cache: OrderedDict = OrderedDict()
|
||||||
self.max_items = max_items
|
self.max_items = max_items
|
||||||
self.stats = {}
|
self.stats = {}
|
||||||
self.enabled = True
|
self.enabled = True
|
||||||
self.lru = []
|
# A lock is crucial to prevent race conditions in an async environment.
|
||||||
self.version = ((42 + 420 + 1984 + 1990 + 10 + 6 + 71 + 3004 + 7245) ^ 1337) + 4
|
self._lock = asyncio.Lock()
|
||||||
|
self.version = ((42 + 420 + 1984 + 1990 + 10 + 6 + 71 + 3004 + 7245) ^ 1337) + 4
|
||||||
|
|
||||||
async def get(self, args):
|
async def get(self, key):
|
||||||
|
"""
|
||||||
|
Retrieves an item from the cache. If found, it's marked as recently used.
|
||||||
|
Returns None if the item is not found or the cache is disabled.
|
||||||
|
"""
|
||||||
if not self.enabled:
|
if not self.enabled:
|
||||||
return None
|
return None
|
||||||
await self.update_stat(args, "get")
|
|
||||||
try:
|
#async with self._lock:
|
||||||
self.lru.pop(self.lru.index(args))
|
if key not in self.cache:
|
||||||
except:
|
await self.update_stat(key, "get")
|
||||||
# print("Cache miss!", args, flush=True)
|
|
||||||
return None
|
return None
|
||||||
self.lru.insert(0, args)
|
|
||||||
while len(self.lru) > self.max_items:
|
# Mark as recently used by moving it to the end of the OrderedDict.
|
||||||
self.cache.pop(self.lru[-1])
|
# This is an O(1) operation.
|
||||||
self.lru.pop()
|
self.cache.move_to_end(key)
|
||||||
# print("Cache hit!", args, flush=True)
|
await self.update_stat(key, "get")
|
||||||
return self.cache[args]
|
return self.cache[key]
|
||||||
|
|
||||||
|
async def set(self, key, value):
|
||||||
|
"""
|
||||||
|
Adds or updates an item in the cache and marks it as recently used.
|
||||||
|
If the cache exceeds its maximum size, the least recently used item is evicted.
|
||||||
|
"""
|
||||||
|
if not self.enabled:
|
||||||
|
return
|
||||||
|
# comment
|
||||||
|
#async with self._lock:
|
||||||
|
is_new = key not in self.cache
|
||||||
|
|
||||||
|
# Add or update the item. If it exists, it's moved to the end.
|
||||||
|
self.cache[key] = value
|
||||||
|
self.cache.move_to_end(key)
|
||||||
|
|
||||||
|
await self.update_stat(key, "set")
|
||||||
|
|
||||||
|
# Evict the least recently used item if the cache is full.
|
||||||
|
# This is an O(1) operation.
|
||||||
|
if len(self.cache) > self.max_items:
|
||||||
|
# popitem(last=False) removes and returns the first (oldest) item.
|
||||||
|
evicted_key, _ = self.cache.popitem(last=False)
|
||||||
|
# Optionally, you could log the evicted key here.
|
||||||
|
|
||||||
|
if is_new:
|
||||||
|
self.version += 1
|
||||||
|
|
||||||
|
async def delete(self, key):
|
||||||
|
"""Removes an item from the cache if it exists."""
|
||||||
|
if not self.enabled:
|
||||||
|
return
|
||||||
|
|
||||||
|
async with self._lock:
|
||||||
|
if key in self.cache:
|
||||||
|
await self.update_stat(key, "delete")
|
||||||
|
# Deleting from OrderedDict is an O(1) operation on average.
|
||||||
|
del self.cache[key]
|
||||||
|
|
||||||
async def get_stats(self):
|
async def get_stats(self):
|
||||||
all_ = []
|
"""Returns statistics for all items currently in the cache."""
|
||||||
for key in self.lru:
|
async with self._lock:
|
||||||
all_.append(
|
stats_list = []
|
||||||
{
|
# Items are iterated from oldest to newest. We reverse to show
|
||||||
"key": key,
|
# most recently used items first.
|
||||||
"set": self.stats[key]["set"],
|
for key in reversed(self.cache):
|
||||||
"get": self.stats[key]["get"],
|
stat_data = self.stats.get(key, {"set": 0, "get": 0, "delete": 0})
|
||||||
"delete": self.stats[key]["delete"],
|
value = self.cache[key]
|
||||||
"value": str(self.serialize(self.cache[key].record)),
|
value_record = value.record if hasattr(value, 'record') else value
|
||||||
}
|
|
||||||
)
|
|
||||||
return all_
|
|
||||||
|
|
||||||
def serialize(self, obj):
|
stats_list.append({
|
||||||
cpy = obj.copy()
|
"key": key,
|
||||||
cpy.pop("created_at", None)
|
"set": stat_data.get("set", 0),
|
||||||
cpy.pop("deleted_at", None)
|
"get": stat_data.get("get", 0),
|
||||||
cpy.pop("email", None)
|
"delete": stat_data.get("delete", 0),
|
||||||
cpy.pop("password", None)
|
"value": str(self.serialize(value_record)),
|
||||||
return cpy
|
})
|
||||||
|
return stats_list
|
||||||
|
|
||||||
async def update_stat(self, key, action):
|
async def update_stat(self, key, action):
|
||||||
if key not in self.stats:
|
"""Updates hit/miss/set counts for a given cache key."""
|
||||||
self.stats[key] = {"set": 0, "get": 0, "delete": 0}
|
# This method is already called within a locked context,
|
||||||
self.stats[key][action] = self.stats[key][action] + 1
|
# but the lock makes it safe if ever called directly.
|
||||||
|
async with self._lock:
|
||||||
|
if key not in self.stats:
|
||||||
|
self.stats[key] = {"set": 0, "get": 0, "delete": 0}
|
||||||
|
self.stats[key][action] += 1
|
||||||
|
|
||||||
|
def serialize(self, obj):
|
||||||
|
"""A synchronous helper to create a serializable representation of an object."""
|
||||||
|
if not isinstance(obj, dict):
|
||||||
|
return obj
|
||||||
|
cpy = obj.copy()
|
||||||
|
for key_to_remove in ["created_at", "deleted_at", "email", "password"]:
|
||||||
|
cpy.pop(key_to_remove, None)
|
||||||
|
return cpy
|
||||||
|
|
||||||
def json_default(self, value):
|
def json_default(self, value):
|
||||||
# if hasattr(value, "to_json"):
|
"""JSON serializer fallback for objects that are not directly serializable."""
|
||||||
# return value.to_json()
|
|
||||||
try:
|
try:
|
||||||
return json.dumps(value.__dict__, default=str)
|
return json.dumps(value.__dict__, default=str)
|
||||||
except:
|
except:
|
||||||
return str(value)
|
return str(value)
|
||||||
|
|
||||||
async def create_cache_key(self, args, kwargs):
|
async def create_cache_key(self, args, kwargs):
|
||||||
|
"""Creates a consistent, hashable cache key from function arguments."""
|
||||||
|
# security.hash is async, so this method remains async.
|
||||||
return await security.hash(
|
return await security.hash(
|
||||||
json.dumps(
|
json.dumps(
|
||||||
{"args": args, "kwargs": kwargs},
|
{"args": args, "kwargs": kwargs},
|
||||||
@ -78,38 +140,8 @@ class Cache:
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
async def set(self, args, result):
|
|
||||||
if not self.enabled:
|
|
||||||
return
|
|
||||||
is_new = args not in self.cache
|
|
||||||
self.cache[args] = result
|
|
||||||
await self.update_stat(args, "set")
|
|
||||||
try:
|
|
||||||
self.lru.pop(self.lru.index(args))
|
|
||||||
except (ValueError, IndexError):
|
|
||||||
pass
|
|
||||||
self.lru.insert(0, args)
|
|
||||||
|
|
||||||
while len(self.lru) > self.max_items:
|
|
||||||
self.cache.pop(self.lru[-1])
|
|
||||||
self.lru.pop()
|
|
||||||
|
|
||||||
if is_new:
|
|
||||||
self.version += 1
|
|
||||||
# print(f"Cache store! {len(self.lru)} items. New version:", self.version, flush=True)
|
|
||||||
|
|
||||||
async def delete(self, args):
|
|
||||||
if not self.enabled:
|
|
||||||
return
|
|
||||||
await self.update_stat(args, "delete")
|
|
||||||
if args in self.cache:
|
|
||||||
try:
|
|
||||||
self.lru.pop(self.lru.index(args))
|
|
||||||
except IndexError:
|
|
||||||
pass
|
|
||||||
del self.cache[args]
|
|
||||||
|
|
||||||
def async_cache(self, func):
|
def async_cache(self, func):
|
||||||
|
"""Decorator to cache the results of an async function."""
|
||||||
@functools.wraps(func)
|
@functools.wraps(func)
|
||||||
async def wrapper(*args, **kwargs):
|
async def wrapper(*args, **kwargs):
|
||||||
cache_key = await self.create_cache_key(args, kwargs)
|
cache_key = await self.create_cache_key(args, kwargs)
|
||||||
@ -119,33 +151,14 @@ class Cache:
|
|||||||
result = await func(*args, **kwargs)
|
result = await func(*args, **kwargs)
|
||||||
await self.set(cache_key, result)
|
await self.set(cache_key, result)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
def async_delete_cache(self, func):
|
def async_delete_cache(self, func):
|
||||||
|
"""Decorator to invalidate a cache entry before running an async function."""
|
||||||
@functools.wraps(func)
|
@functools.wraps(func)
|
||||||
async def wrapper(*args, **kwargs):
|
async def wrapper(*args, **kwargs):
|
||||||
cache_key = await self.create_cache_key(args, kwargs)
|
cache_key = await self.create_cache_key(args, kwargs)
|
||||||
if cache_key in self.cache:
|
await self.delete(cache_key)
|
||||||
try:
|
|
||||||
self.lru.pop(self.lru.index(cache_key))
|
|
||||||
except IndexError:
|
|
||||||
pass
|
|
||||||
del self.cache[cache_key]
|
|
||||||
return await func(*args, **kwargs)
|
return await func(*args, **kwargs)
|
||||||
|
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
def async_cache(func):
|
|
||||||
cache = {}
|
|
||||||
|
|
||||||
@functools.wraps(func)
|
|
||||||
async def wrapper(*args):
|
|
||||||
if args in cache:
|
|
||||||
return cache[args]
|
|
||||||
result = await func(*args)
|
|
||||||
cache[args] = result
|
|
||||||
return result
|
|
||||||
|
|
||||||
return wrapper
|
|
||||||
|
|||||||
@ -1,7 +1,7 @@
|
|||||||
DEFAULT_LIMIT = 30
|
DEFAULT_LIMIT = 30
|
||||||
import asyncio
|
import asyncio
|
||||||
import typing
|
import typing
|
||||||
|
import time
|
||||||
from snek.system.model import BaseModel
|
from snek.system.model import BaseModel
|
||||||
|
|
||||||
|
|
||||||
@ -15,8 +15,6 @@ class BaseMapper:
|
|||||||
def __init__(self, app):
|
def __init__(self, app):
|
||||||
self.app = app
|
self.app = app
|
||||||
|
|
||||||
self.default_limit = self.__class__.default_limit
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def db(self):
|
def db(self):
|
||||||
return self.app.db
|
return self.app.db
|
||||||
@ -27,6 +25,7 @@ class BaseMapper:
|
|||||||
|
|
||||||
async def run_in_executor(self, func, *args, **kwargs):
|
async def run_in_executor(self, func, *args, **kwargs):
|
||||||
use_semaphore = kwargs.pop("use_semaphore", False)
|
use_semaphore = kwargs.pop("use_semaphore", False)
|
||||||
|
start_time = time.time()
|
||||||
|
|
||||||
def _execute():
|
def _execute():
|
||||||
result = func(*args, **kwargs)
|
result = func(*args, **kwargs)
|
||||||
@ -34,10 +33,8 @@ class BaseMapper:
|
|||||||
self.db.commit()
|
self.db.commit()
|
||||||
return result
|
return result
|
||||||
|
|
||||||
return _execute()
|
async with self.semaphore:
|
||||||
#async with self.semaphore:
|
return await asyncio.to_thread(_execute)
|
||||||
# return await self.loop.run_in_executor(None, _execute)
|
|
||||||
|
|
||||||
|
|
||||||
async def new(self):
|
async def new(self):
|
||||||
return self.model_class(mapper=self, app=self.app)
|
return self.model_class(mapper=self, app=self.app)
|
||||||
|
|||||||
@ -1,129 +0,0 @@
|
|||||||
import asyncio
|
|
||||||
from aiohttp import web, WSMsgType
|
|
||||||
|
|
||||||
from datetime import datetime, timedelta, timezone
|
|
||||||
from collections import defaultdict
|
|
||||||
import html
|
|
||||||
|
|
||||||
def create_stats_structure():
|
|
||||||
"""Creates the nested dictionary structure for storing statistics."""
|
|
||||||
def nested_dd():
|
|
||||||
return defaultdict(lambda: defaultdict(int))
|
|
||||||
return defaultdict(nested_dd)
|
|
||||||
|
|
||||||
def get_time_keys(dt: datetime):
|
|
||||||
"""Generates dictionary keys for different time granularities."""
|
|
||||||
return {
|
|
||||||
"hour": dt.strftime('%Y-%m-%d-%H'),
|
|
||||||
"day": dt.strftime('%Y-%m-%d'),
|
|
||||||
"week": dt.strftime('%Y-%W'), # Week number, Monday is first day
|
|
||||||
"month": dt.strftime('%Y-%m'),
|
|
||||||
}
|
|
||||||
|
|
||||||
def update_stats_counters(stats_dict: defaultdict, now: datetime):
|
|
||||||
"""Increments the appropriate time-based counters in a stats dictionary."""
|
|
||||||
keys = get_time_keys(now)
|
|
||||||
stats_dict['by_hour'][keys['hour']] += 1
|
|
||||||
stats_dict['by_day'][keys['day']] += 1
|
|
||||||
stats_dict['by_week'][keys['week']] += 1
|
|
||||||
stats_dict['by_month'][keys['month']] += 1
|
|
||||||
|
|
||||||
def generate_time_series_svg(title: str, data: list[tuple[str, int]], y_label: str) -> str:
|
|
||||||
"""Generates a responsive SVG bar chart for time-series data."""
|
|
||||||
if not data:
|
|
||||||
return f"<h3>{html.escape(title)}</h3><p>No data yet.</p>"
|
|
||||||
max_val = max(item[1] for item in data) if data else 1
|
|
||||||
svg_height, svg_width = 250, 600
|
|
||||||
bar_padding = 5
|
|
||||||
bar_width = (svg_width - 50) / len(data) - bar_padding
|
|
||||||
|
|
||||||
bars = ""
|
|
||||||
labels = ""
|
|
||||||
for i, (key, val) in enumerate(data):
|
|
||||||
bar_height = (val / max_val) * (svg_height - 50) if max_val > 0 else 0
|
|
||||||
x = i * (bar_width + bar_padding) + 40
|
|
||||||
y = svg_height - bar_height - 30
|
|
||||||
|
|
||||||
bars += f'<rect x="{x}" y="{y}" width="{bar_width}" height="{bar_height}" fill="#007BFF"><title>{html.escape(key)}: {val}</title></rect>'
|
|
||||||
labels += f'<text x="{x + bar_width / 2}" y="{svg_height - 15}" font-size="11" text-anchor="middle">{html.escape(key)}</text>'
|
|
||||||
|
|
||||||
return f"""
|
|
||||||
<h3>{html.escape(title)}</h3>
|
|
||||||
<div style="border:1px solid #ccc; padding: 10px; border-radius: 5px;">
|
|
||||||
<svg viewBox="0 0 {svg_width} {svg_height}" style="width:100%; height:auto;">
|
|
||||||
<g>{bars}</g>
|
|
||||||
<g>{labels}</g>
|
|
||||||
<line x1="35" y1="10" x2="35" y2="{svg_height - 30}" stroke="#aaa" stroke-width="1" />
|
|
||||||
<line x1="35" y1="{svg_height - 30}" x2="{svg_width - 10}" y2="{svg_height - 30}" stroke="#aaa" stroke-width="1" />
|
|
||||||
<text x="5" y="{svg_height - 30}" font-size="12">0</text>
|
|
||||||
<text x="5" y="20" font-size="12">{max_val}</text>
|
|
||||||
</svg>
|
|
||||||
</div>
|
|
||||||
"""
|
|
||||||
|
|
||||||
@web.middleware
|
|
||||||
async def middleware(request, handler):
|
|
||||||
"""Middleware to count all incoming HTTP requests."""
|
|
||||||
# Avoid counting requests to the stats page itself
|
|
||||||
if request.path.startswith('/stats.html'):
|
|
||||||
return await handler(request)
|
|
||||||
|
|
||||||
update_stats_counters(request.app['stats']['http_requests'], datetime.now(timezone.utc))
|
|
||||||
return await handler(request)
|
|
||||||
|
|
||||||
def update_websocket_stats(app):
|
|
||||||
update_stats_counters(app['stats']['websocket_requests'], datetime.now(timezone.utc))
|
|
||||||
|
|
||||||
async def pipe_and_count_websocket(ws_from, ws_to, stats_dict):
|
|
||||||
"""This function proxies WebSocket messages AND counts them."""
|
|
||||||
async for msg in ws_from:
|
|
||||||
# This is the key part for monitoring WebSockets
|
|
||||||
update_stats_counters(stats_dict, datetime.now(timezone.utc))
|
|
||||||
|
|
||||||
if msg.type == WSMsgType.TEXT:
|
|
||||||
await ws_to.send_str(msg.data)
|
|
||||||
elif msg.type == WSMsgType.BINARY:
|
|
||||||
await ws_to.send_bytes(msg.data)
|
|
||||||
elif msg.type in (WSMsgType.CLOSE, WSMsgType.ERROR):
|
|
||||||
await ws_to.close(code=ws_from.close_code)
|
|
||||||
break
|
|
||||||
|
|
||||||
|
|
||||||
async def stats_handler(request: web.Request):
|
|
||||||
"""Handler to display the statistics dashboard."""
|
|
||||||
stats = request.app['stats']
|
|
||||||
now = datetime.now(timezone.utc)
|
|
||||||
|
|
||||||
# Helper to prepare data for charts
|
|
||||||
def get_data(source, period, count):
|
|
||||||
data = []
|
|
||||||
for i in range(count - 1, -1, -1):
|
|
||||||
if period == 'hour':
|
|
||||||
dt = now - timedelta(hours=i)
|
|
||||||
key, label = dt.strftime('%Y-%m-%d-%H'), dt.strftime('%H:00')
|
|
||||||
data.append((label, source['by_hour'].get(key, 0)))
|
|
||||||
elif period == 'day':
|
|
||||||
dt = now - timedelta(days=i)
|
|
||||||
key, label = dt.strftime('%Y-%m-%d'), dt.strftime('%a')
|
|
||||||
data.append((label, source['by_day'].get(key, 0)))
|
|
||||||
return data
|
|
||||||
|
|
||||||
http_hourly = get_data(stats['http_requests'], 'hour', 24)
|
|
||||||
ws_hourly = get_data(stats['ws_messages'], 'hour', 24)
|
|
||||||
http_daily = get_data(stats['http_requests'], 'day', 7)
|
|
||||||
ws_daily = get_data(stats['ws_messages'], 'day', 7)
|
|
||||||
|
|
||||||
body = f"""
|
|
||||||
<html><head><title>App Stats</title><meta http-equiv="refresh" content="30"></head>
|
|
||||||
<body>
|
|
||||||
<h2>Application Dashboard</h2>
|
|
||||||
<h3>Last 24 Hours</h3>
|
|
||||||
{generate_time_series_svg("HTTP Requests", http_hourly, "Reqs/Hour")}
|
|
||||||
{generate_time_series_svg("WebSocket Messages", ws_hourly, "Msgs/Hour")}
|
|
||||||
<h3>Last 7 Days</h3>
|
|
||||||
{generate_time_series_svg("HTTP Requests", http_daily, "Reqs/Day")}
|
|
||||||
{generate_time_series_svg("WebSocket Messages", ws_daily, "Msgs/Day")}
|
|
||||||
</body></html>
|
|
||||||
"""
|
|
||||||
return web.Response(text=body, content_type='text/html')
|
|
||||||
|
|
||||||
@ -32,7 +32,7 @@
|
|||||||
<link rel="stylesheet" href="/base.css">
|
<link rel="stylesheet" href="/base.css">
|
||||||
<link rel="icon" type="image/png" href="/image/snek_logo_32x32.png" sizes="32x32">
|
<link rel="icon" type="image/png" href="/image/snek_logo_32x32.png" sizes="32x32">
|
||||||
<link rel="icon" type="image/png" href="/image/snek_logo_64x64.png" sizes="64x64">
|
<link rel="icon" type="image/png" href="/image/snek_logo_64x64.png" sizes="64x64">
|
||||||
<script nonce="{{nonce}}" defer src="https://umami.molodetz.nl/script.js" data-website-id="d127c3e4-dc70-4041-a1c8-bcc32c2492ea"></script>
|
<script nonce="{{nonce}}" defer src="https://umami.molodetz.nl/script.js" data-website-id="d127c3e4-dc70-4041-a1c8-bcc32c2492ea" defer></script>
|
||||||
</head>
|
</head>
|
||||||
<body>
|
<body>
|
||||||
<header>
|
<header>
|
||||||
|
|||||||
97
src/snek/templates/new.html
Normal file
97
src/snek/templates/new.html
Normal file
@ -0,0 +1,97 @@
|
|||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0, interactive-widget=resizes-content">
|
||||||
|
<link rel="manifest" href="/manifest.json" />
|
||||||
|
<style>
|
||||||
|
body{
|
||||||
|
background-color: black;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<script type="module">
|
||||||
|
import { Socket } from "./socket.js";
|
||||||
|
|
||||||
|
class ChatWindow extends HTMLElement {
|
||||||
|
|
||||||
|
constructor() {
|
||||||
|
super();
|
||||||
|
this.component = document.createElement("div");
|
||||||
|
this.message_list = document.createElement("div")
|
||||||
|
this.component.appendChild(this.message_list);
|
||||||
|
this.chat_input = document.createElement("div")
|
||||||
|
this.component.appendChild(this.chat_input);
|
||||||
|
this.channelUid = null
|
||||||
|
this.channelUid = this.getAttribute("channel")
|
||||||
|
this.inputText = document.createElement("textarea")
|
||||||
|
this.inputText.addEventListener("keyup",(e)=>{
|
||||||
|
|
||||||
|
|
||||||
|
this.rpc.sendMessage(this.channelUid, e.target.value,false)
|
||||||
|
if(e.key == "Enter" && !e.shiftKey){
|
||||||
|
|
||||||
|
|
||||||
|
this.rpc.sendMessage(this.channelUid, e.target.value,true)
|
||||||
|
|
||||||
|
e.target.value = ""
|
||||||
|
}else{
|
||||||
|
//this.rpc.sendMessage(this.channelUid, e.target.value, false)
|
||||||
|
|
||||||
|
}
|
||||||
|
})
|
||||||
|
this.component.appendChild(this.inputText)
|
||||||
|
this.ws = new Socket();
|
||||||
|
this.ws.addEventListener("channel-message", this.handleMessage.bind(this))
|
||||||
|
this.rpc = this.ws.client
|
||||||
|
this.ws.addEventListener("update_message_text",this.handleMessage.bind(this))
|
||||||
|
window.chat = this
|
||||||
|
}
|
||||||
|
|
||||||
|
async handleMessage(data,data2) {
|
||||||
|
if(data2 && data2.event)
|
||||||
|
data = data.data
|
||||||
|
console.info(["update-messagettt",data])
|
||||||
|
console.warn(data.uid)
|
||||||
|
if(!data.html)
|
||||||
|
return
|
||||||
|
let div = this.message_list.querySelector('[data-uid="' + data.uid + '"]');
|
||||||
|
console.info(div)
|
||||||
|
if(!div){
|
||||||
|
let temp = document.createElement("chat-message");
|
||||||
|
temp.innerHTML = data.html
|
||||||
|
this.message_list.appendChild(temp)
|
||||||
|
//this.message_list.replace(div,temp)
|
||||||
|
//div.innerHTML = data.html
|
||||||
|
//this.message_list.appendChild(div);
|
||||||
|
}else{
|
||||||
|
// alert("HIERR")
|
||||||
|
let temp = document.createElement("chat-message");
|
||||||
|
temp.innerHTML = data.html;
|
||||||
|
div.innerHTML = temp.innerHTML
|
||||||
|
console.info("REPLACE")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async connectedCallback() {
|
||||||
|
await this.rpc.ping(this.channel)
|
||||||
|
console.info(this.channelUid)
|
||||||
|
this.messages = await this.rpc.getMessages(this.channelUid, 0, 0);
|
||||||
|
this.messages.forEach((msg) => {
|
||||||
|
const temp = document.createElement("div");
|
||||||
|
temp.innerHTML = msg.html;
|
||||||
|
this.message_list.appendChild(temp.firstChild);
|
||||||
|
})
|
||||||
|
this.appendChild(this.component);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
customElements.define("chat-window", ChatWindow);
|
||||||
|
</script>
|
||||||
|
<chat-window channel="df3e1259-7d1a-4184-b75c-3befd5bf08e1"></chat-window>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@ -71,7 +71,10 @@ class AvatarView(BaseView):
|
|||||||
uid = self.request.match_info.get("uid")
|
uid = self.request.match_info.get("uid")
|
||||||
if uid == "unique":
|
if uid == "unique":
|
||||||
uid = str(uuid.uuid4())
|
uid = str(uuid.uuid4())
|
||||||
avatar = multiavatar.multiavatar(uid, True, None)
|
avatar = await self.app.get(uid)
|
||||||
|
if not avatar:
|
||||||
|
avatar = multiavatar.multiavatar(uid, True, None)
|
||||||
|
await self.app.set(uid, avatar)
|
||||||
response = web.Response(text=avatar, content_type="image/svg+xml")
|
response = web.Response(text=avatar, content_type="image/svg+xml")
|
||||||
response.headers["Cache-Control"] = f"public, max-age={1337*42}"
|
response.headers["Cache-Control"] = f"public, max-age={1337*42}"
|
||||||
return response
|
return response
|
||||||
|
|||||||
8
src/snek/view/new.py
Normal file
8
src/snek/view/new.py
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
from snek.system.view import BaseView
|
||||||
|
|
||||||
|
|
||||||
|
class NewView(BaseView):
|
||||||
|
login_required = True
|
||||||
|
|
||||||
|
async def get(self):
|
||||||
|
return await self.render_template("new.html")
|
||||||
@ -6,7 +6,6 @@
|
|||||||
|
|
||||||
# MIT License: Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions.
|
# MIT License: Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions.
|
||||||
|
|
||||||
from snek.system.stats import update_websocket_stats
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
@ -507,9 +506,7 @@ class RPCView(BaseView):
|
|||||||
raise Exception("Method not found")
|
raise Exception("Method not found")
|
||||||
success = True
|
success = True
|
||||||
try:
|
try:
|
||||||
update_websocket_stats(self.app)
|
|
||||||
result = await method(*args)
|
result = await method(*args)
|
||||||
update_websocket_stats(self.app)
|
|
||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
result = {"exception": str(ex), "traceback": traceback.format_exc()}
|
result = {"exception": str(ex), "traceback": traceback.format_exc()}
|
||||||
success = False
|
success = False
|
||||||
|
|||||||
@ -1,78 +0,0 @@
|
|||||||
import asyncio
|
|
||||||
import logging
|
|
||||||
import os
|
|
||||||
|
|
||||||
import asyncssh
|
|
||||||
|
|
||||||
asyncssh.set_debug_level(2)
|
|
||||||
logging.basicConfig(level=logging.DEBUG)
|
|
||||||
# Configuration for SFTP server
|
|
||||||
SFTP_ROOT = "." # Directory to serve
|
|
||||||
USERNAME = "test"
|
|
||||||
PASSWORD = "woeii"
|
|
||||||
HOST = "localhost"
|
|
||||||
PORT = 2225
|
|
||||||
|
|
||||||
|
|
||||||
class MySFTPServer(asyncssh.SFTPServer):
|
|
||||||
def __init__(self, chan):
|
|
||||||
super().__init__(chan)
|
|
||||||
self.root = os.path.abspath(SFTP_ROOT)
|
|
||||||
|
|
||||||
async def stat(self, path):
|
|
||||||
"""Handles 'stat' command from SFTP client"""
|
|
||||||
full_path = os.path.join(self.root, path.lstrip("/"))
|
|
||||||
return await super().stat(full_path)
|
|
||||||
|
|
||||||
async def open(self, path, flags, attrs):
|
|
||||||
"""Handles file open requests"""
|
|
||||||
full_path = os.path.join(self.root, path.lstrip("/"))
|
|
||||||
return await super().open(full_path, flags, attrs)
|
|
||||||
|
|
||||||
async def listdir(self, path):
|
|
||||||
"""Handles directory listing"""
|
|
||||||
full_path = os.path.join(self.root, path.lstrip("/"))
|
|
||||||
return await super().listdir(full_path)
|
|
||||||
|
|
||||||
|
|
||||||
class MySSHServer(asyncssh.SSHServer):
|
|
||||||
"""Custom SSH server to handle authentication"""
|
|
||||||
|
|
||||||
def connection_made(self, conn):
|
|
||||||
print(f"New connection from {conn.get_extra_info('peername')}")
|
|
||||||
|
|
||||||
def connection_lost(self, exc):
|
|
||||||
print("Client disconnected")
|
|
||||||
|
|
||||||
def begin_auth(self, username):
|
|
||||||
return True # No additional authentication steps
|
|
||||||
|
|
||||||
def password_auth_supported(self):
|
|
||||||
return True # Support password authentication
|
|
||||||
|
|
||||||
def validate_password(self, username, password):
|
|
||||||
print(username, password)
|
|
||||||
|
|
||||||
return True
|
|
||||||
return username == USERNAME and password == PASSWORD
|
|
||||||
|
|
||||||
|
|
||||||
async def start_sftp_server():
|
|
||||||
os.makedirs(SFTP_ROOT, exist_ok=True) # Ensure the root directory exists
|
|
||||||
|
|
||||||
await asyncssh.create_server(
|
|
||||||
lambda: MySSHServer(),
|
|
||||||
host=HOST,
|
|
||||||
port=PORT,
|
|
||||||
server_host_keys=["ssh_host_key"],
|
|
||||||
process_factory=MySFTPServer,
|
|
||||||
)
|
|
||||||
print(f"SFTP server running on {HOST}:{PORT}")
|
|
||||||
await asyncio.Future() # Keep running forever
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
try:
|
|
||||||
asyncio.run(start_sftp_server())
|
|
||||||
except (OSError, asyncssh.Error) as e:
|
|
||||||
print(f"Error starting SFTP server: {e}")
|
|
||||||
@ -1,77 +0,0 @@
|
|||||||
import asyncio
|
|
||||||
import os
|
|
||||||
|
|
||||||
import asyncssh
|
|
||||||
|
|
||||||
# SSH Server Configuration
|
|
||||||
HOST = "0.0.0.0"
|
|
||||||
PORT = 2225
|
|
||||||
USERNAME = "user"
|
|
||||||
PASSWORD = "password"
|
|
||||||
SHELL = "/bin/sh" # Change to another shell if needed
|
|
||||||
|
|
||||||
|
|
||||||
class CustomSSHServer(asyncssh.SSHServer):
|
|
||||||
def connection_made(self, conn):
|
|
||||||
print(f"New connection from {conn.get_extra_info('peername')}")
|
|
||||||
|
|
||||||
def connection_lost(self, exc):
|
|
||||||
print("Client disconnected")
|
|
||||||
|
|
||||||
def password_auth_supported(self):
|
|
||||||
return True
|
|
||||||
|
|
||||||
def validate_password(self, username, password):
|
|
||||||
return username == USERNAME and password == PASSWORD
|
|
||||||
|
|
||||||
|
|
||||||
async def custom_bash_process(process):
|
|
||||||
"""Spawns a custom bash shell process"""
|
|
||||||
env = os.environ.copy()
|
|
||||||
env["TERM"] = "xterm-256color"
|
|
||||||
|
|
||||||
# Start the Bash shell
|
|
||||||
bash_proc = await asyncio.create_subprocess_exec(
|
|
||||||
SHELL,
|
|
||||||
"-i",
|
|
||||||
stdin=asyncio.subprocess.PIPE,
|
|
||||||
stdout=asyncio.subprocess.PIPE,
|
|
||||||
stderr=asyncio.subprocess.PIPE,
|
|
||||||
env=env,
|
|
||||||
)
|
|
||||||
|
|
||||||
async def read_output():
|
|
||||||
while True:
|
|
||||||
data = await bash_proc.stdout.read(1)
|
|
||||||
if not data:
|
|
||||||
break
|
|
||||||
process.stdout.write(data)
|
|
||||||
|
|
||||||
async def read_input():
|
|
||||||
while True:
|
|
||||||
data = await process.stdin.read(1)
|
|
||||||
if not data:
|
|
||||||
break
|
|
||||||
bash_proc.stdin.write(data)
|
|
||||||
|
|
||||||
await asyncio.gather(read_output(), read_input())
|
|
||||||
|
|
||||||
|
|
||||||
async def start_ssh_server():
|
|
||||||
"""Starts the AsyncSSH server with Bash"""
|
|
||||||
await asyncssh.create_server(
|
|
||||||
lambda: CustomSSHServer(),
|
|
||||||
host=HOST,
|
|
||||||
port=PORT,
|
|
||||||
server_host_keys=["ssh_host_key"],
|
|
||||||
process_factory=custom_bash_process,
|
|
||||||
)
|
|
||||||
print(f"SSH server running on {HOST}:{PORT}")
|
|
||||||
await asyncio.Future() # Keep running
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
try:
|
|
||||||
asyncio.run(start_ssh_server())
|
|
||||||
except (OSError, asyncssh.Error) as e:
|
|
||||||
print(f"Error starting SSH server: {e}")
|
|
||||||
@ -1,74 +0,0 @@
|
|||||||
#!/usr/bin/env python3.7
|
|
||||||
#
|
|
||||||
# Copyright (c) 2013-2024 by Ron Frederick <ronf@timeheart.net> and others.
|
|
||||||
#
|
|
||||||
# This program and the accompanying materials are made available under
|
|
||||||
# the terms of the Eclipse Public License v2.0 which accompanies this
|
|
||||||
# distribution and is available at:
|
|
||||||
#
|
|
||||||
# http://www.eclipse.org/legal/epl-2.0/
|
|
||||||
#
|
|
||||||
# This program may also be made available under the following secondary
|
|
||||||
# licenses when the conditions for such availability set forth in the
|
|
||||||
# Eclipse Public License v2.0 are satisfied:
|
|
||||||
#
|
|
||||||
# GNU General Public License, Version 2.0, or any later versions of
|
|
||||||
# that license
|
|
||||||
#
|
|
||||||
# SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
|
|
||||||
#
|
|
||||||
# Contributors:
|
|
||||||
# Ron Frederick - initial implementation, API, and documentation
|
|
||||||
|
|
||||||
# To run this program, the file ``ssh_host_key`` must exist with an SSH
|
|
||||||
# private key in it to use as a server host key. An SSH host certificate
|
|
||||||
# can optionally be provided in the file ``ssh_host_key-cert.pub``.
|
|
||||||
#
|
|
||||||
# The file ``ssh_user_ca`` must exist with a cert-authority entry of
|
|
||||||
# the certificate authority which can sign valid client certificates.
|
|
||||||
|
|
||||||
import asyncio
|
|
||||||
import sys
|
|
||||||
|
|
||||||
import asyncssh
|
|
||||||
|
|
||||||
|
|
||||||
async def handle_client(process: asyncssh.SSHServerProcess) -> None:
|
|
||||||
|
|
||||||
width, height, pixwidth, pixheight = process.term_size
|
|
||||||
|
|
||||||
process.stdout.write(
|
|
||||||
f"Terminal type: {process.term_type}, " f"size: {width}x{height}"
|
|
||||||
)
|
|
||||||
if pixwidth and pixheight:
|
|
||||||
process.stdout.write(f" ({pixwidth}x{pixheight} pixels)")
|
|
||||||
process.stdout.write("\nTry resizing your window!\n")
|
|
||||||
|
|
||||||
while not process.stdin.at_eof():
|
|
||||||
try:
|
|
||||||
await process.stdin.read()
|
|
||||||
except asyncssh.TerminalSizeChanged as exc:
|
|
||||||
process.stdout.write(f"New window size: {exc.width}x{exc.height}")
|
|
||||||
if exc.pixwidth and exc.pixheight:
|
|
||||||
process.stdout.write(f" ({exc.pixwidth}" f"x{exc.pixheight} pixels)")
|
|
||||||
process.stdout.write("\n")
|
|
||||||
|
|
||||||
|
|
||||||
async def start_server() -> None:
|
|
||||||
await asyncssh.listen(
|
|
||||||
"",
|
|
||||||
2230,
|
|
||||||
server_host_keys=["ssh_host_key"],
|
|
||||||
# authorized_client_keys='ssh_user_ca',
|
|
||||||
process_factory=handle_client,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
loop = asyncio.new_event_loop()
|
|
||||||
|
|
||||||
try:
|
|
||||||
loop.run_until_complete(start_server())
|
|
||||||
except (OSError, asyncssh.Error) as exc:
|
|
||||||
sys.exit("Error starting server: " + str(exc))
|
|
||||||
|
|
||||||
loop.run_forever()
|
|
||||||
@ -1,90 +0,0 @@
|
|||||||
#!/usr/bin/env python3.7
|
|
||||||
#
|
|
||||||
# Copyright (c) 2013-2024 by Ron Frederick <ronf@timeheart.net> and others.
|
|
||||||
#
|
|
||||||
# This program and the accompanying materials are made available under
|
|
||||||
# the terms of the Eclipse Public License v2.0 which accompanies this
|
|
||||||
# distribution and is available at:
|
|
||||||
#
|
|
||||||
# http://www.eclipse.org/legal/epl-2.0/
|
|
||||||
#
|
|
||||||
# This program may also be made available under the following secondary
|
|
||||||
# licenses when the conditions for such availability set forth in the
|
|
||||||
# Eclipse Public License v2.0 are satisfied:
|
|
||||||
#
|
|
||||||
# GNU General Public License, Version 2.0, or any later versions of
|
|
||||||
# that license
|
|
||||||
#
|
|
||||||
# SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
|
|
||||||
#
|
|
||||||
# Contributors:
|
|
||||||
# Ron Frederick - initial implementation, API, and documentation
|
|
||||||
|
|
||||||
# To run this program, the file ``ssh_host_key`` must exist with an SSH
|
|
||||||
# private key in it to use as a server host key. An SSH host certificate
|
|
||||||
# can optionally be provided in the file ``ssh_host_key-cert.pub``.
|
|
||||||
|
|
||||||
import asyncio
|
|
||||||
import sys
|
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
import asyncssh
|
|
||||||
import bcrypt
|
|
||||||
|
|
||||||
passwords = {
|
|
||||||
"guest": b"", # guest account with no password
|
|
||||||
"user": bcrypt.hashpw(b"user", bcrypt.gensalt()),
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def handle_client(process: asyncssh.SSHServerProcess) -> None:
|
|
||||||
username = process.get_extra_info("username")
|
|
||||||
process.stdout.write(f"Welcome to my SSH server, {username}!\n")
|
|
||||||
# process.exit(0)
|
|
||||||
|
|
||||||
|
|
||||||
class MySSHServer(asyncssh.SSHServer):
|
|
||||||
def connection_made(self, conn: asyncssh.SSHServerConnection) -> None:
|
|
||||||
peername = conn.get_extra_info("peername")[0]
|
|
||||||
print(f"SSH connection received from {peername}.")
|
|
||||||
|
|
||||||
def connection_lost(self, exc: Optional[Exception]) -> None:
|
|
||||||
if exc:
|
|
||||||
print("SSH connection error: " + str(exc), file=sys.stderr)
|
|
||||||
else:
|
|
||||||
print("SSH connection closed.")
|
|
||||||
|
|
||||||
def begin_auth(self, username: str) -> bool:
|
|
||||||
# If the user's password is the empty string, no auth is required
|
|
||||||
return passwords.get(username) != b""
|
|
||||||
|
|
||||||
def password_auth_supported(self) -> bool:
|
|
||||||
return True
|
|
||||||
|
|
||||||
def validate_password(self, username: str, password: str) -> bool:
|
|
||||||
if username not in passwords:
|
|
||||||
return False
|
|
||||||
pw = passwords[username]
|
|
||||||
if not password and not pw:
|
|
||||||
return True
|
|
||||||
return bcrypt.checkpw(password.encode("utf-8"), pw)
|
|
||||||
|
|
||||||
|
|
||||||
async def start_server() -> None:
|
|
||||||
await asyncssh.create_server(
|
|
||||||
MySSHServer,
|
|
||||||
"",
|
|
||||||
2231,
|
|
||||||
server_host_keys=["ssh_host_key"],
|
|
||||||
process_factory=handle_client,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
loop = asyncio.new_event_loop()
|
|
||||||
|
|
||||||
try:
|
|
||||||
loop.run_until_complete(start_server())
|
|
||||||
except (OSError, asyncssh.Error) as exc:
|
|
||||||
sys.exit("Error starting server: " + str(exc))
|
|
||||||
|
|
||||||
loop.run_forever()
|
|
||||||
@ -1,112 +0,0 @@
|
|||||||
#!/usr/bin/env python3.7
|
|
||||||
#
|
|
||||||
# Copyright (c) 2016-2024 by Ron Frederick <ronf@timeheart.net> and others.
|
|
||||||
#
|
|
||||||
# This program and the accompanying materials are made available under
|
|
||||||
# the terms of the Eclipse Public License v2.0 which accompanies this
|
|
||||||
# distribution and is available at:
|
|
||||||
#
|
|
||||||
# http://www.eclipse.org/legal/epl-2.0/
|
|
||||||
#
|
|
||||||
# This program may also be made available under the following secondary
|
|
||||||
# licenses when the conditions for such availability set forth in the
|
|
||||||
# Eclipse Public License v2.0 are satisfied:
|
|
||||||
#
|
|
||||||
# GNU General Public License, Version 2.0, or any later versions of
|
|
||||||
# that license
|
|
||||||
#
|
|
||||||
# SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
|
|
||||||
#
|
|
||||||
# Contributors:
|
|
||||||
# Ron Frederick - initial implementation, API, and documentation
|
|
||||||
|
|
||||||
# To run this program, the file ``ssh_host_key`` must exist with an SSH
|
|
||||||
# private key in it to use as a server host key. An SSH host certificate
|
|
||||||
# can optionally be provided in the file ``ssh_host_key-cert.pub``.
|
|
||||||
#
|
|
||||||
# The file ``ssh_user_ca`` must exist with a cert-authority entry of
|
|
||||||
# the certificate authority which can sign valid client certificates.
|
|
||||||
|
|
||||||
import asyncio
|
|
||||||
import sys
|
|
||||||
from typing import List, cast
|
|
||||||
|
|
||||||
import asyncssh
|
|
||||||
|
|
||||||
|
|
||||||
class ChatClient:
|
|
||||||
_clients: List["ChatClient"] = []
|
|
||||||
|
|
||||||
def __init__(self, process: asyncssh.SSHServerProcess):
|
|
||||||
self._process = process
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
async def handle_client(cls, process: asyncssh.SSHServerProcess):
|
|
||||||
await cls(process).run()
|
|
||||||
|
|
||||||
async def readline(self) -> str:
|
|
||||||
return cast(str, self._process.stdin.readline())
|
|
||||||
|
|
||||||
def write(self, msg: str) -> None:
|
|
||||||
self._process.stdout.write(msg)
|
|
||||||
|
|
||||||
def broadcast(self, msg: str) -> None:
|
|
||||||
for client in self._clients:
|
|
||||||
if client != self:
|
|
||||||
client.write(msg)
|
|
||||||
|
|
||||||
def begin_auth(self, username: str) -> bool:
|
|
||||||
# If the user's password is the empty string, no auth is required
|
|
||||||
# return False
|
|
||||||
return True # passwords.get(username) != b''
|
|
||||||
|
|
||||||
def password_auth_supported(self) -> bool:
|
|
||||||
return True
|
|
||||||
|
|
||||||
def validate_password(self, username: str, password: str) -> bool:
|
|
||||||
# if username not in passwords:
|
|
||||||
# return False
|
|
||||||
# pw = passwords[username]
|
|
||||||
# if not password and not pw:
|
|
||||||
# return True
|
|
||||||
return True
|
|
||||||
# return bcrypt.checkpw(password.encode('utf-8'), pw)
|
|
||||||
|
|
||||||
async def run(self) -> None:
|
|
||||||
self.write("Welcome to chat!\n\n")
|
|
||||||
|
|
||||||
self.write("Enter your name: ")
|
|
||||||
name = (await self.readline()).rstrip("\n")
|
|
||||||
|
|
||||||
self.write(f"\n{len(self._clients)} other users are connected.\n\n")
|
|
||||||
|
|
||||||
self._clients.append(self)
|
|
||||||
self.broadcast(f"*** {name} has entered chat ***\n")
|
|
||||||
|
|
||||||
try:
|
|
||||||
async for line in self._process.stdin:
|
|
||||||
self.broadcast(f"{name}: {line}")
|
|
||||||
except asyncssh.BreakReceived:
|
|
||||||
pass
|
|
||||||
|
|
||||||
self.broadcast(f"*** {name} has left chat ***\n")
|
|
||||||
self._clients.remove(self)
|
|
||||||
|
|
||||||
|
|
||||||
async def start_server() -> None:
|
|
||||||
await asyncssh.listen(
|
|
||||||
"",
|
|
||||||
2235,
|
|
||||||
server_host_keys=["ssh_host_key"],
|
|
||||||
process_factory=ChatClient.handle_client,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
loop = asyncio.new_event_loop()
|
|
||||||
|
|
||||||
try:
|
|
||||||
loop.run_until_complete(start_server())
|
|
||||||
except (OSError, asyncssh.Error) as exc:
|
|
||||||
sys.exit("Error starting server: " + str(exc))
|
|
||||||
|
|
||||||
loop.run_forever()
|
|
||||||
Loading…
Reference in New Issue
Block a user