Added a basic input validation for shortUrl

This commit is contained in:
Przemek Dragańczuk 2020-02-16 16:05:09 +01:00
parent 8cd399d2e9
commit 1322569cf6
3 changed files with 23 additions and 6 deletions

View File

@ -1,5 +1,6 @@
package tk.draganczuk.url;
import org.eclipse.jetty.http.HttpStatus;
import spark.Request;
import spark.Response;
@ -29,15 +30,21 @@ public class Routes {
shortUrl = Utils.randomString();
}
return urlFile.addUrl(longUrl, shortUrl);
if (Utils.validate(shortUrl)) {
return urlFile.addUrl(longUrl, shortUrl);
} else {
res.status(HttpStatus.BAD_REQUEST_400);
return "shortUrl not valid ([a-z0-9]+)";
}
}
public static String goToLongUrl(Request req, Response res){
public static String goToLongUrl(Request req, Response res) {
String shortUrl = req.params("shortUrl");
var longUrlOpt = urlFile
.findForShortUrl(shortUrl);
.findForShortUrl(shortUrl);
if(longUrlOpt.isEmpty()){
if (longUrlOpt.isEmpty()) {
res.status(404);
return "";
}

View File

@ -1,10 +1,14 @@
package tk.draganczuk.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);
public static String randomString() {
int leftLimit = 48; // numeral '0'
int rightLimit = 122; // letter 'z'
@ -18,4 +22,9 @@ public class Utils {
StringBuilder::append)
.toString();
}
public static boolean validate(String shortUrl) {
return PATTERN.matcher(shortUrl)
.matches();
}
}

View File

@ -30,11 +30,12 @@
<legend>Add new URL</legend>
<div class="pure-control-group">
<label for="longUrl">Long URL</label>
<input type="text" name="longUrl" id="longUrl" placeholder="Long URL" required/>
<input type="url" name="longUrl" id="longUrl" placeholder="Long URL" required/>
</div>
<div class="pure-control-group">
<label for="shortUrl">Short URL (Optional)</label>
<input type="text" name="shortUrl" id="shortUrl" placeholder="Short URL (optional)"/>
<input type="text" name="shortUrl" id="shortUrl" placeholder="Short URL (optional)"
pattern="[a-z0-9]+"/>
</div>
<div class="pure-controls">
<button class="pure-button pure-button-primary">Submit</button>