Renamed one folder
This commit is contained in:
SinTan1729 committed
1 parent
b803f5805d
commit
8956519a10
6 files changed
+7
-7
No files matched your search
@@ -0,0 +1,44 @@
|
||||
package tk.SinTan1729.url;
|
||||
|
||||
import static spark.Spark.*;
|
||||
|
||||
public class App {
|
||||
|
||||
public static void main(String[] args) {
|
||||
// Useful for developing the frontend
|
||||
// http://sparkjava.com/documentation#examples-and-faq -> How do I enable automatic refresh of static files?
|
||||
if (System.getenv("dev") != null) {
|
||||
String projectDir = System.getProperty("user.dir");
|
||||
String staticDir = "/src/main/resources/public";
|
||||
staticFiles.externalLocation(projectDir + staticDir);
|
||||
} else {
|
||||
staticFiles.location("/public");
|
||||
}
|
||||
|
||||
port(Integer.parseInt(System.getenv().getOrDefault("port", "4567")));
|
||||
|
||||
// Add GZIP compression
|
||||
after(Filters::addGZIP);
|
||||
|
||||
// No need to auth in dev
|
||||
if (System.getenv("dev") == null && Utils.isPasswordEnabled()) {
|
||||
// Authenticate
|
||||
before("/api/*", Filters.createAuthFilter());
|
||||
}
|
||||
|
||||
get("/", (req, res) -> {
|
||||
res.redirect("/index.html");
|
||||
return "Redirect";
|
||||
});
|
||||
|
||||
|
||||
path("/api", () -> {
|
||||
get("/all", Routes::getAll);
|
||||
post("/new", Routes::addUrl);
|
||||
delete("/:shortUrl", Routes::delete);
|
||||
get("/site", Routes::getSiteUrl);
|
||||
});
|
||||
|
||||
get("/:shortUrl", Routes::goToLongUrl);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,20 @@
|
||||
package tk.SinTan1729.url;
|
||||
|
||||
import com.qmetric.spark.authentication.AuthenticationDetails;
|
||||
import com.qmetric.spark.authentication.BasicAuthenticationFilter;
|
||||
import spark.Filter;
|
||||
import spark.Request;
|
||||
import spark.Response;
|
||||
|
||||
public class Filters {
|
||||
public static void addGZIP(Request request, Response response) {
|
||||
response.header("Content-Encoding", "gzip");
|
||||
}
|
||||
|
||||
public static Filter createAuthFilter() {
|
||||
String username = System.getenv("username");
|
||||
String password = System.getenv("password");
|
||||
|
||||
return new BasicAuthenticationFilter(new AuthenticationDetails(username, password));
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,76 @@
|
||||
package tk.SinTan1729.url;
|
||||
|
||||
import org.eclipse.jetty.http.HttpStatus;
|
||||
import spark.Request;
|
||||
import spark.Response;
|
||||
|
||||
public class Routes {
|
||||
|
||||
private static final UrlRepository urlRepository;
|
||||
|
||||
static {
|
||||
urlRepository = new UrlRepository();
|
||||
}
|
||||
|
||||
public static String getAll(Request req, Response res) {
|
||||
return String.join("\n", urlRepository.getAll());
|
||||
}
|
||||
|
||||
public static String addUrl(Request req, Response res) {
|
||||
var body = req.body();
|
||||
var split = body.split(";");
|
||||
String longUrl = split[0];
|
||||
boolean unique = false;
|
||||
|
||||
String shortUrl;
|
||||
try {
|
||||
shortUrl = split[1];
|
||||
shortUrl = shortUrl.toLowerCase();
|
||||
if (urlRepository.findForShortUrl(shortUrl).isEmpty()) {
|
||||
unique = true;
|
||||
}
|
||||
} catch (ArrayIndexOutOfBoundsException e) {
|
||||
do {
|
||||
shortUrl = Utils.randomName();
|
||||
if (urlRepository.findForShortUrl(shortUrl).isEmpty()) {
|
||||
unique = true;
|
||||
}
|
||||
} while (unique == false);
|
||||
}
|
||||
|
||||
if (unique && Utils.validate(shortUrl)) {
|
||||
return urlRepository.addUrl(longUrl, shortUrl);
|
||||
} else {
|
||||
res.status(HttpStatus.BAD_REQUEST_400);
|
||||
return "shortUrl not valid or already in use";
|
||||
}
|
||||
}
|
||||
|
||||
public static String getSiteUrl(Request req, Response res) {
|
||||
return System.getenv().getOrDefault("site_url", "unset");
|
||||
}
|
||||
|
||||
|
||||
public static String goToLongUrl(Request req, Response res) {
|
||||
String shortUrl = req.params("shortUrl");
|
||||
shortUrl = shortUrl.toLowerCase();
|
||||
var longUrlOpt = urlRepository
|
||||
.findForShortUrl(shortUrl);
|
||||
|
||||
if (longUrlOpt.isEmpty()) {
|
||||
res.status(404);
|
||||
return "";
|
||||
}
|
||||
urlRepository.addHit(shortUrl);
|
||||
res.redirect(longUrlOpt.get(), HttpStatus.PERMANENT_REDIRECT_308);
|
||||
|
||||
return "";
|
||||
}
|
||||
|
||||
public static String delete(Request req, Response res) {
|
||||
String shortUrl = req.params("shortUrl");
|
||||
|
||||
urlRepository.deleteEntry(shortUrl);
|
||||
return "";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,117 @@
|
||||
package tk.SinTan1729.url;
|
||||
|
||||
import java.sql.*;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Optional;
|
||||
|
||||
|
||||
public class UrlRepository {
|
||||
private static final String INSERT_ROW_SQL = "INSERT INTO urls (long_url, short_url, hits) VALUES (?, ?, ?)";
|
||||
private static final String ADD_HIT_SQL = "UPDATE urls SET hits = hits + 1 WHERE short_url = ?";
|
||||
private static final String CREATE_TABLE_SQL = "CREATE TABLE IF NOT EXISTS urls\n" +
|
||||
"(\n" +
|
||||
" id INTEGER PRIMARY KEY AUTOINCREMENT,\n" +
|
||||
" long_url TEXT NOT NULL,\n" +
|
||||
" short_url TEXT NOT NULL,\n" +
|
||||
" hits INTEGER NOT NULL\n" +
|
||||
");";
|
||||
private static final String SELECT_FOR_SHORT_SQL = "SELECT long_url FROM urls WHERE short_url = ?";
|
||||
private static final String DELETE_ROW_SQL = "DELETE FROM urls WHERE short_url = ?";
|
||||
|
||||
private final String databaseUrl;
|
||||
|
||||
|
||||
public UrlRepository() {
|
||||
String path = System.getenv().getOrDefault("db_url", "/urls.sqlite");
|
||||
|
||||
databaseUrl = "jdbc:sqlite:" + path;
|
||||
|
||||
try (Connection conn = DriverManager.getConnection(databaseUrl)) {
|
||||
if (conn != null) {
|
||||
DatabaseMetaData meta = conn.getMetaData();
|
||||
|
||||
conn.createStatement()
|
||||
.execute(CREATE_TABLE_SQL);
|
||||
|
||||
System.out.println("Database initialised");
|
||||
}
|
||||
|
||||
} catch (SQLException e) {
|
||||
System.out.println(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
public List<String> getAll() {
|
||||
try (final var con = DriverManager.getConnection(databaseUrl)) {
|
||||
var statement = con.createStatement();
|
||||
|
||||
statement.execute("SELECT * FROM urls");
|
||||
ResultSet rs = statement.getResultSet();
|
||||
|
||||
List<String> result = new ArrayList<>();
|
||||
|
||||
while (rs.next()) {
|
||||
result.add(String.format("%s,%s,%s", rs.getString("short_url"), rs.getString("long_url"), String.valueOf(rs.getInt("hits"))));
|
||||
}
|
||||
|
||||
return result;
|
||||
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return List.of();
|
||||
}
|
||||
|
||||
public String addUrl(String longURL, String shortUrl) {
|
||||
try (final var con = DriverManager.getConnection(databaseUrl)) {
|
||||
final var stmt = con.prepareStatement(INSERT_ROW_SQL);
|
||||
stmt.setString(1, longURL);
|
||||
stmt.setString(2, shortUrl);
|
||||
stmt.setInt(3, 0);
|
||||
if (stmt.execute()) {
|
||||
return String.format("%s,%s", shortUrl, longURL);
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return shortUrl;
|
||||
}
|
||||
|
||||
public void addHit(String shortURL) {
|
||||
try (final var con = DriverManager.getConnection(databaseUrl)) {
|
||||
final var stmt = con.prepareStatement(ADD_HIT_SQL);
|
||||
stmt.setString(1, shortURL);
|
||||
stmt.execute();
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
public Optional<String> findForShortUrl(String shortUrl) {
|
||||
try (final var con = DriverManager.getConnection(databaseUrl)) {
|
||||
final var stmt = con.prepareStatement(SELECT_FOR_SHORT_SQL);
|
||||
stmt.setString(1, shortUrl);
|
||||
if (stmt.execute()) {
|
||||
ResultSet rs = stmt.getResultSet();
|
||||
if (rs.next()) {
|
||||
return Optional.of(rs.getString("long_url"));
|
||||
}
|
||||
}
|
||||
return Optional.empty();
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return Optional.empty();
|
||||
}
|
||||
|
||||
public void deleteEntry(String shortUrl) {
|
||||
try (final var con = DriverManager.getConnection(databaseUrl)) {
|
||||
final var stmt = con.prepareStatement(DELETE_ROW_SQL);
|
||||
stmt.setString(1, shortUrl);
|
||||
stmt.execute();
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,63 @@
|
||||
package tk.SinTan1729.url;
|
||||
|
||||
import java.util.Random;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
public class Utils {
|
||||
private static final Random random = new Random(System.currentTimeMillis());
|
||||
|
||||
private static final String SHORT_URL_PATTERN = "[a-z0-9-_]+";
|
||||
private static final Pattern PATTERN = Pattern.compile(SHORT_URL_PATTERN);
|
||||
|
||||
// The following lists are modified versions of the strings in
|
||||
// https://github.com/moby/moby/blob/master/pkg/namesgenerator/names-generator.go
|
||||
|
||||
private static final String[] adjective = new String[] {"admiring", "adoring", "affectionate", "agitated", "amazing", "angry", "awesome", "beautiful",
|
||||
"blissful", "bold", "boring", "brave", "busy", "charming", "clever", "compassionate", "competent", "condescending", "confident", "cool",
|
||||
"cranky", "crazy", "dazzling", "determined", "distracted", "dreamy", "eager", "ecstatic", "elastic", "elated", "elegant", "eloquent", "epic",
|
||||
"exciting", "fervent", "festive", "flamboyant", "focused", "friendly", "frosty", "funny", "gallant", "gifted", "goofy", "gracious",
|
||||
"great", "happy", "hardcore", "heuristic", "hopeful", "hungry", "infallible", "inspiring", "intelligent", "interesting", "jolly",
|
||||
"jovial", "keen", "kind", "laughing", "loving", "lucid", "magical", "modest", "musing", "mystifying", "naughty", "nervous", "nice",
|
||||
"nifty", "nostalgic", "objective", "optimistic", "peaceful", "pedantic", "pensive", "practical", "priceless", "quirky", "quizzical",
|
||||
"recursing", "relaxed", "reverent", "romantic", "sad", "serene", "sharp", "silly", "sleepy", "stoic", "strange", "stupefied", "suspicious",
|
||||
"sweet", "tender", "thirsty", "trusting", "unruffled", "upbeat", "vibrant", "vigilant", "vigorous", "wizardly", "wonderful", "xenodochial",
|
||||
"youthful", "zealous", "zen"};
|
||||
|
||||
private static final String[] name = new String[] {"agnesi", "albattani", "allen", "almeida", "antonelli", "archimedes", "ardinghelli", "aryabhata", "austin",
|
||||
"babbage", "banach", "banzai", "bardeen", "bartik", "bassi", "beaver", "bell", "benz", "bhabha", "bhaskara", "black", "blackburn", "blackwell",
|
||||
"bohr", "booth", "borg", "bose", "bouman", "boyd", "brahmagupta", "brattain", "brown", "buck", "burnell", "cannon", "carson", "cartwright",
|
||||
"carver", "cauchy", "cerf", "chandrasekhar", "chaplygin", "chatelet", "chatterjee", "chaum", "chebyshev", "clarke", "cohen", "colden", "cori",
|
||||
"cray", "curie", "curran", "darwin", "davinci", "dewdney", "dhawan", "diffie", "dijkstra", "dirac", "driscoll", "dubinsky", "easley", "edison",
|
||||
"einstein", "elbakyan", "elgamal", "elion", "ellis", "engelbart", "euclid", "euler", "faraday", "feistel", "fermat", "fermi", "feynman", "franklin",
|
||||
"gagarin", "galileo", "galois", "ganguly", "gates", "gauss", "germain", "goldberg", "goldstine", "goldwasser", "golick", "goodall", "gould", "greider",
|
||||
"grothendieck", "haibt", "hamilton", "hardy", "haslett", "hawking", "heisenberg", "hellman", "hermann", "herschel", "hertz", "heyrovsky", "hodgkin",
|
||||
"hofstadter", "hoover", "hopper", "hugle", "hypatia", "ishizaka", "jackson", "jang", "jemison", "jennings", "jepsen", "johnson", "joliot", "jones",
|
||||
"kalam", "kapitsa", "kare", "keldysh", "keller", "kepler", "khayyam", "khorana", "kilby", "kirch", "knuth", "kowalevski", "lalande", "lamarr",
|
||||
"lamport", "leakey", "leavitt", "lederberg", "lehmann", "lewin", "lichterman", "liskov", "lovelace", "lumiere", "mahavira", "margulis", "matsumoto",
|
||||
"maxwell", "mayer", "mccarthy", "mcclintock", "mclaren", "mclean", "mcnulty", "meitner", "mendel", "mendeleev", "meninsky", "merkle", "mestorf",
|
||||
"mirzakhani", "montalcini", "moore", "morse", "moser", "murdock", "napier", "nash", "neumann", "newton", "nightingale", "nobel", "noether", "northcutt",
|
||||
"noyce", "panini", "pare", "pascal", "pasteur", "payne", "perlman", "pike", "poincare", "poitras", "proskuriakova", "ptolemy", "raman", "ramanujan",
|
||||
"rhodes", "ride", "riemann", "ritchie", "robinson", "roentgen", "rosalind", "rubin", "saha", "sammet", "sanderson", "satoshi", "shamir", "shannon",
|
||||
"shaw", "shirley", "shockley", "shtern", "sinoussi", "snyder", "solomon", "spence", "stonebraker", "sutherland", "swanson", "swartz", "swirles",
|
||||
"taussig", "tesla", "tharp", "thompson", "torvalds", "tu", "turing", "varahamihira", "vaughan", "vaughn", "villani", "visvesvaraya", "volhard",
|
||||
"wescoff", "weierstrass", "wilbur", "wiles", "williams", "williamson", "wilson", "wing", "wozniak", "wright", "wu", "yalow", "yonath", "zhukovsky"};
|
||||
|
||||
public static boolean validate(String shortUrl) {
|
||||
return PATTERN.matcher(shortUrl)
|
||||
.matches();
|
||||
}
|
||||
|
||||
public static boolean isPasswordEnabled(){
|
||||
String disablePasswordEnv = System.getenv("INSECURE_DISABLE_PASSWORD");
|
||||
|
||||
if(disablePasswordEnv != null && disablePasswordEnv.equals("I_KNOW_ITS_BAD")){
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public static String randomName() {
|
||||
return adjective[random.nextInt(adjective.length)]+"-"+name[random.nextInt(name.length)];
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user