Initial commit
This commit is contained in:
commit
b8998f6090
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
||||
ruff
|
||||
5
go.mod
Normal file
5
go.mod
Normal file
@ -0,0 +1,5 @@
|
||||
module git.tilde.town/diff/ruff
|
||||
|
||||
go 1.14
|
||||
|
||||
require github.com/mdp/qrterminal v1.0.1
|
||||
4
go.sum
Normal file
4
go.sum
Normal file
@ -0,0 +1,4 @@
|
||||
github.com/mdp/qrterminal v1.0.1 h1:07+fzVDlPuBlXS8tB0ktTAyf+Lp1j2+2zK3fBOL5b7c=
|
||||
github.com/mdp/qrterminal v1.0.1/go.mod h1:Z33WhxQe9B6CdW37HaVqcRKzP+kByF3q/qLxOGe12xQ=
|
||||
rsc.io/qr v0.2.0 h1:6vBLea5/NRMVTz8V66gipeLycZMl/+UlFmk8DvqQ6WY=
|
||||
rsc.io/qr v0.2.0/go.mod h1:IF+uZjkb9fqyeF/4tlBoynqmQxUoPfWEKh921coOuXs=
|
||||
198
main.go
Normal file
198
main.go
Normal file
@ -0,0 +1,198 @@
|
||||
// Ruff provides a pop-up web server to Retrieve/Upload Files Fast over
|
||||
// LAN, inspired by WOOF (Web Offer One File) by Simon Budig.
|
||||
//
|
||||
// It's based on the principle that not everyone has <insert neat file transfer
|
||||
// utility here>, just about every device that can network has an HTTP client,
|
||||
// making a hyper-simple HTTP server a viable option for file transfer with
|
||||
// zero notice or setup as long as *somebody* has a copy of RUFF.
|
||||
//
|
||||
// Why create RUFF when WOOF exists? WOOF is no longer in the debian repos and
|
||||
// it's easier to `go get` a tool than it is to hunt down Simon's website for
|
||||
// the latest copy.
|
||||
package main
|
||||
|
||||
import (
|
||||
"net"
|
||||
"net/url"
|
||||
"net/http"
|
||||
"time"
|
||||
"context"
|
||||
"html/template"
|
||||
|
||||
"path"
|
||||
"os"
|
||||
"io"
|
||||
|
||||
"fmt"
|
||||
"flag"
|
||||
"errors"
|
||||
"github.com/mdp/qrterminal"
|
||||
)
|
||||
|
||||
// Config stores all settings for an instance of RUFF.
|
||||
type Config struct {
|
||||
downloads int
|
||||
port int
|
||||
filePath string
|
||||
fileName string
|
||||
hideQR bool
|
||||
uploading bool
|
||||
}
|
||||
|
||||
func getConfig() Config {
|
||||
conf := Config{
|
||||
downloads: 1,
|
||||
port: 8008,
|
||||
hideQR: false,
|
||||
uploading: false,
|
||||
}
|
||||
|
||||
flag.IntVar(&conf.downloads, "count", conf.downloads, "number of downloads before exiting. set to -1 for unlimited downloads.")
|
||||
flag.IntVar(&conf.port, "port", conf.port, "port to serve file on.")
|
||||
flag.BoolVar(&conf.hideQR, "hide-qr", conf.hideQR, "hide the QR code.")
|
||||
flag.BoolVar(&conf.uploading, "upload", false, "upload files instead of downloading")
|
||||
|
||||
flag.IntVar(&conf.downloads, "c", conf.downloads, "number of downloads before exiting. set to -1 for unlimited downloads. (shorthand)")
|
||||
flag.IntVar(&conf.port, "p", conf.port, "port to serve file on. (shorthand)")
|
||||
flag.BoolVar(&conf.hideQR, "q", conf.hideQR, "hide the QR code. (shorthand)")
|
||||
flag.BoolVar(&conf.uploading, "u", false, "upload files instead of downloading (shorthand)")
|
||||
|
||||
flag.Parse()
|
||||
conf.filePath = flag.Arg(0)
|
||||
conf.fileName = path.Base(conf.filePath)
|
||||
|
||||
return conf
|
||||
}
|
||||
|
||||
func getIP() (string, error) {
|
||||
conn, err := net.Dial("udp", "8.8.8.8:80")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
localAddr, ok := conn.LocalAddr().(*net.UDPAddr)
|
||||
if !ok {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return localAddr.IP.String(), nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
conf := getConfig()
|
||||
|
||||
server := &http.Server{
|
||||
Addr: fmt.Sprintf(":%v", conf.port),
|
||||
ReadTimeout: 10*time.Second,
|
||||
WriteTimeout: 10*time.Second,
|
||||
}
|
||||
|
||||
if conf.uploading {
|
||||
setupUpload(server, &conf)
|
||||
} else {
|
||||
setupDownload(server, &conf)
|
||||
}
|
||||
|
||||
ip, err := getIP()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
|
||||
url := fmt.Sprintf("http://%s:%v/%s", ip, conf.port, conf.fileName)
|
||||
if !conf.hideQR {
|
||||
qrterminal.GenerateHalfBlock(url, qrterminal.M, os.Stdout)
|
||||
}
|
||||
fmt.Println(url)
|
||||
|
||||
err = server.ListenAndServe()
|
||||
if err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func setupDownload(server *http.Server, conf *Config) {
|
||||
downloads := conf.downloads
|
||||
http.Handle("/", http.RedirectHandler("/"+conf.fileName, http.StatusFound)) // 302 redirect
|
||||
http.HandleFunc("/"+conf.fileName, func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Disposition", "attachment; filename=\""+url.PathEscape(conf.fileName)+"\"")
|
||||
http.ServeFile(w, r, conf.filePath)
|
||||
|
||||
downloads--
|
||||
if downloads == 0 {
|
||||
server.Shutdown(context.Background())
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
var baseHeader = `<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<title>{{.}}</title>
|
||||
<style>
|
||||
body {
|
||||
margin: 18pt auto;
|
||||
font-size: 16pt;
|
||||
color: #212121;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>`
|
||||
|
||||
var baseFooter = `</body>
|
||||
</html>`
|
||||
|
||||
var uploadTemplate = `{{template "BaseHeader" "RUFF Upload Form"}}
|
||||
<form enctype="multipart/form-data" action="/" method="post">
|
||||
<label for="file">Select a file for upload:</label>
|
||||
<input type="file" name="file">
|
||||
<input type="submit" value="Upload"{{if .multiple}} multiple{{end}}>
|
||||
</form>
|
||||
{{template "BaseFooter"}}`
|
||||
|
||||
var errorTemplate = `{{template "BaseHeader" "UploadError"}}
|
||||
<p>{{.}}</p>
|
||||
<p><a href="/">Go back</a></p>
|
||||
{{template "BaseFooter"}}`
|
||||
|
||||
func setupUpload(server *http.Server, conf *Config) {
|
||||
tpl := template.Must(template.New("BaseHeader").Parse(baseHeader))
|
||||
template.Must(tpl.New("BaseFooter").Parse(baseFooter))
|
||||
template.Must(tpl.New("UploadForm").Parse(uploadTemplate))
|
||||
template.Must(tpl.New("UploadError").Parse(errorTemplate))
|
||||
|
||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
// Upload form
|
||||
if r.Method != http.MethodPost {
|
||||
err := tpl.ExecuteTemplate(w, "UploadForm", conf)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Handle upload
|
||||
r.ParseMultipartForm(20 << 20) // Buffer a maximum of 20MB in memory.
|
||||
|
||||
inFile, header, err := r.FormFile("file")
|
||||
if err != nil {
|
||||
tpl.ExecuteTemplate(w, "UploadError", err)
|
||||
return
|
||||
}
|
||||
defer inFile.Close()
|
||||
|
||||
outFile, err := os.Create(header.Filename)
|
||||
if err != nil {
|
||||
tpl.ExecuteTemplate(w, "UploadError", err)
|
||||
return
|
||||
}
|
||||
defer outFile.Close()
|
||||
|
||||
_, err = io.Copy(outFile, inFile)
|
||||
if err != nil {
|
||||
tpl.ExecuteTemplate(w, "UploadError", err)
|
||||
return
|
||||
}
|
||||
})
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user