feat: add basic impl
This commit is contained in:
commit
630140521e
6 changed files with 111 additions and 0 deletions
4
.gitignore
vendored
Normal file
4
.gitignore
vendored
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
.env
|
||||
.vscode/
|
||||
files/
|
||||
main/
|
||||
13
Dockerfile
Normal file
13
Dockerfile
Normal file
|
|
@ -0,0 +1,13 @@
|
|||
FROM golang:1.17.11-alpine3.16
|
||||
|
||||
ENV APP_ROOT /opt/sfu
|
||||
|
||||
RUN mkdir -p "$APP_ROOT"
|
||||
WORKDIR "$APP_ROOT"
|
||||
|
||||
COPY go.mod .
|
||||
COPY main.go .
|
||||
|
||||
RUN go build \
|
||||
&& rm -r go.mod main.go
|
||||
ENTRYPOINT [ "/bin/sh", "-c", "$APP_ROOT/main"]
|
||||
10
docker-compose.yml
Normal file
10
docker-compose.yml
Normal file
|
|
@ -0,0 +1,10 @@
|
|||
---
|
||||
version: "3.9"
|
||||
services:
|
||||
app:
|
||||
build:
|
||||
context: .
|
||||
ports:
|
||||
- "8080:80"
|
||||
env_file:
|
||||
- .env
|
||||
3
go.mod
Normal file
3
go.mod
Normal file
|
|
@ -0,0 +1,3 @@
|
|||
module main
|
||||
|
||||
go 1.17
|
||||
BIN
main
Executable file
BIN
main
Executable file
Binary file not shown.
81
main.go
Normal file
81
main.go
Normal file
|
|
@ -0,0 +1,81 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
|
||||
var Info = log.New(os.Stdout, "\u001b[34mINFO: \u001B[0m", log.LstdFlags|log.Lshortfile)
|
||||
var Warning = log.New(os.Stdout, "\u001b[33mWARNING: \u001B[0m", log.LstdFlags|log.Lshortfile)
|
||||
var Error = log.New(os.Stdout, "\u001b[31mERROR: \u001b[0m", log.LstdFlags|log.Lshortfile)
|
||||
var Debug = log.New(os.Stdout, "\u001b[36mDEBUG: \u001B[0m", log.LstdFlags|log.Lshortfile)
|
||||
|
||||
func upload_file(w http.ResponseWriter, req *http.Request) {
|
||||
file, fileHeader, err := req.FormFile("file")
|
||||
if err != nil {
|
||||
Error.Println(err)
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
f, err := os.OpenFile(fmt.Sprintf("%v/%v", get_envvar_or_fatal("SFU_FILES_DIR"), fileHeader.Filename), os.O_WRONLY|os.O_CREATE, 0666)
|
||||
defer f.Close()
|
||||
io.Copy(f, file)
|
||||
}
|
||||
|
||||
func list_uploaded_files(w http.ResponseWriter, req *http.Request) {
|
||||
files_dir := get_envvar_or_fatal("SFU_FILES_DIR")
|
||||
Info.Println(fmt.Sprintf("server: will list uploaded files on %v", files_dir))
|
||||
files, err := ioutil.ReadDir(files_dir)
|
||||
if err != nil {
|
||||
Warning.Println(fmt.Sprintf("%v does not exist", files_dir))
|
||||
Info.Println(fmt.Sprintf("will create %v", files_dir))
|
||||
_ = os.Mkdir(files_dir, os.ModePerm)
|
||||
}
|
||||
for _, f := range files {
|
||||
fmt.Fprintf(w, "%v\n", f.Name())
|
||||
}
|
||||
}
|
||||
|
||||
func get_envvar_or_fatal(envvar_name string) string {
|
||||
envvar_value, isSet := os.LookupEnv(envvar_name)
|
||||
if !isSet {
|
||||
Error.Println(fmt.Sprintf("%v is not set", envvar_name))
|
||||
os.Exit(1)
|
||||
}
|
||||
return envvar_value
|
||||
}
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
|
||||
path := req.URL.Path
|
||||
Info.Println(fmt.Sprintf("received %v on %v", req.Method, path))
|
||||
if path != "/" {
|
||||
err_msg := fmt.Sprintf("path %v does not exist", path)
|
||||
http.Error(w, err_msg, http.StatusNotFound)
|
||||
Warning.Println(err_msg)
|
||||
Error.Println(fmt.Sprintf("will return %v", http.StatusNotFound))
|
||||
return
|
||||
}
|
||||
switch req.Method {
|
||||
case http.MethodGet:
|
||||
list_uploaded_files(w, req)
|
||||
case http.MethodPost:
|
||||
upload_file(w, req)
|
||||
default:
|
||||
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
|
||||
Warning.Println(fmt.Sprintf("%v not allowed", req.Method))
|
||||
Error.Println(fmt.Sprintf("will return %v", http.StatusMethodNotAllowed))
|
||||
}
|
||||
})
|
||||
port := fmt.Sprintf(":%v", get_envvar_or_fatal("SFU_PORT"))
|
||||
Info.Println(fmt.Sprintf("running SFU on port %v", port))
|
||||
err := http.ListenAndServe(port, nil)
|
||||
if err != nil {
|
||||
Error.Println(fmt.Sprintf("%v port may not be available", port))
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue