cameron/camdn
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Improve environment.

Browse source
This commit is contained in:
Cameron Redmore 2025-02-15 13:49:18 +00:00
parent d0e6ad2c8e
commit 03bfd46be7
2 changed files with 27 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

6
.env.example
View file

@ -1,3 +1,7 @@
UPLOAD_DIR=uploads UPLOAD_DIR=uploads
API_KEY=TestAPIKey API_KEY=TestAPIKey
HOST=http://localhost:3000 HOST=http://localhost:3000
FILE_SIZE_LIMIT=1
SITE_NAME=CamDN
LISTEN_HOST=0.0.0.0
LISTEN_PORT=3000

34
index.js
View file

@ -20,6 +20,16 @@ const db = new sqlite3.Database(dbFile)
const execAsync = promisify(exec) const execAsync = promisify(exec)
const {
UPLOAD_DIR,
FILE_SIZE_LIMIT,
API_KEY,
HOST,
SITE_NAME,
LISTEN_HOST,
LISTEN_PORT
} = process.env
// Load environment variables // Load environment variables
dotenv.config() dotenv.config()
@ -27,7 +37,7 @@ const fastify = Fastify({
logger: true logger: true
}) })
let uploadDir = process.env.UPLOAD_DIR || 'uploads' let uploadDir = UPLOAD_DIR || 'uploads'
//Convert to absolute path //Convert to absolute path
if (!path.isAbsolute(uploadDir)) { if (!path.isAbsolute(uploadDir)) {
@ -107,7 +117,7 @@ fastify.get('/s/:date/:filename', async (request, reply) => {
filename = sanitize(filename) filename = sanitize(filename)
date = sanitize(date) date = sanitize(date)
const uploadDir = process.env.UPLOAD_DIR || 'uploads' const uploadDir = UPLOAD_DIR || 'uploads'
const filePath = path.join(uploadDir, date, filename) const filePath = path.join(uploadDir, date, filename)
const joinedPath = path.join(date, filename) const joinedPath = path.join(date, filename)
@ -122,8 +132,8 @@ fastify.get('/s/:date/:filename', async (request, reply) => {
const mimeType = mime.lookup(filename) || 'application/octet-stream' const mimeType = mime.lookup(filename) || 'application/octet-stream'
let fileContent = ''; let fileContent = '';
let ogTags = `<meta property="og:title" content="${process.env.SITE_NAME || 'CamDN'} - ${filename}" /> let ogTags = `<meta property="og:title" content="${SITE_NAME || 'CamDN'} - ${filename}" />
<meta property="og:url" content="${process.env.HOST || ''}/s/${date}/${filename}" />\n`; <meta property="og:url" content="${HOST || ''}/s/${date}/${filename}" />\n`;
if (mimeType.startsWith('image/')) { if (mimeType.startsWith('image/')) {
//If the user agent is DiscordBot, return the image directly //If the user agent is DiscordBot, return the image directly
@ -167,7 +177,7 @@ fastify.get('/s/:date/:filename', async (request, reply) => {
.replaceAll('{{ogTags}}', ogTags) .replaceAll('{{ogTags}}', ogTags)
.replaceAll('{{filename}}', filename) .replaceAll('{{filename}}', filename)
.replaceAll('{{joinedPath}}', joinedPath) .replaceAll('{{joinedPath}}', joinedPath)
.replaceAll('{{siteName}}', process.env.SITE_NAME || 'CamDN') .replaceAll('{{siteName}}', SITE_NAME || 'CamDN')
reply.code(200).header('Content-Type', 'text/html').send(html) reply.code(200).header('Content-Type', 'text/html').send(html)
@ -198,7 +208,7 @@ fastify.get('/l/:shortId', async (request, reply) => {
} }
}); });
const sizeLimit = process.env.FILE_SIZE_LIMIT || 16; //Size limit in GB const sizeLimit = FILE_SIZE_LIMIT || 16; //Size limit in GB
// Register multipart support // Register multipart support
fastify.register(fastifyMultipart, { fastify.register(fastifyMultipart, {
@ -212,7 +222,7 @@ fastify.register(fastifyMultipart, {
fastify.put('/upload', { fastify.put('/upload', {
preHandler: async (request, reply) => { preHandler: async (request, reply) => {
const apiKey = request.headers.authorization const apiKey = request.headers.authorization
if (!apiKey || apiKey !== process.env.API_KEY) { if (!apiKey || apiKey !== API_KEY) {
reply.code(401).send({ error: 'Unauthorized' }) reply.code(401).send({ error: 'Unauthorized' })
} }
} }
@ -224,7 +234,7 @@ fastify.put('/upload', {
return return
} }
const uploadDir = process.env.UPLOAD_DIR || 'uploads' const uploadDir = UPLOAD_DIR || 'uploads'
const date = new Date() const date = new Date()
const dateFolder = `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, '0')}-${String(date.getDate()).padStart(2, '0')}` const dateFolder = `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, '0')}-${String(date.getDate()).padStart(2, '0')}`
const fileName = data.filename const fileName = data.filename
@ -239,7 +249,7 @@ fastify.put('/upload', {
data.file, data.file,
createWriteStream(filePath) createWriteStream(filePath)
) )
const host = process.env.HOST || '' const host = HOST || ''
reply.code(200).send({ reply.code(200).send({
message: 'File uploaded successfully', message: 'File uploaded successfully',
fileName: host.replace(/\/$/, '') + '/s/' + path.join(dateFolder, fileName) fileName: host.replace(/\/$/, '') + '/s/' + path.join(dateFolder, fileName)
@ -264,7 +274,7 @@ fastify.put('/upload', {
fastify.put('/shorten', { fastify.put('/shorten', {
preHandler: async (request, reply) => { preHandler: async (request, reply) => {
const apiKey = request.headers.authorization const apiKey = request.headers.authorization
if (!apiKey || apiKey !== process.env.API_KEY) { if (!apiKey || apiKey !== API_KEY) {
reply.code(401).send({ error: 'Unauthorized' }) reply.code(401).send({ error: 'Unauthorized' })
} }
} }
@ -281,7 +291,7 @@ fastify.put('/shorten', {
//Insert the short URL into the database //Insert the short URL into the database
db.run('INSERT INTO short_urls (short_id, url) VALUES (?, ?)', [shortId, url]) db.run('INSERT INTO short_urls (short_id, url) VALUES (?, ?)', [shortId, url])
const host = process.env.HOST || '' const host = HOST || ''
reply.code(200).send({ url: host.replace(/\/$/, '') + '/l/' + shortId }) reply.code(200).send({ url: host.replace(/\/$/, '') + '/l/' + shortId })
}); });
@ -305,7 +315,7 @@ const start = async () => {
try { try {
await initDb(); await initDb();
await fastify.listen({ host: process.env.LISTEN_HOST || '0.0.0.0', port: process.env.LISTEN_PORT || 3000 }) await fastify.listen({ host: LISTEN_HOST || '0.0.0.0', port: LISTEN_PORT || 3000 })
} catch (err) { } catch (err) {
fastify.log.error(err) fastify.log.error(err)
process.exit(1) process.exit(1)