This repository has been archived on 2023-03-15. You can view files and clone it, but cannot push or open issues or pull requests.
infrastructure/docker/netiquette/index.mjs

60 lines
1.7 KiB
JavaScript
Raw Normal View History

2019-06-01 14:02:49 +00:00
'use strict'
import consul from 'consul'
import { exec } from './src/io/run.mjs'
import { readFile } from './src/io/files.mjs'
2019-12-04 17:04:30 +00:00
import ctlg_control_loop from './src/catalog/control_loop.mjs'
2019-06-01 14:02:49 +00:00
import ctlg_consul from './src/catalog/consul.mjs'
import inj_iptables from './src/injector/iptables.mjs'
const get_args = () => process
.argv
.slice(2)
.map(a => a.split('='))
.reduce((dict, tuple) => {
dict[tuple[0]] = tuple.length > 1 ? tuple[1] : null
return dict
}, {})
/**
* If we have multiple catalogs
* we cache the results of the other ones
*/
function* notifications_aggregator(injectors) {
const states = []
for(let idx = 0; true; idx++) {
yield async (tag_list) => {
states[idx] = tag_list
2019-12-04 17:04:30 +00:00
const merged = states.reduce((acc, tag) => [...acc, ...tag], [])
2019-12-06 08:53:59 +00:00
await Promise.all(injectors.map(notify => notify(merged)))
2019-06-01 14:02:49 +00:00
}
}
}
const main = async () => {
try {
const args = get_args()
// Initialize all injectors
const injectors = [
2019-12-04 17:04:30 +00:00
await inj_iptables(args.ipt_base, readFile, exec, console.log),
// await inj_upnp
2019-06-01 14:02:49 +00:00
]
// Initialize all catalogs and map them to the injectors
const aggr = notifications_aggregator(injectors)
const catalogs = [
2019-12-04 17:04:30 +00:00
// this catalog is used to defeat deriving config due to single resource updated async. by multiple prog or by external program not tracked by catalogs
await ctlg_control_loop(setInterval, 60000, aggr.next().value),
2019-06-01 14:02:49 +00:00
await ctlg_consul(args.node, consul(), console.log, aggr.next().value)
]
console.log("[main] initialized")
} catch(e) {
console.error("initialization failed", e)
process.exit(1)
}
}
main()