main.go

package main

import (
	"database/sql"
	"flag"
	"github.com/aagat/attic/config"
	"github.com/aagat/attic/fetcher"
	"github.com/aagat/attic/helpers"
	"github.com/aagat/attic/models"
	"github.com/aagat/attic/search"
	"github.com/aagat/attic/web"
	"github.com/gorilla/mux"
	_ "github.com/mattn/go-sqlite3"
	"log"
	"net/http"
)

var addr = flag.String("listen", ":8000", "HTTP ip/port to listen to")
var dbPath = flag.String("db", ":memory:", "Database file to use")
var indexPath = flag.String("index", "attic.index", "Index location to use")
var importFromFile = flag.String("import", "", "Bookmark file to import from")
var workers = flag.Int("workers", 5, "Web workers to fetch your pages.")

func main() {

	flag.Parse()

	var app config.Config

	r := mux.NewRouter()

	db, err := sql.Open("sqlite3", *dbPath)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	app.DB = db

	app.Models, _ = models.Init(&app)

	app.Search, err = search.Init(*indexPath)
	searchHandler := app.Search.(*search.Search).SearchHandler()

	if err != nil {
		log.Fatal(err)
	}

	jobs := make(chan string, 128)
	results := make(chan *models.BookmarkMeta, 128)
	errors := make(chan string, 128)

	app.Fetcher = fetcher.Init(&app, jobs, results, errors)
	app.Fetcher.(*fetcher.Fetcher).Boot(*workers)

	go func() {
		for {
			select {
			case meta := <-results:
				log.Println("Downloaded:", meta.Url)
				err = meta.Insert()

				if err != nil {
					log.Println(err)
				}
			case err := <-errors:
				log.Println("Error fetching bookmark:", err)
			}
		}
	}()

	app.Web = web.Init(&app)
	app.Helpers = helpers.Init(&app)

	utils := app.Helpers.(*helpers.Helpers)
	handler := app.Web.(*web.Handler)

	err = utils.CreateTables()
	if err != nil {
		log.Fatal(err)
	}

	if *importFromFile != "" {
		utils.ImportBookmarks(importFromFile)
	}

	r.HandleFunc("/", handler.Index)
	r.HandleFunc("/show/{id:[0-9]+}", handler.BookmarkById).Methods("GET")
	r.HandleFunc("/show/{hash}", handler.BookmarkByHash).Methods("GET")
	r.HandleFunc("/add", handler.NewBookmark).Methods("POST")
	r.HandleFunc("/update/{id:[0-9]+}", handler.UpdateBookmarkById).Methods("POST")
	r.HandleFunc("/update/{hash}", handler.UpdateBookmarkByHash).Methods("POST")
	r.Handle("/bookmarks/search", searchHandler)
	r.HandleFunc("/delete/{id:[0-9]+}", handler.DeleteBookmarkById).Methods("POST")
	r.HandleFunc("/delete/{hash}", handler.DeleteBookmarkByHash).Methods("POST")

	log.Printf("Listening and serving on port %v\n", *addr)
	http.ListenAndServe(*addr, r)
}