344 lines
8.9 KiB
Go
344 lines
8.9 KiB
Go
// gemini
|
|
// a tool for gemini authors/creators/builders/admins
|
|
// (c) 2020 Brian Evans
|
|
// Available under the terms of the MIT license:
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
|
// The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
SPEC_URL string = "gemini://gemini.circumlunar.space:1965/docs/spec-spec.txt"
|
|
VERSION string = "0.1"
|
|
WARNING int = iota
|
|
ERROR
|
|
)
|
|
|
|
type issue struct {
|
|
message string
|
|
context string
|
|
lineNumber int
|
|
errType int
|
|
}
|
|
|
|
var inPreBlock bool = false
|
|
var lineNumber int = 0
|
|
var issues []issue
|
|
|
|
func min(a, b int) int {
|
|
if a < b {
|
|
return a
|
|
}
|
|
return b
|
|
}
|
|
|
|
func isWhiteSpace(c rune) bool {
|
|
switch c {
|
|
case ' ', '\t':
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func firstIndexNotChar(ln string, ch rune) int {
|
|
for i, c := range ln {
|
|
if c != ch {
|
|
return i
|
|
}
|
|
}
|
|
return -1
|
|
}
|
|
|
|
// Makes sure only 3 heading levels are used, adds a space after the heading declaration, before the text
|
|
func heading(ln string) string {
|
|
ln = strings.TrimSpace(ln)
|
|
if strings.HasPrefix(ln, "####") {
|
|
length := min(len(ln), 50)
|
|
issues = append(issues, issue{"Too many heading levels", ln[:length], lineNumber, ERROR})
|
|
}
|
|
endHeading := firstIndexNotChar(ln, '#')
|
|
return fmt.Sprintf("%s %s", ln[:endHeading], strings.TrimSpace(ln[endHeading:]))
|
|
}
|
|
|
|
func link(ln string) string {
|
|
ln = strings.TrimSpace(ln)
|
|
if len(ln) < 3 {
|
|
issues = append(issues, issue{"Link initiated but not followed with a url", ln, lineNumber, ERROR})
|
|
}
|
|
linkln := strings.TrimSpace(ln[2:])
|
|
linkln = strings.Replace(linkln, "\t", " ", -1)
|
|
split := strings.SplitN(linkln, " ", 2)
|
|
if len(split) == 1 {
|
|
return fmt.Sprintf("=> %s", strings.TrimSpace(split[0]))
|
|
}
|
|
return fmt.Sprintf("=> %s %s", strings.TrimSpace(split[0]), strings.TrimSpace(split[1]))
|
|
|
|
}
|
|
|
|
// Makes sure there is one space between the list declaration and the text
|
|
func list(ln string) string {
|
|
ln = strings.TrimSpace(ln)
|
|
if len(ln) < 2 {
|
|
return ln
|
|
}
|
|
return fmt.Sprintf("* %s", strings.TrimSpace(ln[1:]))
|
|
}
|
|
|
|
// preBlockToggle toggles the preBlock state as well as providing proper trimming of extra space
|
|
// if text is provided with an opening ``` a space is put in between the ``` and the text
|
|
func preBlockToggle(ln string) string {
|
|
ln = strings.TrimSpace(ln)
|
|
inPreBlock = !inPreBlock
|
|
if len(ln) > 3 {
|
|
alt := strings.TrimSpace(ln[3:])
|
|
|
|
if len(alt) == 0 {
|
|
return "```"
|
|
}
|
|
|
|
// If there is alt text and we are at a closing ```:
|
|
if !inPreBlock {
|
|
length := min(len(ln), 50)
|
|
issues = append(issues, issue{"Alt text cannot be included when closing a block, put it with the opening ```", ln[:length], lineNumber, ERROR})
|
|
return ln
|
|
}
|
|
// If there is alt text and this is opening a ```:
|
|
return fmt.Sprintf("``` %s", strings.TrimSpace(ln[3:]))
|
|
}
|
|
return ln
|
|
}
|
|
|
|
// preText just removes excess space on the right hand side
|
|
func preText(ln string) string {
|
|
return strings.TrimRight(ln, " \t")
|
|
}
|
|
|
|
// nonPreText normalizes spacing by eliminating instances of multiple spaces or tabs in a row.
|
|
func nonPreText(ln string) string {
|
|
ln = strings.TrimSpace(ln)
|
|
var output strings.Builder
|
|
ws := false
|
|
for _, c := range ln {
|
|
if !isWhiteSpace(c) {
|
|
ws = false
|
|
} else if ws {
|
|
continue
|
|
} else {
|
|
ws = true
|
|
}
|
|
output.WriteRune(c)
|
|
}
|
|
|
|
return output.String()
|
|
}
|
|
|
|
func lineRouter(ln string) string {
|
|
if len(ln) == 0 {
|
|
return ln
|
|
}
|
|
|
|
// Handle special strings
|
|
if ln[0] == '#' {
|
|
return heading(ln)
|
|
} else if ln[0] == '*' {
|
|
return list(ln)
|
|
} else if strings.HasPrefix(ln, "```") {
|
|
return preBlockToggle(ln)
|
|
} else if strings.HasPrefix(ln, "=>") {
|
|
return link(ln)
|
|
}
|
|
|
|
// Handle text
|
|
if inPreBlock {
|
|
return preText(ln)
|
|
}
|
|
return nonPreText(ln)
|
|
}
|
|
|
|
func fmtFile(path string) (string, error) {
|
|
b, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
lines := strings.SplitN(string(b), "\n", -1)
|
|
for i, ln := range lines {
|
|
lines[i] = lineRouter(ln)
|
|
lineNumber = i
|
|
}
|
|
return strings.Join(lines, "\n"), nil
|
|
}
|
|
|
|
func printIssues() {
|
|
for _, issue := range issues {
|
|
color := ""
|
|
kind := "Error"
|
|
if issue.errType == ERROR {
|
|
color = "\033[41;97;1m"
|
|
} else {
|
|
kind = "Warning"
|
|
color = "\033[30;103;1m"
|
|
}
|
|
fmt.Fprintf(os.Stderr, "%s%s:\033[0m %s\n%sLine %d:\033[0m %s\n\n", color, kind, issue.message, color, issue.lineNumber, issue.context)
|
|
}
|
|
}
|
|
|
|
func LoadCertificate(cert, key string) (tls.Certificate, error) {
|
|
certificate, err := tls.LoadX509KeyPair(cert, key)
|
|
if err != nil {
|
|
return tls.Certificate{}, err
|
|
}
|
|
return certificate, nil
|
|
}
|
|
|
|
func get(resource, cert, key string, headerOnly, currentSpec bool) (string, error) {
|
|
hasScheme := strings.Contains(resource, "://")
|
|
if !hasScheme {
|
|
resource = "gemini://" + resource
|
|
}
|
|
u, err := url.Parse(resource)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if u.Scheme != "gemini" {
|
|
return "", fmt.Errorf("URL scheme must be 'gemini'")
|
|
}
|
|
if u.Port() == "" {
|
|
u.Host = u.Host + ":1965"
|
|
}
|
|
conf := &tls.Config{
|
|
MinVersion: tls.VersionTLS12,
|
|
InsecureSkipVerify: true,
|
|
}
|
|
// TODO add client cert if available
|
|
|
|
conn, err := tls.Dial("tcp", u.Host, conf)
|
|
if err != nil {
|
|
return "", fmt.Errorf("TLS Dial Error: %s", err.Error())
|
|
}
|
|
defer conn.Close()
|
|
send := u.String() + "\r\n"
|
|
|
|
_, err = conn.Write([]byte(send))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
result, err := ioutil.ReadAll(conn)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
text := string(result)
|
|
endHeader := strings.Index(text, "\n")
|
|
|
|
if endHeader < 1 {
|
|
return "", fmt.Errorf("Invalid response from server")
|
|
}
|
|
|
|
if text[0] == '3' {
|
|
newUrl := strings.TrimSpace(text[3:endHeader])
|
|
return get(newUrl, cert, key, headerOnly, currentSpec)
|
|
}
|
|
|
|
head := strings.TrimSpace(text[:endHeader])
|
|
body := strings.TrimSpace(text[endHeader:])
|
|
if headerOnly {
|
|
return head, nil
|
|
}
|
|
|
|
if body == "" {
|
|
return "", fmt.Errorf("Error: Body was empty. Response header:\n%s\n", head)
|
|
}
|
|
|
|
return body, nil
|
|
}
|
|
|
|
func main() {
|
|
fmtCmd := flag.NewFlagSet("fmt", flag.ExitOnError)
|
|
fmtOut := fmtCmd.String("o", "", "Path to output file")
|
|
fmtSupress := fmtCmd.Bool("s", false, "Supress error messaging")
|
|
|
|
getCmd := flag.NewFlagSet("get", flag.ExitOnError)
|
|
getOut := getCmd.String("o", "", "Path to output file")
|
|
getMime := getCmd.Bool("header", false, "Only retrieve header")
|
|
getCert := getCmd.String("cert", "", "Path to certificate file")
|
|
getKey := getCmd.String("key", "", "Path to key file")
|
|
|
|
specCmd := flag.NewFlagSet("spec", flag.ExitOnError)
|
|
specOut := specCmd.String("o", "", "Path to output file")
|
|
|
|
if len(os.Args) < 2 {
|
|
fmt.Println("expected command: fmt, get, spec, or version")
|
|
os.Exit(1)
|
|
}
|
|
|
|
switch os.Args[1] {
|
|
case "fmt":
|
|
fmtCmd.Parse(os.Args[2:])
|
|
if len(fmtCmd.Args()) != 1 {
|
|
fmt.Printf("Incorrect number of positional arguments expected 1, got %d\n", len(fmtCmd.Args()))
|
|
os.Exit(1)
|
|
}
|
|
text, err := fmtFile(fmtCmd.Args()[0])
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
if *fmtOut == "" {
|
|
fmt.Print(text)
|
|
fmt.Fprint(os.Stderr, "\n") // So that the terminal input is always on a new line, but this LF doesnt get piped into a document
|
|
} else {
|
|
// TODO handle file writing
|
|
}
|
|
|
|
if !*fmtSupress {
|
|
printIssues()
|
|
}
|
|
case "get":
|
|
getCmd.Parse(os.Args[2:])
|
|
if len(getCmd.Args()) != 1 {
|
|
fmt.Printf("Incorrect number of positional arguments expected 1, got %d\n", len(fmtCmd.Args()))
|
|
os.Exit(1)
|
|
}
|
|
text, err := get(getCmd.Args()[0], *getCert, *getKey, *getMime, false)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
if *getOut == "" {
|
|
fmt.Print(text)
|
|
fmt.Print("\n")
|
|
} else {
|
|
// TODO handle file writing
|
|
}
|
|
case "spec":
|
|
specCmd.Parse(os.Args[2:])
|
|
if len(getCmd.Args()) != 0 {
|
|
fmt.Printf("Incorrect number of positional arguments expected 0, got %d\n", len(fmtCmd.Args()))
|
|
os.Exit(1)
|
|
}
|
|
text, err := get(SPEC_URL, "", "", false, true)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err.Error())
|
|
}
|
|
if *specOut == "" {
|
|
fmt.Print(text)
|
|
fmt.Print("\n")
|
|
} else {
|
|
// TODO handle file writing
|
|
}
|
|
case "version":
|
|
fmt.Printf("Gemini Tool v%s\n", VERSION)
|
|
}
|
|
}
|