andreimarcu
9 years ago
commit
2dbe318b18
10 changed files with 317 additions and 0 deletions
-
24.gitignore
-
24README.md
-
54display.go
-
17pages.go
-
47server.go
-
13templates/base.html
-
6templates/display/file.html
-
6templates/display/image.html
-
8templates/index.html
-
118upload.go
@ -0,0 +1,24 @@ |
|||||
|
# Compiled Object files, Static and Dynamic libs (Shared Objects) |
||||
|
*.o |
||||
|
*.a |
||||
|
*.so |
||||
|
|
||||
|
# Folders |
||||
|
_obj |
||||
|
_test |
||||
|
|
||||
|
# Architecture specific extensions/prefixes |
||||
|
*.[568vq] |
||||
|
[568vq].out |
||||
|
|
||||
|
*.cgo1.go |
||||
|
*.cgo2.c |
||||
|
_cgo_defun.c |
||||
|
_cgo_gotypes.go |
||||
|
_cgo_export.* |
||||
|
|
||||
|
_testmain.go |
||||
|
|
||||
|
*.exe |
||||
|
*.test |
||||
|
*.prof |
||||
@ -0,0 +1,24 @@ |
|||||
|
Linx Server |
||||
|
====== |
||||
|
Soon-to-be opensource replacement of Linx |
||||
|
|
||||
|
**Please do not use yet! Consider it in pre-alpha development stages.** |
||||
|
|
||||
|
Build & Run |
||||
|
---------------- |
||||
|
|
||||
|
1. ```go get -u github.com/andreimarcu/linx-server ``` |
||||
|
2. ```cd $GOPATH/src/github.com/andreimarcu/linx-server ``` |
||||
|
3. ```go build && ./linx-server``` |
||||
|
|
||||
|
By default linx will bind to ```http://127.0.0.1:8080/```, use the "files/" files directory and the "linx" sitename. |
||||
|
Configurable flags can be found in ```server.go```. |
||||
|
|
||||
|
|
||||
|
TODO |
||||
|
-------- |
||||
|
Please refer to the [main TODO issue](https://github.com/andreimarcu/linx-server/issues/1) |
||||
|
|
||||
|
Author |
||||
|
------- |
||||
|
Andrei Marcu, http://andreim.net/ |
||||
@ -0,0 +1,54 @@ |
|||||
|
package main |
||||
|
|
||||
|
import ( |
||||
|
"net/http" |
||||
|
"os" |
||||
|
"path" |
||||
|
"strings" |
||||
|
|
||||
|
"github.com/flosch/pongo2" |
||||
|
"github.com/rakyll/magicmime" |
||||
|
"github.com/zenazn/goji/web" |
||||
|
) |
||||
|
|
||||
|
func fileDisplayHandler(c web.C, w http.ResponseWriter, r *http.Request) { |
||||
|
filename := c.URLParams["name"] |
||||
|
absPath := path.Join(Config.filesDir, filename) |
||||
|
fileInfo, err := os.Stat(absPath) |
||||
|
|
||||
|
if os.IsNotExist(err) { |
||||
|
http.Error(w, http.StatusText(404), 404) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
if err := magicmime.Open(magicmime.MAGIC_MIME_TYPE | |
||||
|
magicmime.MAGIC_SYMLINK | |
||||
|
magicmime.MAGIC_ERROR); err != nil { |
||||
|
http.Error(w, err.Error(), http.StatusInternalServerError) |
||||
|
} |
||||
|
defer magicmime.Close() |
||||
|
|
||||
|
mimetype, err := magicmime.TypeByFile(absPath) |
||||
|
if err != nil { |
||||
|
http.Error(w, err.Error(), http.StatusInternalServerError) |
||||
|
} |
||||
|
|
||||
|
var tpl *pongo2.Template |
||||
|
|
||||
|
if strings.HasPrefix(mimetype, "image/") { |
||||
|
tpl = pongo2.Must(pongo2.FromCache("templates/display/image.html")) |
||||
|
} else { |
||||
|
tpl = pongo2.Must(pongo2.FromCache("templates/display/file.html")) |
||||
|
} |
||||
|
|
||||
|
err = tpl.ExecuteWriter(pongo2.Context{ |
||||
|
"mime": mimetype, |
||||
|
"sitename": Config.siteName, |
||||
|
"filename": filename, |
||||
|
"size": fileInfo.Size(), |
||||
|
}, w) |
||||
|
|
||||
|
if err != nil { |
||||
|
http.Error(w, err.Error(), http.StatusInternalServerError) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,17 @@ |
|||||
|
package main |
||||
|
|
||||
|
import ( |
||||
|
"github.com/flosch/pongo2" |
||||
|
"net/http" |
||||
|
|
||||
|
"github.com/zenazn/goji/web" |
||||
|
) |
||||
|
|
||||
|
func indexHandler(c web.C, w http.ResponseWriter, r *http.Request) { |
||||
|
indexTpl := pongo2.Must(pongo2.FromCache("templates/index.html")) |
||||
|
|
||||
|
err := indexTpl.ExecuteWriter(pongo2.Context{"sitename": Config.siteName}, w) |
||||
|
if err != nil { |
||||
|
http.Error(w, err.Error(), http.StatusInternalServerError) |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,47 @@ |
|||||
|
package main |
||||
|
|
||||
|
import ( |
||||
|
"flag" |
||||
|
"fmt" |
||||
|
"log" |
||||
|
"net" |
||||
|
"net/http" |
||||
|
"regexp" |
||||
|
|
||||
|
"github.com/zenazn/goji" |
||||
|
// "github.com/zenazn/goji/web"
|
||||
|
) |
||||
|
|
||||
|
var Config struct { |
||||
|
bind string |
||||
|
filesDir string |
||||
|
siteName string |
||||
|
} |
||||
|
|
||||
|
func main() { |
||||
|
flag.StringVar(&Config.bind, "b", "127.0.0.1:8080", |
||||
|
"host to bind to (default: 127.0.0.1:8080)") |
||||
|
flag.StringVar(&Config.filesDir, "d", "files/", |
||||
|
"path to files directory (default: files/)") |
||||
|
flag.StringVar(&Config.siteName, "n", "linx", |
||||
|
"name of the site") |
||||
|
flag.Parse() |
||||
|
|
||||
|
fmt.Printf("About to listen on http://%s\n", Config.bind) |
||||
|
|
||||
|
nameRe := regexp.MustCompile(`^/(?P<name>[a-z0-9-\.]+)$`) |
||||
|
selifRe := regexp.MustCompile(`^/selif/(?P<name>[a-z0-9-\.]+)$`) |
||||
|
|
||||
|
goji.Get("/", indexHandler) |
||||
|
goji.Post("/upload", uploadPostHandler) |
||||
|
goji.Put("/upload", uploadPutHandler) |
||||
|
goji.Get(nameRe, fileDisplayHandler) |
||||
|
goji.Handle(selifRe, http.StripPrefix("/selif/", http.FileServer(http.Dir(Config.filesDir)))) |
||||
|
|
||||
|
listener, err := net.Listen("tcp", Config.bind) |
||||
|
if err != nil { |
||||
|
log.Fatal("Could not bind: ", err) |
||||
|
} |
||||
|
|
||||
|
goji.ServeListener(listener) |
||||
|
} |
||||
@ -0,0 +1,13 @@ |
|||||
|
<html> |
||||
|
<head> |
||||
|
<title>{% block title %}{{ sitename }}{% endblock %}</title> |
||||
|
</head> |
||||
|
<body> |
||||
|
|
||||
|
<div> |
||||
|
{% block content %}{% endblock %} |
||||
|
</div> |
||||
|
|
||||
|
|
||||
|
</body> |
||||
|
</html> |
||||
@ -0,0 +1,6 @@ |
|||||
|
{% extends "../base.html" %} |
||||
|
|
||||
|
{% block content %} |
||||
|
<p>Viewing file {{ filename }} of mime: {{ mime }}</p> |
||||
|
<p><a href="/selif/{{ filename }}">Click to download</a></p> |
||||
|
{% endblock %} |
||||
@ -0,0 +1,6 @@ |
|||||
|
{% extends "../base.html" %} |
||||
|
|
||||
|
{% block content %} |
||||
|
<p>Viewing file {{ filename }}</p> |
||||
|
<img src="/selif/{{ filename }}" /> |
||||
|
{% endblock %} |
||||
@ -0,0 +1,8 @@ |
|||||
|
{% extends "base.html" %} |
||||
|
|
||||
|
{% block content %} |
||||
|
<form enctype="multipart/form-data" action="/upload" method="post"> |
||||
|
<input type="file" name="file" /> |
||||
|
<input type="submit" value="upload" /> |
||||
|
</form> |
||||
|
{% endblock %} |
||||
@ -0,0 +1,118 @@ |
|||||
|
package main |
||||
|
|
||||
|
import ( |
||||
|
"errors" |
||||
|
"fmt" |
||||
|
"io" |
||||
|
"net/http" |
||||
|
"os" |
||||
|
"path" |
||||
|
"regexp" |
||||
|
"strings" |
||||
|
|
||||
|
"code.google.com/p/go-uuid/uuid" |
||||
|
"github.com/zenazn/goji/web" |
||||
|
) |
||||
|
|
||||
|
type UploadRequest struct { |
||||
|
src io.Reader |
||||
|
filename string |
||||
|
expiry int |
||||
|
randomBarename bool |
||||
|
} |
||||
|
|
||||
|
type Upload struct { |
||||
|
Filename string |
||||
|
Size int64 |
||||
|
Expiry int |
||||
|
} |
||||
|
|
||||
|
func uploadPostHandler(c web.C, w http.ResponseWriter, r *http.Request) { |
||||
|
upReq := UploadRequest{} |
||||
|
|
||||
|
file, headers, err := r.FormFile("file") |
||||
|
if err != nil { |
||||
|
fmt.Fprintf(w, err.Error()) |
||||
|
return |
||||
|
} |
||||
|
defer file.Close() |
||||
|
|
||||
|
upReq.src = file |
||||
|
upReq.filename = headers.Filename |
||||
|
|
||||
|
upload, err := processUpload(upReq) |
||||
|
if err != nil { |
||||
|
fmt.Fprintf(w, "Failed to upload: %v", err) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
fmt.Fprintf(w, "File %s uploaded successfully.", upload.Filename) |
||||
|
} |
||||
|
|
||||
|
func uploadPutHandler(c web.C, w http.ResponseWriter, r *http.Request) { |
||||
|
upReq := UploadRequest{} |
||||
|
|
||||
|
defer r.Body.Close() |
||||
|
upReq.src = r.Body |
||||
|
|
||||
|
upload, err := processUpload(upReq) |
||||
|
if err != nil { |
||||
|
fmt.Fprintf(w, "Failed to upload") |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
fmt.Fprintf(w, "File %s uploaded successfully.", upload.Filename) |
||||
|
} |
||||
|
|
||||
|
func processUpload(upReq UploadRequest) (upload Upload, err error) { |
||||
|
barename, extension := barePlusExt(upReq.filename) |
||||
|
|
||||
|
if upReq.randomBarename || len(barename) == 0 { |
||||
|
barename = generateBarename() |
||||
|
} |
||||
|
|
||||
|
if len(extension) == 0 { |
||||
|
extension = "ext" |
||||
|
} |
||||
|
|
||||
|
upload.Filename = strings.Join([]string{barename, extension}, ".") |
||||
|
|
||||
|
dst, ferr := os.Create(path.Join("files/", upload.Filename)) |
||||
|
defer dst.Close() |
||||
|
if ferr != nil { |
||||
|
err = ferr |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
bytes, cerr := io.Copy(dst, upReq.src) |
||||
|
if cerr != nil { |
||||
|
err = cerr |
||||
|
return |
||||
|
} else if bytes == 0 { |
||||
|
err = errors.New("Empty file") |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
upload.Size = bytes |
||||
|
|
||||
|
return |
||||
|
} |
||||
|
|
||||
|
func generateBarename() string { |
||||
|
return uuid.New()[:8] |
||||
|
} |
||||
|
|
||||
|
func barePlusExt(filename string) (barename, extension string) { |
||||
|
re := regexp.MustCompile(`[^A-Za-z0-9\-]`) |
||||
|
|
||||
|
filename = strings.TrimSpace(filename) |
||||
|
filename = strings.ToLower(filename) |
||||
|
|
||||
|
extension = path.Ext(filename) |
||||
|
barename = filename[:len(filename)-len(extension)] |
||||
|
|
||||
|
extension = re.ReplaceAllString(extension, "") |
||||
|
barename = re.ReplaceAllString(barename, "") |
||||
|
|
||||
|
return |
||||
|
} |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue