history.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"os"

	"github.com/Tomoka64/go-pkg-seeker/model"
)

//DBPath is a path to data file. all the data will be saved in json format
const DBPath = "config/data.json"

//History is slice of model.Result
type History struct {
	datas []model.Result
}

func newHistory(items ...string) (Driver, error) {
	datas, err := ListAll()
	if err != nil {
		return &History{}, err
	}
	return &History{
		datas: datas,
	}, nil
}

//Run writes the history data into terminal
func (r *History) Run() error {
	for _, data := range r.datas {
		fmt.Printf("%v\t%v\t%d\t%v\n", data.Filename, data.Keyword, data.Line, data.Detail)
	}
	return nil
}

//FileGetContents reads a file and gives back in the format of slice of bytes
func FileGetContents(filename string) []byte {
	contents, err := ioutil.ReadFile(filename)
	if err != nil {
		log.Fatalln(err)
	}
	return contents
}

//ListAll decodes all the Data from data file and shows it in terminal
func ListAll() ([]model.Result, error) {
	var datas []model.Result
	contents := FileGetContents(DBPath)
	dec := json.NewDecoder(bytes.NewReader(contents))

	for {
		var data model.Result
		if err := dec.Decode(&data); err == io.EOF {
			break
		} else if err != nil {
			return []model.Result{}, err
		}
		datas = append(datas, data)
	}
	return datas, nil
}

//SaveToFile saves all the searched results into config/data.json
func SaveToFile(r *model.Result) error {
	f, err := os.OpenFile(DBPath, os.O_APPEND|os.O_WRONLY, 0600)
	if err != nil {
		log.Fatalln(err)
	}
	defer f.Close()
	b, err := json.Marshal(&r)
	if err != nil {
		log.Fatalf("Error: %v", err)
	}
	if _, err = f.WriteString(string(b[:])); err != nil {
		log.Fatalln(err)
	}
	return err
}