Gemini
/
mastoGem
1
0
Fork 0
Code Issues 6 Pull Requests Projects Releases 7 Wiki Activity

Can be configured

This commit is contained in:
Romain de Laage 2021-02-28 15:19:15 +01:00
parent 7c48322d3a
commit 3c80ec8864
Signed by: rdelaage
GPG Key ID: 534845FADDF0C329
2 changed files with 49 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
config.json Normal file
View File

@ -0,0 +1,6 @@
{
"listen": "0.0.0.0:1965",
"cert_path": "cert.pem",
"key_path": "key.rsa",
"base_url": "https://mamot.fr"
}

55
server.go
View File

@ -5,8 +5,8 @@ import (
"log"
"net"
"net/http"
"os"
"fmt"
"mime"
"encoding/json"
"io/ioutil"
"strings"
@ -20,17 +20,48 @@ type Blog struct {
Date string `json:"created_at"`
}
type Config struct {
Listen string `json:"listen"`
CertPath string `json:"cert_path"`
KeyPath string `json:"key_path"`
BaseURL string `json:"base_url"`
}
func main() {
err := mime.AddExtensionType(".gmi", "text/gemini")
if err != nil {
log.Fatalln("mime: %s", err)
config := getConfig()
listener := listen(config.Listen, config.CertPath, config.KeyPath)
log.Println("Server successfully started")
log.Println("Server is listening at " + config.Listen)
serve(listener, config.BaseURL)
}
func getConfig() Config {
configPath := os.Getenv("MASTOGEM_CONFIG_PATH")
if configPath == "" {
log.Println("MASTOGEM_CONFIG_PATH was not set, using default settings")
config := Config{
Listen: "127.0.0.1:1965",
CertPath: "cert.pem",
KeyPath: "key.rsa",
BaseURL: "https://mamot.fr",
}
return config
}
listener := listen("0.0.0.0:1965", "cert.pem", "key.rsa")
log.Println("Server successfully started")
log.Println("Server is listening at 0.0.0.0:1956")
configFile, err := ioutil.ReadFile(configPath)
if err != nil {
log.Fatalln("config file: %s", err)
}
serve(listener)
var config Config
json.Unmarshal(configFile, &config)
return config
}
func listen(address, certFile, keyFile string) net.Listener {
@ -53,21 +84,21 @@ func listen(address, certFile, keyFile string) net.Listener {
return listener
}
func serve(listener net.Listener) {
func serve(listener net.Listener, baseURL string) {
for {
conn, err := listener.Accept()
if err != nil {
log.Println(err)
}
go handleConn(conn.(*tls.Conn))
go handleConn(conn.(*tls.Conn), baseURL)
}
}
func handleConn(conn *tls.Conn) {
func handleConn(conn *tls.Conn, baseURL string) {
defer conn.Close()
blogs := getBlog("https://mamot.fr", "138624")
blogs := getBlog(baseURL, "138624")
_, err := fmt.Fprintf(conn, "20 text/gemini\r\n# Picasoft account toots\n")
if err != nil {