segment-node.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"os"

	pwl "github.com/justjanne/powerline-go/powerline"
)

const pkgfile = "./package.json"

type packageJSON struct {
	Version string `json:"version"`
}

func segmentNode(p *powerline) []pwl.Segment {
	stat, err := os.Stat(pkgfile)
	if err != nil {
		return []pwl.Segment{}
	}
	if stat.IsDir() {
		return []pwl.Segment{}
	}
	pkg := packageJSON{"!"}
	raw, err := ioutil.ReadFile(pkgfile)
	if err != nil {
		return []pwl.Segment{}
	}
	err = json.Unmarshal(raw, &pkg)
	if err != nil {
		return []pwl.Segment{}
	}
	return []pwl.Segment{{
		Name:       "node-segment",
		Content:    pkg.Version + " \u2B22",
		Foreground: p.theme.NodeFg,
		Background: p.theme.NodeBg,
	}}
}