import express, { Request, Response } from "express";
import Shortify from "../src";
import dotenv from "dotenv";
// Load environment variables
dotenv.config();
// Create Express app
const app = express();
app.use(express.json());
app.use(express.urlencoded({ extended: true }));
// Create Shortify instance
const shortify = new Shortify(
process.env.BASE_URL || "http://localhost:3000/",
process.env.MONGODB_URI || "mongodb://localhost:27017/shortify"
);
// Connect to database before starting server
async function startServer() {
try {
// Connect to MongoDB
await shortify.connect();
console.log("Connected to MongoDB");
// Set up routes
// Home page - Simple form to create shortened URLs
app.get("/", (req, res) => {
res.send(`
URL Shortener
URL Shortener
`);
});
// Shorten URL endpoint
app.post("/shorten", async (req: Request, res: Response) => {
try {
const { url, customId, expiry } = req.body;
if (!url) {
return res.status(400).json({ error: "URL is required" });
}
const options: any = {};
if (customId) {
options.customUrlId = customId;
}
if (expiry) {
options.expiresInDays = parseInt(expiry);
}
const result = await shortify.shorten(url, options);
res.send(`
URL Shortened
URL Shortened
Original URL: ${result.originalUrl}
Shortened URL:
${
result.shortUrl
}
URL ID: ${result.urlId}
${
result.expiresAt
? `
Expires at: ${new Date(
result.expiresAt
).toLocaleString()}
`
: ""
}
← Back to homepage
`);
} catch (error) {
console.error("Error shortening URL:", error);
res.status(500).json({ error: "Failed to shorten URL" });
}
});
// Redirect shortened URLs
app.get(
"/:urlId",
async (req: Request<{ urlId: string }>, res: Response) => {
try {
const { urlId } = req.params;
const originalUrl = await shortify.resolve(urlId);
if (originalUrl) {
return res.redirect(originalUrl);
} else {
return res.status(404).send(`
URL Not Found
URL Not Found
The shortened URL you're looking for doesn't exist or has expired.
← Go to homepage
`);
}
} catch (error) {
console.error("Error resolving URL:", error);
res.status(500).json({ error: "Failed to resolve URL" });
}
}
);
// API endpoint to get URL stats
app.get(
"/api/stats/:urlId",
async (req: Request<{ urlId: string }>, res: Response) => {
try {
const { urlId } = req.params;
const stats = await shortify.getStats(urlId);
if (stats) {
return res.json(stats);
} else {
return res.status(404).json({ error: "URL not found" });
}
} catch (error) {
console.error("Error getting URL stats:", error);
res.status(500).json({ error: "Failed to get URL stats" });
}
}
);
// Start the server
const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
console.log(`Server running at http://localhost:${PORT}`);
});
} catch (error) {
console.error("Failed to start server:", error);
process.exit(1);
}
}
// Graceful shutdown
process.on("SIGINT", async () => {
console.log("Shutting down server...");
await shortify.disconnect();
console.log("Disconnected from MongoDB");
process.exit(0);
});
// Start the server
startServer();