You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.5 KiB
JavaScript
85 lines
2.5 KiB
JavaScript
const http = require('http');
|
|
const url = require('url');
|
|
const fs = require('fs');
|
|
const ws = require('ws');
|
|
const redis = require('redis')
|
|
|
|
//redis client
|
|
const redis_client = redis.createClient()
|
|
redis_client.on("error", function (error) {
|
|
console.error(error)
|
|
})
|
|
|
|
//HTTP server
|
|
http.createServer(function (req, res) {
|
|
const q = url.parse(req.url, true);
|
|
let filename = "./public" + q.pathname;
|
|
if (filename === "./public/") filename = "./public/index.html";
|
|
fs.readFile(filename, function(err, data) {
|
|
if (err) {
|
|
res.writeHead(404, {'Content-Type': 'text/html'});
|
|
return res.end("404 Not Found");
|
|
}
|
|
res.writeHead(200, {'Content-Type': 'text/html'});
|
|
res.write(data);
|
|
return res.end();
|
|
});
|
|
}).listen(8080);
|
|
|
|
//WS server
|
|
const wss = new ws.Server({
|
|
port: 8081,
|
|
perMessageDeflate: {
|
|
zlibDeflateOptions: {
|
|
// See zlib defaults.
|
|
chunkSize: 1024,
|
|
memLevel: 7,
|
|
level: 3
|
|
},
|
|
zlibInflateOptions: {
|
|
chunkSize: 10 * 1024
|
|
},
|
|
// Other options settable:
|
|
clientNoContextTakeover: true, // Defaults to negotiated value.
|
|
serverNoContextTakeover: true, // Defaults to negotiated value.
|
|
serverMaxWindowBits: 10, // Defaults to negotiated value.
|
|
// Below options specified as default values.
|
|
concurrencyLimit: 10, // Limits zlib concurrency for perf.
|
|
threshold: 1024 // Size (in bytes) below which messages
|
|
// should not be compressed.
|
|
}
|
|
});
|
|
|
|
//WS handler
|
|
wss.on('connection', ws => {
|
|
ws.on('message', message => {
|
|
console.log(`Received message => ${message}`)
|
|
let msg = `${message}`.split(";", 2)
|
|
if (msg[0] === 'long_url') {
|
|
if (msg[1] === '') ws.send('error;url is empty')
|
|
if (msg[1].length > 2000) ws.send('error;your url is too big')
|
|
else{
|
|
let ran_key = get_key(8)
|
|
//redis_client.set("key", "value", redis.print)
|
|
ws.send('short_url;http://127.0.0.1'+ran_key)
|
|
}
|
|
}
|
|
//ws.send('got your request: '+msg)
|
|
console.log(msg[0])
|
|
})
|
|
ws.send('websocket connected')
|
|
})
|
|
|
|
//random key
|
|
const forbidden_array = ['sym', 'admin', 'stats']
|
|
function get_key(length) {
|
|
let forbidden = false; let output = ''
|
|
let characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
|
|
do{
|
|
for (let i = 0; i < length; i++ )
|
|
output += characters.charAt(Math.floor(Math.random() * characters.length))
|
|
for (let i = 0; i < forbidden_array.length; i++) if (output === forbidden_array[i]) forbidden = true
|
|
} while (forbidden)
|
|
return output
|
|
}
|