backend update

master
adb-sh 4 years ago
parent 6d662d1e68
commit 40c668b6c5

@ -5,16 +5,23 @@ const ws = require('ws');
const redis = require('redis')
//redis client
const redis_client = redis.createClient()
redis_client.on("error", function (error) {
const redis_cli = redis.createClient()
redis_cli.on("error", function (error) {
console.error(error)
})
const outpath = ['sym', '']
//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";
//q.pathname.split("/", 2)[1] === "sym" || )
if (function valid_path(){
for (let i = 0; i < outpath.length; i++) if (q.pathname.split("/", 2)[1] === outpath[i]) return true
return false
}() === true){
fs.readFile(filename, function(err, data) {
if (err) {
res.writeHead(404, {'Content-Type': 'text/html'});
@ -23,7 +30,13 @@ http.createServer(function (req, res) {
res.writeHead(200, {'Content-Type': 'text/html'});
res.write(data);
return res.end();
});
})
}else{
let url = redis_cli.hget("surl;"+q.pathname.split("/", 2)[1], "url")
res.writeHead(302, {'Location': 'http://adb.sh'});
//res.write(data);
return res.end();
}
}).listen(8080);
//WS server
@ -56,12 +69,15 @@ wss.on('connection', ws => {
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{
if (function valid_url(){
if (msg[1] === ''){ws.send('error;url is empty'); return false}
else if (msg[1].length > 2000){ws.send('error;your url is too long'); return false}
else return true
}() === true){
let ran_key = get_key(8)
//redis_client.set("key", "value", redis.print)
ws.send('short_url;http://127.0.0.1'+ran_key)
redis_cli.hmset("surl;"+ran_key, "url", msg[1], "time", Date.now())
ws.send('short_url;http://127.0.0.1/'+ran_key)
}
}
//ws.send('got your request: '+msg)

Loading…
Cancel
Save