func init() {
hashRouter, _ := nstuff.NewRouter("/hash$", Hash)
crawlRouter, _ := nstuff.NewRouter("/crawl$", Crawl)
indexRouter, _ := nstuff.NewRouter("$", Index)
var s nstuff.Host
s.AddRouters(crawlRouter, hashRouter, indexRouter)
s.Run()
}
func Hash(s *nstuff.Host) {
s.Print(`<!doctype html>
<html lang="vn">
<head>
<meta charset="utf-8">
<title>Image Index Uploader</title>
<script>
function submit() {
document.getElementById("form").submit();
}
`)
file, header, err := s.File("file")
if err == nil {
if err == nil {
filetype := header.Header.Get("Content-Type")
if filetype == "image/jpeg" || filetype == "image/png" || filetype == "image/gif" {
m, _, err := tipimage.Decode(file)
if err == nil {
h := md5.New()
io.Copy(h, file)
checksum := base64.URLEncoding.EncodeToString(h.Sum(nil))
phash, p := PHash(m)
FindImage(s, checksum, phash, p)
} else {
s.Print("error=invalid_image&err_detail=", err.Error())
}
}
} else {
s.Print("error=invalid_input")
}
}
s.Print(`
</script>
</head>
<body>
<form id="form" method="post" enctype="multipart/form-data" action="hash">
<input id="uploader" type="file" name="file" onchange="submit();" />
</form>
</body>
</html>
`)
}
func Crawl(s *nstuff.Host) {
page := s.Post("url")
if !ValidURL(page) {
s.Print("error: invalid image url.")
return
}
deep, err := strconv.Atoi(s.Post("deep"))
if err != nil {
deep = 0
}
NewImageIndex(s, page, deep).Index()
}
func Index(s *nstuff.Host) {
html := `<!doctype html>
<html lang="vn">
<head>
<meta charset="utf-8">
<title>Image Index</title>
<link href="/statics/css/cupertino/jquery-ui-1.9.1.custom.css" rel="stylesheet">
<script src="/statics/js/jquery-1.8.2.js"></script>
<script src="/statics/js/jquery-ui-1.9.1.custom.js"></script>
<script>
$(function() {
$("#upload_link").click(function(){
$('#upload_frame').contents().find("#uploader").click();
});
});
function ShowResult(query) {
var arr = jQuery.parseJSON(query);
appendImage(arr, arr.length, 0);
}
function appendImage(arr, len, step) {
if (step >= len) {
return;
}
$("<img/>", {
"class": "img_result",
src: arr[step].SavedLocation,
title: arr[step].PageLocation
}).tooltip().hide().appendTo("#results").show(
'slide',
'easeInOutCubic',
1000,
function() {
appendImage(arr, len, step+1)
});
}
</script>
<style>
#layout {
width: 70%;
max-width: 700px;
margin-left: auto;
margin-right: auto;
}
#file_browser {
height: 100px;
border: 5px solid #d7ebf9;
border-radius: 25px;
text-align: center;
}
.img_result {
width: 100px;
margin: 10px;
border-radius: 10px;
}
</style>
</head>
<body>
<div id="layout">
<div class="ui-widget-content" id="file_browser">
<h1 id="upload_link">Select an Image!</h1>
<iframe id="upload_frame" class="ui-helper-hidden" src="hash"></iframe>
</div>
<div id="results"></div>
<form method="post" action="crawl">
<input type="text" name="url" />
<select name="deep">
<option value="-1">All</option>
<option value="0" selected="selected">Only this</option>
<option value="1">1 more levels</option>
<option value="3">3 more levels</option>
<option value="9">9 more levels</option>
<option value="27">27 more levels</option>
</select>
<input type="submit" value="upload" />
</form>
</div>
</body>
</html>`
s.Print(html)
}