catApi/index.js
2022-01-24 18:39:57 +03:00

99 lines
2.3 KiB
JavaScript

const express = require('express');
const app = express();
const fs = require('fs');
require('dotenv').config();
const port = process.env.PORT;
const customurl = process.env.CUSTOMURL;
app.listen(port, () => {
console.log('App listening at http://localhost:' + port || 3005);
console.log('Godspeed, little fella!');
});
app.get('/', (req, res) => {
res.sendFile(getRandomImagePath(), (err) => {
if (err) {
res.status(err.status).end();
}
});
});
app.get('/:id', (req, res) => {
const image = getImageById(req.params.id);
if (image) {
res.sendFile(image, (err) => {
if (err) {
res.status(err.status).end();
}
});
} else {
return res.json({
error: 'Image not found'
});
}
});
app.get('/api/list', (req, res) => {
return res.json(getAllImageIds());
});
app.get('/api/random', (req, res) => {
return res.json(getRandomImageApi());
});
app.get('/api/:id', (req, res) => {
const image = getImageById(req.params.id, false);
if (image) {
return res.json({
id: parseInt(req.params.id),
url: 'https://' + customurl + req.params.id,
});
} else {
return res.json({
error: 'Image not found'
});
}
});
const getRandomImagePath = () => {
const images = fs.readdirSync('./images');
return __dirname + '/images/' + images[Math.floor(Math.random() * images.length)];
};
const getImageById = (id, path = true) => {
const images = fs.readdirSync('./images');
for (const image of images) {
if (image.substring(0, image.length - 4).split('-')[1] === id) {
if (path) {
return __dirname + '/images/' + image;
} else {
return image;
}
}
}
return null;
};
const getAllImageIds = () => {
const ids = [];
fs.readdirSync('./images').forEach(image => {
ids.push(parseInt(image.substring(0, image.length - 4) .split('-')[1]));
});
return ids.sort((a, b) => a - b);
};
const getRandomImageApi = () => {
const path = getRandomImagePath();
const id = path.substring(0, path.length - 4).split('-')[1];
return {
id: parseInt(id),
url: 'https://' + customurl + id,
};
};