RageStudio/linebridge
1
0
Fork 0
mirror of https://github.com/ragestudio/linebridge.git synced 2025-06-09 18:44:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

added ids & limit per address

Browse Source
This commit is contained in:
srgooglo 2021-06-15 17:41:05 +02:00
parent dfb8735a15
commit c1c9e74df5
1 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
packages/nethub/src/index.js
View File

@ -11,6 +11,7 @@ const { websocket } = require("corenode").net
//* GLOBALS //* GLOBALS
const SERVER_REGISTRY = "server.registry" const SERVER_REGISTRY = "server.registry"
const SERVER_GENFILE = "origin.server" const SERVER_GENFILE = "origin.server"
const LIMIT_PER_ADDRESS = 5
const SERVER_REGISTRYPATH = path.resolve(process.cwd(), SERVER_REGISTRY) const SERVER_REGISTRYPATH = path.resolve(process.cwd(), SERVER_REGISTRY)
const SERVER_GENFILEPATH = path.resolve(process.cwd(), SERVER_GENFILE) const SERVER_GENFILEPATH = path.resolve(process.cwd(), SERVER_GENFILE)
@ -20,9 +21,14 @@ const SERVER = require("express")()
//* SERVER HUB REGISTRY //* SERVER HUB REGISTRY
const HUB = { const HUB = {
ids: {},
addresses: {}, // not storaged to registry
entries: [], entries: [],
oids: [], oids: [],
registry: {}, registry: {},
add: (payload) => {
},
set: (oid, data) => { set: (oid, data) => {
HUB.registry[oid] = data HUB.registry[oid] = data
@ -30,7 +36,9 @@ const HUB = {
}, },
del: (oid) => { del: (oid) => {
const addressIndex = HUB.oids.indexOf(oid) const addressIndex = HUB.oids.indexOf(oid)
const item = HUB.registry[oid]
delete HUB.ids[item.id]
delete HUB.registry[oid] delete HUB.registry[oid]
delete HUB.oids[addressIndex] delete HUB.oids[addressIndex]
delete HUB.entries[addressIndex] delete HUB.entries[addressIndex]
@ -39,6 +47,7 @@ const HUB = {
}, },
update: () => { update: () => {
const data = { const data = {
ids: HUB.ids,
entries: HUB.entries, entries: HUB.entries,
oids: HUB.oids, oids: HUB.oids,
registry: HUB.registry, registry: HUB.registry,
@ -106,7 +115,7 @@ function getUptime() {
return now - lastStart return now - lastStart
} }
function getRegistryFromEntry(entry){ function getRegistryFromEntry(entry) {
const index = HUB.entries.indexOf(entry) const index = HUB.entries.indexOf(entry)
const oid = HUB.oids[index] const oid = HUB.oids[index]
@ -203,11 +212,21 @@ function start() {
}) })
SERVER.put("/registry", (req, res, next) => { SERVER.put("/registry", (req, res, next) => {
let { entry, oid } = req.body let { entry, oid, id } = req.body
const address = req.headers['x-real-ip'] || req.connection.remoteAddress const address = req.headers['x-real-ip'] || req.connection.remoteAddress
let mutation = {} let mutation = {}
//? check address quota usage
if (typeof HUB.addresses[address] !== "undefined") {
if (HUB.addresses[address] >= LIMIT_PER_ADDRESS) {
res.status(403)
return res.json({
error: `[${address}] This address has exceeded the maximum number of registries [MAX ${LIMIT_PER_ADDRESS}]`
})
}
}
//? validate oid token //? validate oid token
if (typeof oid !== "undefined") { if (typeof oid !== "undefined") {
if (!TOKENIZER.valid(oid)) { if (!TOKENIZER.valid(oid)) {
@ -245,13 +264,24 @@ function start() {
oid = TOKENIZER.generate(address) oid = TOKENIZER.generate(address)
} }
//? add to hub
HUB.entries.push(entry) HUB.entries.push(entry)
HUB.oids.push(oid) HUB.oids.push(oid)
HUB.addresses[address] = Number(addresses[address]) + 1 ?? Number(1)
mutation.oid = oid mutation.oid = oid
mutation.created = Date.now() mutation.created = Date.now()
} }
if (typeof id === "string") {
if (typeof HUB.ids[id] !== "undefined") {
return false
}
HUB.ids[id] = oid
mutation.id = id
}
mutation["address"] = address mutation["address"] = address
mutation["entry"] = entry mutation["entry"] = entry