golangtest/main.go
2019-11-21 10:08:17 +01:00

122 lines
3.0 KiB
Go

package main
import (
"encoding/json"
"fmt"
"image/png"
"io/ioutil"
"log"
"net/http"
"os"
"path"
"strings"
"github.com/boombuler/barcode"
"github.com/boombuler/barcode/qr"
"github.com/disintegration/imaging"
"github.com/gorilla/mux"
)
var StartUpPath string
// our main function
func main() {
StartUpPath = "C:/Fotos" //dir
router := mux.NewRouter()
router.HandleFunc("/folder/{foldername:.*}", GetFolderContent).Methods("GET")
router.HandleFunc("/qr/{qrlink:.*}", GetQRCode).Methods("GET")
router.HandleFunc("/thumb/{imgname:.*}", GetThumbNail).Methods("GET")
router.PathPrefix("/img/").Handler(http.StripPrefix("/img/", http.FileServer(http.Dir("C:/Fotos")))).Methods("GET")
router.PathPrefix("/").Handler(http.FileServer(http.Dir("web"))).Methods("GET")
log.Fatal(http.ListenAndServe(":8000", router))
}
func GetFolderContent(w http.ResponseWriter, r *http.Request) {
params := mux.Vars(r)
pFolder := params["foldername"]
fmt.Println(path.Join(StartUpPath, pFolder))
files, err := ioutil.ReadDir(path.Join(StartUpPath, pFolder))
if err != nil {
log.Fatal(err)
}
var dateien []string
for _, f := range files {
if !strings.HasSuffix(f.Name(), ".thumb.jpg") {
dateien = append(dateien, f.Name())
}
}
json.NewEncoder(w).Encode(dateien)
}
func GetQRCode(w http.ResponseWriter, r *http.Request) {
// Variablen auswerten
params := mux.Vars(r)
pQRLink := params["qrlink"]
// QR-Code erstellen
qrCode, _ := qr.Encode(pQRLink, qr.M, qr.Auto)
// Größe festlegen
qrCode, _ = barcode.Scale(qrCode, 200, 200)
// Datei erstellen
file, _ := os.Create("qrcode.png")
defer file.Close()
// Datei ausgeben
png.Encode(w, qrCode)
}
func GetThumbNail(w http.ResponseWriter, r *http.Request) {
// Variablen auswerten
params := mux.Vars(r)
pIMGName := params["imgname"]
// Das Thumbnail soll nur generiert werden wenn es noch nicht existiert
if !fileExists(StartUpPath + "/" + pIMGName + ".thumb.jpg") {
// Originaldatei öffnen, welche angefragt wird
src, err := imaging.Open(StartUpPath + "/" + pIMGName)
if err != nil {
log.Fatalf("failed to open image: %v", err)
}
// Foto auf 200px skalieren
src = imaging.Resize(src, 200, 0, imaging.NearestNeighbor)
err = imaging.Save(src, StartUpPath+"/"+pIMGName+".thumb.jpg")
if err != nil {
log.Fatalf("failed to save image: %v", err)
}
// Datei ausgeben
png.Encode(w, src)
} else {
src, err := imaging.Open(StartUpPath + "/" + pIMGName + ".thumb.jpg")
if err != nil {
log.Fatalf("failed to open image: %v", err)
}
// Datei ausgeben
png.Encode(w, src)
}
}
//--------------------------------------------------------------------------
// Hilfsfunktionen
//--------------------------------------------------------------------------
func fileExists(filename string) bool {
info, err := os.Stat(filename)
if os.IsNotExist(err) {
return false
}
return !info.IsDir()
}