server.js

require("dotenv").config();

const mongoose = require("mongoose");

const multer = require("multer");

const express = require("express");

const bcrypt = require("bcrypt");

const TinyURL = require('tinyurl');


const File = require("./models/Files");

const app = express();
app.use(express.urlencoded({extended: true}))
app.use(express.static('views'))

const upload = multer({dest: "upload"})

// console.log(process.env.DATABASE_URL);
mongoose.connect(process.env.DATABASE_URL);


app.set("view engine","ejs");


app.get("/", (req,res)=>{
    res.render("landing");
})

app.get("/aboutus", (req,res)=>{
    res.render("aboutus");
})

// contactus.ejs
app.get("/contactus", (req,res)=>{
    res.render("contactus");
})

app.get("/index", (req,res) => {
    res.render("index");
})

app.get("/privacyPolicy", (req,res) => {
    res.render("privacyPolicy");
})
// termsandServices.ejs
app.get("/termsandServices", (req,res) => {
    res.render("termsandServices");
})


app.post("/upload",upload.single("file"),async (req,res) => {
    const fileData = {
        path: req.file.path,
        originalName: req.file.originalname
    }

    if(req.body.password !== null && req.body.password !== ""){
        fileData.password = await bcrypt.hash(req.body.password,10)
    }

    const file = await File.create(fileData);
    // console.log(file);
    // res.send(file.originalName)
    // res.render("index",{fileLink: `${req.headers.origin}/file/${file.id}`})
    res.render("index",{fileLink: `${await shortenUrl(`${req.headers.origin}/file/${file.id}`)}`})
    
})
// console.log(Number(process.env.PORT));
// console.log(process.env.DATABASE_URL);


// Tiny url code
 async function shortenUrl(url) {
    try{
        const shortUrl = await TinyURL.shorten(url);
        console.log(`original URL: ${url}`);
        console.log(`Shorten URL: ${shortUrl}`);
        return shortUrl;
    }catch(error){
        console.log(`Error shortening ${url}:`,error);
        throw error;
    }
}

// app.get("/file/:id",handleDownload);
// app.post("/file/:id",handleDownload);

// we can write the above two lines in a shorter format like below
app.route("/file/:id").get(handleDownload).post(handleDownload);


async function handleDownload(req,res){
    const file = await File.findById(req.params.id);
    // console.log(req.body);
    if(file.password != null){
        if(req.body.password == null){
            res.render("password");
            return
        }
        if(!(await bcrypt.compare(req.body.password, file.password))){
            res.render("password",{error: true});
            return
            }
    }

    file.downloadCount++;
    await file.save();
    console.log(file.downloadCount);

    res.download(file.path, file.originalName);
}

app.listen(Number(process.env.PORT));