gmail2go.go

package main

import (
	"bufio"
	"bytes"
	"encoding/json"
	"flag"
	"fmt"
	"log"
	"os"
	"os/exec"
	"path"
	"strings"

	"github.com/rif/gmail2go/passwd"
	"github.com/rif/gmail2go/rss"
)

var (
	config      = flag.String("config", path.Join(os.Getenv("HOME"), ".gmail2gorc"), "the user accounts file")
	account     = flag.String("account", "", "adds/updates/deletes user:password to the accounts file (leave password empty to delete)")
	color       = flag.Bool("color", false, "use terminal output colors")
	notify      = flag.Bool("notify", false, "send libnotify message")
	accountsMap = make(map[string]string)
)

func main() {
	flag.Parse()
	fin, err := os.Open(*config)
	if err == nil {
		// if the config file was opened decrypt it and unmarshal accounts map
		r := bufio.NewReader(fin)
		res, err := passwd.Decrypt(r, make([]byte, 16), make([]byte, 16))
		if err != nil {
			log.Fatal("Could not decrypt accounts file: ", err)
		} else {
			dec := json.NewDecoder(res)
			if err := dec.Decode(&accountsMap); err != nil {
				log.Fatal("Could not decode accounts content to json: ", err)
			}
		}
	}
	// if account parameter has a value operate the changes on the map and write the encrypted json
	if *account != "" {
		fin, err = os.Create(*config)
		if err != nil {
			log.Fatal("Cannot open account file: ", err)
		}
		defer fin.Close()
		w := bufio.NewWriter(fin)
		up := strings.SplitN(*account, ":", 2)
		if len(up) != 2 {
			log.Fatal("Incorrect set string use user:pass")
		}
		if up[1] != "" {
			accountsMap[up[0]] = up[1]
		} else {
			delete(accountsMap, up[0])
		}
		var out bytes.Buffer
		enc := json.NewEncoder(&out)
		err := enc.Encode(accountsMap)
		if err != nil {
			log.Fatal("Could not encode accounts to json", err)
		} else {
			err = passwd.Encrypt(w, &out, make([]byte, 16), make([]byte, 16))
			if err != nil {
				log.Fatal("Could not encrypt accounts json string to file", err)
			}
			w.Flush()
		}
	}

	// prepare the colors for display
	yellow, green, red, reset := "", "", "", ""
	if *color {
		yellow, green, red, reset = "\033[1;33m", "\033[0;32m", "\033[0:31m", "\033[0m"
	}

	emailCountMap := make(map[string][]string)
	// show accounts
	index := 0
	for user, _ := range accountsMap {
		if index > 0 {
			fmt.Print(green + " | ")
		}
		fmt.Print(yellow + user)
		index++
	}
	fmt.Println(reset + "\n")
	// show unread mails
	for user, pass := range accountsMap {
		mails, err := rss.Read("https://mail.google.com/mail/feed/atom", user, pass)
		if err != nil {
			fmt.Println(reset)
			log.Print(user+" error: ", err)
			continue
		}
		// iterate over mails
		for _, m := range mails {
			fmt.Println(yellow+"["+user+"] "+red, m.Author, ": "+reset, m.Title, "\n\t\t", m.Summary)
			emailCountMap[user] = append(emailCountMap[user], m.Title)
		}
	}

	// show the notification
	if len(emailCountMap) > 0 {
		if *notify {
			message := ""
			for k, v := range emailCountMap {
				// pute entire title if only one message
				info := fmt.Sprintf("%d", len(v))
				if len(v) == 1 {
					info = v[0]
				}
				message += fmt.Sprintf("%v(%v), ", k, info)
			}
			message = strings.TrimRight(message, ", ")
			cmd := exec.Command("/usr/bin/notify-send",
				"-i",
				"/usr/share/notify-osd/icons/gnome/scalable/status/notification-message-email.svg",
				"-a",
				"gmail2go",
				"You have new mails:",
				message)
			cmd.Run()
		}
		os.Exit(0)
	}
	os.Exit(1)
}

func isPressent(mp map[string]int, searched string) bool {
	for k := range mp {
		if k == searched {
			return true
		}
	}
	return false
}

func getKeyList(mp map[string]int) (keys []string) {
	for k := range mp {
		keys = append(keys, k)
	}
	return
}