spongy/spongy.cgi/network.go

package main

import (
	"bufio"
	"fmt"
	"path"
	"strconv"
	"os"
)

type Network struct {
	name string
	currentLog string
	lineno int64

	basePath string
}

type Update struct {
	Lines []string
	LastEventId string
}

func NewNetwork(basePath string) (*Network) {
	return &Network{
		basePath: basePath,
	}
}

func (nw *Network) LastEventId() string {
	return fmt.Sprintf("%s:%s:%d", nw.name, nw.currentLog, nw.lineno)
}

func (nw *Network) SetPosition(filename string, lineno int64) {
	nw.currentLog = filename
	nw.lineno = lineno
}

func (nw *Network) Tail(out chan<- *Update) error {
	if nw.currentLog == "" {
		currentfn := path.Join(nw.basePath, "log", "current")
		nw.currentfn, err := os.Readlink(currentfn)
		if err != nil {
			return err
		}
	}
	
	filepath := path.Join(nw.basePath, "log", nw.currentLog)
	f, err := os.Open(filepath)
	if err != nil {
		return err
	}
	defer f.Close()
	
	watcher, err := fsnotify.NewWatcher()
	if err != nil {
		return err
	}
	defer watcher.Close()
	
	watcher.add(filepath)
	
	for {
		lines := make([]string, 0)
		bf := bufio.NewScanner(f)
		for bf.Scan() {
			t := bf.Text()
			nw.lineno += 1
			
			parts := strings.Split(t, " ")
			if (len(parts) >= 4) && (parts[2] == "NEXTLOG") {
				watcher.Remove(filepath)
				filename = parts[3]
				filepath = path.Join(NetworkDir, filename)
				f.Close()
				f, err = os.Open(filepath)
				if err != nil {
					return err
				}
				watcher.Add(filepath)
				nw.lineno = 0
			}
			lines = append(lines, t))
		}
		if len(update.Lines) > 0 {
			update := Update{
				Lines: lines,
				LastEventId: nw.LastEventId(),
			}
			out <- &update
		}
		
		select {
		case _ = <-watcher.Events:
			// Somethin' happened!
		case err := <-watcher.Errors:
			return err
		}
	}
	
	return nil
}

func (nw *Network) Write(line string) {
	fn := path.Join(nw.baseDir