gosuki/browsers.go

package main

import (
	"fmt"
	"path"

	"github.com/fsnotify/fsnotify"
)

type BrowserType uint8

// Browser types
const (
	TChrome BrowserType = iota
	TFirefox
)

// Chrome details
var Chrome = struct {
	BookmarkFile string
	BookmarkDir  string
}{
	"Bookmarks",
	"/home/spike/.config/google-chrome/Default/",
}

type IWatchable interface {
	Watch() bool
	Watcher() *fsnotify.Watcher // returns browser watcher
	Parse()                     // Main parsing method
	GetPath() string            //returns bookmark path
	GetDir() string             // returns bookmarks dir
}

type IBrowser interface {
	SetupWatcher() // Starts watching bookmarks and runs Load on change
	Watch() bool
	InitBuffer() // init buffer db, should be defered to close after call
	Load()       // Loads bookmarks to db without watching
	//Parse(...ParseHook) // Main parsing method with different parsing hooks
	Close() // Gracefully finish work and stop watching
}

// Base browser class serves as reference for implmented browser types
// Browser should contain enough data internally to not rely on any global
// variable or constant if possible.
// To create new browsers, you must implement a New<BrowserType>() function

type BaseBrowser struct {
	watcher    *fsnotify.Watcher
	baseDir    string
	bkFile     string
	bufferDB   *DB
	stats      *ParserStats
	bType      BrowserType
	name       string
	isWatching bool
}

func (bw *BaseBrowser) Watcher() *fsnotify.Watcher {
	return bw.watcher
}

func (bw *BaseBrowser) Load() {
	log.Debug("BaseBrowser Load()")
}

func (bw *BaseBrowser) GetPath() string {
	return path.Join(bw.baseDir, bw.bkFile)
}

func (bw *BaseBrowser) GetDir() string {
	return bw.baseDir
}

func (bw *BaseBrowser) SetupWatcher() {
	var err error
	bw.watcher, err = fsnotify.NewWatcher()
	logPanic(err)
	err = bw.watcher.Add(bw.baseDir)
	logPanic(err)
}

func (bw *BaseBrowser) Close() {
	err := bw.watcher.Close()
	bw.bufferDB.Close()
	logPanic(err)
}

func (b *BaseBrowser) InitBuffer() {
	bufferName := fmt.Sprintf("buffer_%s", b.name)
	bufferPath := fmt.Sprintf(DBBufferFmt, bufferName)
	b.bufferDB = DB{}.New(bufferName, bufferPath)
	b.bufferDB.Init()
}