hashcash/cmd/hashcash/hashcash.go

package main

import (
	"encoding/json"
	"fmt"
	"os"
	"strconv"
	"time"

	"git.rootprojects.org/root/hashcash"
)

func help() {
	fmt.Println("Usage:")
	fmt.Println("\thashcash new [subject *] [expires in 5m] [difficulty 10]")
	fmt.Println("\thashcash parse <hashcash>")
	fmt.Println("\thashcash solve <hashcash>")
	fmt.Println("\thashcash verify <hashcash> [subject *]")
}

func main() {
	args := os.Args[:]

	if len(args) < 2 {
		help()
		os.Exit(1)
		return
	}

	switch args[1] {
	case "new":
		var subject string
		if len(args) > 3 {
			subject = args[2]
		}

		var difficulty int
		if len(args) > 4 {
			var err error
			difficulty, err = strconv.Atoi(args[3])
			if nil != err {
				help()
				os.Exit(1)
			}
		}

		var expIn time.Duration
		if len(args) > 5 {
			var err error
			expIn, err = time.ParseDuration(args[4])
			if nil != err {
				help()
				os.Exit(1)
			}
		} else {
			expIn = 5 * time.Minute
		}

		h := hashcash.New(hashcash.Hashcash{
			Subject:    subject,
			Difficulty: difficulty,
			ExpiresAt:  time.Now().Add(expIn),
		})

		fmt.Println(h.String())
		return
	case "parse":
		var token string
		if 3 != len(args) {
			help()
			os.Exit(1)
			return
		}
		token = args[2]
		h, err := hashcash.Parse(token)
		if nil != err {
			fmt.Fprintf(os.Stderr, "%s\n", err)
			return
		}
		b, _ := json.MarshalIndent(h, "", "  ")
		fmt.Println(string(b))
		return
	case "solve":
		var token string
		if 3 != len(args) {
			help()
			os.Exit(1)
			return
		}
		token = args[2]
		h, err := hashcash.Parse(token)
		if nil != err {
			fmt.Fprintf(os.Stderr, "%s\n", err)
			os.Exit(1)
			return
		}
		err = h.Solve(22)
		if nil != err {
			fmt.Fprintf(os.Stderr, "%s\n", err)
			os.Exit(1)
			return
		}

		fmt.Println(h.String())
		return
	case "verify":
		var token string
		if len(args) < 3 {
			help()
			os.Exit(1)
			return
		}
		token = args[2]
		h, err := hashcash.Parse(token)
		if nil != err {
			fmt.Fprintf(os.Stderr, "%s\n", err)
			return
		}

		subject := "*"
		if len(args) > 3 {
			subject = args[3]
		}

		err = h.Verify(subject)
		if nil != err {
			fmt.Fprintf(os.Stderr, "%s\n", err)
			return
		}

		var duration string
		dur := h.ExpiresAt.Sub(time.Now())
		if dur > 365*24*time.Hour {
			duration = "...a long, long time"
		} else {
			duration = dur.Truncate(time.Second).String()
		}
		fmt.Println("valid for", duration)
		return
	default:
		help()
		os.Exit(1)
	}
}