zevdg / 2018-codemash-go Goto Github PK

package main

import (
	"fmt"
	"math/rand"
	"time"
	"strings"
	"flag"
	"os"
	"log"
	"net/http"
	"github.com/zevdg/2018-codemash-go/labs/01/wordbank"
)

func ipsumHandler(wordbankFile string, words int, sentenceLength int) http.Handler {
	fn := func(w http.ResponseWriter, r *http.Request) {
		ipsum, err := generateIpsum(wordbankFile, words, sentenceLength)
		if err != nil {
			log.Fatal(err)
			panic(err)
		}
		fmt.Fprintf(w, ipsum)
	}
	return http.HandlerFunc(fn)
}

func main() {
	var words int
	flag.IntVar(&words, "words", 100, "an int")

	var sentenceLength int
	flag.IntVar(&sentenceLength, "sentence-length", 6, "an int")

	var wordbankFile string
	flag.StringVar(&wordbankFile, "wordbank", "nope", "File path")

	flag.Parse()

	rand.Seed(time.Now().UnixNano())

	ih := ipsumHandler(wordbankFile, words, sentenceLength)

	http.HandleFunc("/ipsum", ih)
	http.ListenAndServe(":12345", nil)
}

func generateIpsum(wordbankFile string, wordCount int, sentenceLength int) (string, error) {
	file, err := os.Open(wordbankFile)
	if err != nil {
		return "", fmt.Errorf("calculation failed: %s", err)
	}
	defer file.Close()

	wb, err := wordbank.NewWordBank(file)

	if err != nil {
		return "", fmt.Errorf("calculation failed: %s", err)
	}
	var s []string
	ipsum := make(chan string)
	go func() {
		for i := 0; i < wordCount/sentenceLength; i++ {

			ipsum <- generateSentence(wb, sentenceLength)
		}

		if wordCount%sentenceLength > 0 {
			ipsum <- generateSentence(wb, wordCount%sentenceLength)
		}
		close(ipsum)
	}()

	for elem := range ipsum {
		s = append(s, elem)
	}
	return strings.Join(s, ". ") + ".", nil
}

func generateSentence(wb *wordbank.WordBank, wordCount int) string {
	var s []string
	for i := 0; i < wordCount; i++ {
		s = append(s, wb.GetWord())
	}
	return strings.Join(s, " ")
}