initial commit
This commit is contained in:
commit
5885b58dcd
|
|
@ -0,0 +1,236 @@
|
|||
package contented
|
||||
|
||||
import (
|
||||
"crypto/sha512"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"log"
|
||||
"mime/multipart"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/boltdb/bolt"
|
||||
)
|
||||
|
||||
var SERVER_HEADER string = `contented/0.1`
|
||||
|
||||
type ServerOptions struct {
|
||||
DataDirectory string
|
||||
DBPath string
|
||||
AppTitle string
|
||||
MaxUploadBytes int64
|
||||
}
|
||||
|
||||
type Server struct {
|
||||
opts ServerOptions
|
||||
db *bolt.DB
|
||||
metadataBucket []byte
|
||||
}
|
||||
|
||||
func NewServer(opts *ServerOptions) (*Server, error) {
|
||||
s := &Server{
|
||||
opts: *opts,
|
||||
metadataBucket: []byte(`METADATA`),
|
||||
}
|
||||
|
||||
b, err := bolt.Open(opts.DBPath, 0644, bolt.DefaultOptions)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = b.Update(func(tx *bolt.Tx) error {
|
||||
_, err := tx.CreateBucketIfNotExists(s.metadataBucket)
|
||||
return err
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
s.db = b
|
||||
|
||||
return s, nil
|
||||
}
|
||||
|
||||
type Metadata struct {
|
||||
FileSize int64
|
||||
UploadTime time.Time
|
||||
UploadIP string
|
||||
Filename string
|
||||
}
|
||||
|
||||
func (this *Server) handleUpload(src multipart.File, hdr *multipart.FileHeader, UploadIP string) (string, error) {
|
||||
|
||||
// Get file length
|
||||
srcLen, err := src.Seek(0, io.SeekEnd)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
_, err = src.Seek(0, io.SeekStart)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// Get file hash
|
||||
hasher := sha512.New512_256()
|
||||
_, err = io.CopyN(hasher, src, int64(srcLen))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
_, err = src.Seek(0, io.SeekStart)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// Save file to disk
|
||||
fileID := hex.EncodeToString(hasher.Sum(nil))
|
||||
dest, err := os.OpenFile(filepath.Join(this.opts.DataDirectory, fileID), os.O_CREATE|os.O_WRONLY, 0600)
|
||||
if err != nil {
|
||||
if os.IsExist(err) {
|
||||
return fileID, nil // hash matches existing upload
|
||||
}
|
||||
|
||||
return "", err
|
||||
}
|
||||
defer dest.Close()
|
||||
|
||||
_, err = io.CopyN(dest, src, int64(srcLen))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// Persist metadata to DB
|
||||
m := Metadata{
|
||||
Filename: hdr.Filename,
|
||||
UploadTime: time.Now(),
|
||||
UploadIP: UploadIP,
|
||||
FileSize: srcLen,
|
||||
}
|
||||
err = this.SetMetadata(fileID, m)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// Done
|
||||
return fileID, nil
|
||||
}
|
||||
|
||||
func (this *Server) Metadata(fileID string) (*Metadata, error) {
|
||||
var m Metadata
|
||||
err := this.db.View(func(tx *bolt.Tx) error {
|
||||
content := tx.Bucket(this.metadataBucket).Get([]byte(fileID))
|
||||
if len(content) == 0 {
|
||||
return os.ErrNotExist
|
||||
}
|
||||
|
||||
return json.Unmarshal(content, &m)
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &m, nil
|
||||
}
|
||||
|
||||
func (this *Server) SetMetadata(fileID string, m Metadata) error {
|
||||
jb, err := json.Marshal(m)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return this.db.Update(func(tx *bolt.Tx) error {
|
||||
return tx.Bucket(this.metadataBucket).Put([]byte(fileID), jb)
|
||||
})
|
||||
}
|
||||
|
||||
func (this *Server) handleView(w http.ResponseWriter, r *http.Request, fileID string) error {
|
||||
// Load file
|
||||
f, err := os.Open(filepath.Join(this.opts.DataDirectory, fileID))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer f.Close()
|
||||
|
||||
// Load metadata
|
||||
m, err := this.Metadata(fileID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
http.ServeContent(w, r, m.Filename, m.UploadTime, f)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (this *Server) handleInformation(w http.ResponseWriter, fileID string) {
|
||||
m, err := this.Metadata(fileID)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
http.Error(w, "Not found", 404)
|
||||
return
|
||||
}
|
||||
|
||||
log.Println(err.Error())
|
||||
http.Error(w, "Internal error", 500)
|
||||
return
|
||||
}
|
||||
|
||||
jb, err := json.Marshal(m)
|
||||
if err != nil {
|
||||
log.Println(err.Error())
|
||||
http.Error(w, "Internal error", 500)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set(`Content-Type`, `application/json`)
|
||||
w.WriteHeader(200)
|
||||
w.Write(jb)
|
||||
}
|
||||
|
||||
func (this *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
r.Header.Set(`Server`, SERVER_HEADER)
|
||||
if this.opts.MaxUploadBytes > 0 {
|
||||
r.Body = http.MaxBytesReader(w, r.Body, this.opts.MaxUploadBytes)
|
||||
}
|
||||
|
||||
if r.Method == "GET" && strings.HasPrefix(r.URL.Path, `/view/`) {
|
||||
err := this.handleView(w, r, r.URL.Path[6:])
|
||||
if err != nil {
|
||||
log.Printf("%s View failed: %s\n", r.RemoteAddr, err.Error())
|
||||
if os.IsNotExist(err) {
|
||||
http.Error(w, "File not found", 404)
|
||||
} else {
|
||||
http.Error(w, "Couldn't provide content", 500)
|
||||
}
|
||||
}
|
||||
|
||||
} else if r.Method == "GET" && strings.HasPrefix(r.URL.Path, `/info/`) {
|
||||
this.handleInformation(w, r.URL.Path[6:])
|
||||
|
||||
} else if r.Method == "POST" && r.URL.Path == `/upload` {
|
||||
mp, mph, err := r.FormFile("f")
|
||||
if err != nil {
|
||||
log.Printf("%s Invalid request: %s\n", r.RemoteAddr, err.Error())
|
||||
http.Error(w, "Invalid request", 400)
|
||||
}
|
||||
path, err := this.handleUpload(mp, mph, strings.TrimRight(r.RemoteAddr, "0123456789:")) // strip remote-port
|
||||
if err != nil {
|
||||
log.Printf("%s Upload failed: %s\n", r.RemoteAddr, err.Error())
|
||||
http.Error(w, "Upload failed", 500)
|
||||
} else {
|
||||
http.Redirect(w, r, `/view/`+path, http.StatusFound)
|
||||
}
|
||||
|
||||
} else if r.Method == "GET" && r.URL.Path == `/` {
|
||||
this.handleHomepage(w)
|
||||
|
||||
} else {
|
||||
http.Error(w, "Not found", 404)
|
||||
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,38 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
|
||||
"code.ivysaur.me/contented"
|
||||
)
|
||||
|
||||
func main() {
|
||||
cwd, _ := os.Getwd()
|
||||
|
||||
listenAddr := flag.String("listen", "127.0.0.1:80", "")
|
||||
dataDir := flag.String("data", cwd, "Directory for stored content")
|
||||
dbPath := flag.String("db", "contented.db", "Path for metadata database")
|
||||
appTitle := flag.String("title", "contented", "Title used in web interface")
|
||||
maxUploadMb := flag.Int("max", 8, "Maximum size of uploaded files in MiB (set zero for unlimited)")
|
||||
flag.Parse()
|
||||
|
||||
svr, err := contented.NewServer(&contented.ServerOptions{
|
||||
DataDirectory: *dataDir,
|
||||
DBPath: *dbPath,
|
||||
AppTitle: *appTitle,
|
||||
MaxUploadBytes: int64(*maxUploadMb) * 1024 * 1024,
|
||||
})
|
||||
if err != nil {
|
||||
log.Println(err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
err = http.ListenAndServe(*listenAddr, svr)
|
||||
if err != nil {
|
||||
log.Println(err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,143 @@
|
|||
package contented
|
||||
|
||||
import (
|
||||
"html"
|
||||
"net/http"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func (this *Server) handleHomepage(w http.ResponseWriter) {
|
||||
extraText := ""
|
||||
if this.opts.MaxUploadBytes > 0 {
|
||||
extraText = " (max " + strconv.Itoa(int(this.opts.MaxUploadBytes/(1024*1024))) + " MiB)"
|
||||
}
|
||||
|
||||
w.Header().Set(`Content-Type`, `text/html;charset=UTF-8`)
|
||||
w.WriteHeader(200)
|
||||
w.Write([]byte(`<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<title>` + html.EscapeString(this.opts.AppTitle) + `</title>
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||||
<style type="text/css">
|
||||
html, body {
|
||||
font-family: sans-serif;
|
||||
}
|
||||
#img {
|
||||
width:64px;
|
||||
height:64px;
|
||||
}
|
||||
a {
|
||||
color:blue;
|
||||
text-decoration:underline;
|
||||
cursor:pointer;
|
||||
}
|
||||
form {
|
||||
text-align: center;
|
||||
border: 8px dashed lightgrey;
|
||||
padding: 12px;
|
||||
margin: 12px;
|
||||
}
|
||||
form .if-supports-drag {
|
||||
display:none;
|
||||
}
|
||||
form .if-nodrag {
|
||||
display:block;
|
||||
}
|
||||
form.supports-drag .if-supports-drag {
|
||||
display:block;
|
||||
}
|
||||
form.supports-drag .if-nodrag {
|
||||
display:none;
|
||||
}
|
||||
form.is-dragging {
|
||||
background: lightblue;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<form id="form" method="POST" action="/upload" enctype="multipart/form-data">
|
||||
<input type="hidden" name="MAX_FILE_SIZE" value="` + strconv.Itoa(int(this.opts.MaxUploadBytes)) + `" />
|
||||
|
||||
<svg id="img" viewBox="0 0 24 24">
|
||||
<path fill="#000000" d="M5,20H19V18H5M19,9H15V3H9V9H5L12,16L19,9Z" />
|
||||
</svg>
|
||||
<div class="if-supports-drag">
|
||||
<label id="form-label">Drop file(s) to upload ` + extraText + `</label>
|
||||
<br>
|
||||
<br>
|
||||
<a id="classic-uploader">Classic uploader</a>
|
||||
</div>
|
||||
|
||||
<div class="if-nodrag">
|
||||
<input id="form-upload" name="f" type="file" multiple>
|
||||
<input type="submit" value="Upload »">
|
||||
</div>
|
||||
</form>
|
||||
<script type="text/javascript">
|
||||
|
||||
"use strict";
|
||||
|
||||
var onLoad = function() {
|
||||
var form = document.getElementById("form");
|
||||
var input = document.getElementById("form-upload");
|
||||
var label = document.getElementById("form-label");
|
||||
|
||||
var makeCaptionText = function( files ) {
|
||||
return files.length > 1 ? ( input.getAttribute( 'data-multiple-caption' ) || '' ).replace( '{count}', files.length ) : files[ 0 ].name;
|
||||
};
|
||||
|
||||
var refreshCaption = function() {
|
||||
label.textContent = makeCaptionText( input.files );
|
||||
};
|
||||
input.addEventListener('change', refreshCaption);
|
||||
|
||||
//
|
||||
document.getElementById("classic-uploader").addEventListener("click", function() {
|
||||
form.classList.remove("supports-drag");
|
||||
});
|
||||
|
||||
var supportsDrag = ('ondrop' in window && 'FormData' in window && 'FileReader' in window);
|
||||
if (supportsDrag) {
|
||||
|
||||
// Handle CSS
|
||||
|
||||
form.classList.add('supports-drag');
|
||||
|
||||
var handleDragState = function(event_name, classEnabled) {
|
||||
form.addEventListener(event_name, function(e) {
|
||||
e.preventDefault();
|
||||
e.stopPropagation();
|
||||
if (classEnabled) {
|
||||
form.classList.add( 'is-dragover' );
|
||||
} else {
|
||||
form.classList.remove( 'is-dragover' );
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
handleDragState('dragover', true);
|
||||
handleDragState('dragenter', true);
|
||||
handleDragState('dragleave', false);
|
||||
handleDragState('dragend', false);
|
||||
handleDragState('drop', false);
|
||||
|
||||
// Handle uploads
|
||||
|
||||
form.addEventListener('drop', function(e) {
|
||||
input.files = e.dataTransfer.files; // the files that were dropped
|
||||
refreshCaption();
|
||||
form.submit();
|
||||
});
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
// Entry point
|
||||
onLoad();
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
`))
|
||||
}
|
||||
Loading…
Reference in New Issue