mirror of
https://github.com/nbd-wtf/satdress.git
synced 2024-11-13 17:30:35 +00:00
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/sha256"
|
|
_ "embed"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/cockroachdb/pebble"
|
|
"github.com/gorilla/mux"
|
|
"github.com/kelseyhightower/envconfig"
|
|
_ "github.com/lib/pq"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type Settings struct {
|
|
Host string `envconfig:"HOST" default:"0.0.0.0"`
|
|
Port string `envconfig:"PORT" required:"true"`
|
|
Domain string `envconfig:"DOMAIN" required:"true"`
|
|
Secret string `envconfig:"SECRET" required:"true"`
|
|
}
|
|
|
|
var s Settings
|
|
var db *pebble.DB
|
|
var router = mux.NewRouter()
|
|
var log = zerolog.New(os.Stderr).Output(zerolog.ConsoleWriter{Out: os.Stderr})
|
|
|
|
//go:embed index.html
|
|
var html string
|
|
|
|
func main() {
|
|
err := envconfig.Process("", &s)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("couldn't process envconfig.")
|
|
}
|
|
|
|
db, err = pebble.Open(s.Domain, nil)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Str("path", s.Domain).Msg("failed to open db.")
|
|
}
|
|
|
|
router.Path("/.well-known/lnurlp/{username}").Methods("GET").
|
|
HandlerFunc(handleLNURL)
|
|
|
|
router.Path("/").HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("content-type", "text/html")
|
|
fmt.Fprintf(w, html)
|
|
},
|
|
)
|
|
|
|
router.Path("/grab").HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
name := []byte(r.FormValue("name"))
|
|
|
|
mac := hmac.New(sha256.New, []byte(s.Secret))
|
|
mac.Write(name)
|
|
pin := hex.EncodeToString(mac.Sum(nil))
|
|
|
|
if _, closer, err := db.Get(name); err == nil {
|
|
w.WriteHeader(401)
|
|
fmt.Fprint(w,
|
|
"name already exists! must provide pin (contact support).")
|
|
return
|
|
} else if err == nil {
|
|
closer.Close()
|
|
}
|
|
|
|
data, _ := json.Marshal(Params{
|
|
Kind: r.FormValue("kind"),
|
|
Host: r.FormValue("host"),
|
|
Key: r.FormValue("key"),
|
|
})
|
|
|
|
if err := db.Set(name, data, pebble.Sync); err != nil {
|
|
w.WriteHeader(500)
|
|
fmt.Fprint(w, "error! "+err.Error())
|
|
return
|
|
}
|
|
|
|
fmt.Fprintf(w,
|
|
"name saved! this is your secret pin key for this name: %s",
|
|
pin)
|
|
},
|
|
)
|
|
|
|
srv := &http.Server{
|
|
Handler: router,
|
|
Addr: s.Host + ":" + s.Port,
|
|
WriteTimeout: 15 * time.Second,
|
|
ReadTimeout: 15 * time.Second,
|
|
}
|
|
log.Debug().Str("addr", srv.Addr).Msg("listening")
|
|
srv.ListenAndServe()
|
|
}
|