Handlers WIP

hugolib/handler_file.go

@@ -13,8 +13,11 @@
 
 package hugolib
 
+import "github.com/spf13/hugo/source"
+
 var Filer interface {
-	Read()
+	Read(*source.File)
 	Render()
 	Convert()
+	Extensions() []string
 }

hugolib/handler_page.go

@@ -13,8 +13,31 @@
 
 package hugolib
 
+import "github.com/spf13/hugo/source"
+
 var Pager interface {
-	Read()
+	Read(*source.File)
 	Render()
 	Convert()
+	Extensions() []string
+}
+
+var markdown = Handle{
+	extensions: []string{"mdown", "markdown", "md"},
+	readrun: func(f *source.File, results HandleResults) {
+		page, err := NewPage(f.Path())
+		if err != nil {
+			results <- HandledResult{file: f, err: err}
+		}
+
+		if err := page.ReadFrom(f.Contents); err != nil {
+			results <- HandledResult{file: f, err: err}
+		}
+
+		results <- HandledResult{file: f, page: page, err: err}
+	},
+}
+
+func init() {
+	RegisterHandler(markdown)
 }

hugolib/handlers.go

@@ -13,15 +13,40 @@
 
 package hugolib
 
-var Handler interface {
+import "github.com/spf13/hugo/source"
-	Read()
+
-	Render()
+type Handler interface {
-	Convert()
+	Read(*source.File, HandleResults)
-	Extensions()
+	//Render()
+	//Convert()
+	Extensions() []string
+}
+
+type HandledResult struct {
+	page *Page
+	file *source.File
+	err  error
+}
+
+type HandleResults chan<- HandledResult
+
+type ReadFunc func(*source.File, HandleResults)
+
+type Handle struct {
+	extensions []string
+	readrun    ReadFunc
 }
 
 var handlers []Handler
 
+func (h Handle) Extensions() []string {
+	return h.extensions
+}
+
+func (h Handle) Read(s *source.File, results HandleResults) {
+	h.readrun(s, results)
+}
+
 func RegisterHandler(h Handler) {
 	handlers = append(handlers, h)
 }
@@ -30,16 +55,16 @@ func Handlers() []Handler {
 	return handlers
 }
 
-func Handler(ext string) Handler {
+func FindHandler(ext string) Handler {
 	for _, h := range Handlers() {
-		if h.Match(ext) {
+		if HandlerMatch(h, ext) {
 			return h
 		}
 	}
 	return nil
 }
 
-func (h Handler) Match(ext string) bool {
+func HandlerMatch(h Handler, ext string) bool {
 	for _, x := range h.Extensions() {
 		if ext == x {
 			return true

hugolib/page.go

@@ -54,20 +54,12 @@ type Page struct {
 	frontmatter []byte
 	rawContent  []byte
 	plain       string // TODO should be []byte
-	//sourceFrontmatter []byte
-	//sourceContent     []byte
 	PageMeta
-	//SourceFile source.File
 	Source
 	Position
 	Node
-	//Destination source.File
 }
 
-//type File struct {
-//Name, FileName, Extension, Dir, UniqueId string
-//}
-
 type Source struct {
 	Frontmatter []byte
 	Content     []byte

hugolib/site.go

@@ -331,7 +331,7 @@ func (s *Site) CreatePages() error {
 
 	files := s.Source.Files()
 
-	results := make(chan pageResult)
+	results := make(chan HandledResult)
 	filechan := make(chan *source.File)
 
 	procs := getGoMaxProcs()
@@ -361,7 +361,7 @@ func (s *Site) CreatePages() error {
 
 	readErrs := <-errs
 
-	results = make(chan pageResult)
+	results = make(chan HandledResult)
 	pagechan := make(chan *Page)
 
 	wg = &sync.WaitGroup{}
@@ -397,33 +397,23 @@ func (s *Site) CreatePages() error {
 	return fmt.Errorf("%s\n%s", readErrs, renderErrs)
 }
 
-func sourceReader(s *Site, files <-chan *source.File, results chan<- pageResult, wg *sync.WaitGroup) {
+func sourceReader(s *Site, files <-chan *source.File, results chan<- HandledResult, wg *sync.WaitGroup) {
 	defer wg.Done()
 	for file := range files {
-		// TODO: Switch here on extension
+		h := FindHandler(file.Extension())
-		h := handlers.Handler(file.Extension())
 		if h != nil {
-
+			h.Read(file, results)
 		} else {
 			jww.ERROR.Println("Unsupported File Type", file.Path())
 		}
 
-		page, err := NewPage(file.Path())
+		// TODO: Figure out Site stuff
-		if err != nil {
+		//page.Site = &s.Info
-			results <- pageResult{nil, err}
+		//page.Tmpl = s.Tmpl
-			continue
-		}
-		page.Site = &s.Info
-		page.Tmpl = s.Tmpl
-		if err := page.ReadFrom(file.Contents); err != nil {
-			results <- pageResult{nil, err}
-			continue
-		}
-		results <- pageResult{page, nil}
 	}
 }
 
-func pageConverter(s *Site, pages <-chan *Page, results chan<- pageResult, wg *sync.WaitGroup) {
+func pageConverter(s *Site, pages <-chan *Page, results HandleResults, wg *sync.WaitGroup) {
 	defer wg.Done()
 	for page := range pages {
 		//Handling short codes prior to Conversion to HTML
@@ -431,14 +421,14 @@ func pageConverter(s *Site, pages <-chan *Page, results chan<- pageResult, wg *s
 
 		err := page.Convert()
 		if err != nil {
-			results <- pageResult{nil, err}
+			results <- HandledResult{err: err}
 			continue
 		}
-		results <- pageResult{page, nil}
+		results <- HandledResult{err: err}
 	}
 }
 
-func converterCollator(s *Site, results <-chan pageResult, errs chan<- error) {
+func converterCollator(s *Site, results <-chan HandledResult, errs chan<- error) {
 	errMsgs := []string{}
 	for r := range results {
 		if r.err != nil {
@@ -453,7 +443,7 @@ func converterCollator(s *Site, results <-chan pageResult, errs chan<- error) {
 	errs <- fmt.Errorf("Errors rendering pages: %s", strings.Join(errMsgs, "\n"))
 }
 
-func readCollator(s *Site, results <-chan pageResult, errs chan<- error) {
+func readCollator(s *Site, results <-chan HandledResult, errs chan<- error) {
 	errMsgs := []string{}
 	for r := range results {
 		if r.err != nil {