avens/main.go at main · FrappeFortyTwo/avens · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package main

import (
	"io/ioutil"
	"time"

	"github.com/getlantern/systray"
)

const (
	// files
	fconfig = "config.csv"
	ficon   = "assets/logo.png"
	fwall   = "assets/desktop.png"
	// source
	sUnsplash = "https://source.unsplash.com/collection/"
	sKeywords = "76468395/"
)

var (
	err      error
	sz       string        // screen resolution
	de       string        // desktop environment
	fpath    string        // path of desktop.png
	delay    time.Duration = 1
	record   int           = 0
	nextFlag               = false
)

func main() {
	onExit := func() { println("exiting...") }
	// start systray process
	systray.Run(onReady, onExit)
}

func onReady() {
	// paint interface
	go func() {
		// set icon & tooltip
		icon, err := ioutil.ReadFile(ficon)
		checkErr("unable to open icon file : assets/logo.ico", err)
		systray.SetIcon(icon)
		systray.SetTooltip("Avens : Dynamic Wallpapers")

		// set menu options
		m := make(map[int]*systray.MenuItem)
		m[1] = systray.AddMenuItem("1 hr delay", "")
		m[4] = systray.AddMenuItem("4 hrs delay", "")
		m[6] = systray.AddMenuItem("6 hrs delay", "")
		m[10] = systray.AddMenuItem("10 hrs delay", "")
		systray.AddSeparator()
		nextImg := systray.AddMenuItem("Next Wallpaper", "")
		systray.AddSeparator()
		saveImg := systray.AddMenuItem("Save Wallpaper", "")
		systray.AddSeparator()
		quitTray := systray.AddMenuItem("Quit", "")

		// toggle delay options
		toggle := func(k int) {
			delay = time.Duration(k)
			updateConfig()
			for i := range m {
				if i == k {
					m[i].Disable()
				} else {
					m[i].Enable()
				}
			}
		}

		toggle(int(delay))

		// render loop
		for {
			// wait until a case satisfies
			select {
			case <-m[1].ClickedCh:
				toggle(1)
			case <-m[4].ClickedCh:
				toggle(4)
			case <-m[6].ClickedCh:
				toggle(6)
			case <-m[10].ClickedCh:
				toggle(10)
			case <-nextImg.ClickedCh:
				nextWall()
			case <-saveImg.ClickedCh:
				saveWall()
			case <-quitTray.ClickedCh:
				systray.Quit()
			}
		}
	}()

	go func() {
		for {
			time.Sleep(delay * time.Hour)
			nextWall()
		}
	}()
}