main.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strconv"
	"time"

	"github.com/google/uuid"
	log "github.com/sirupsen/logrus"
)

func main() {
	// Discover all services on the network (e.g. _workstation._tcp)
	log.SetLevel(log.DebugLevel)
	port := os.Getenv("PORT")
	if port == "" {
		port = "8080"
	}

	dbname := os.Getenv("DBNAME")
	if dbname == "" {
		dbname = "disaster.db"
	}

	id := uuid.New()

	s, err := NewStore(dbname)
	if err != nil {
		fmt.Println(err)
		fmt.Println("couldn't create store")
	}

	tmp := MyInfo{
		UserID: id,
		Name:   "me1",
		Email:  "me@example.com",
		Phone:  "(555) 555 5555",
		Lat:    -1,
		Long:   -1,
		Time:   time.Now(),
	}
	s.SetMyInfo(&tmp)
	a := NewServer("0.0.0.0", port, id, s)

	portInt, _ := strconv.Atoi(port)

	go RegisterService(id, portInt)
	peerChan := make(chan Peer)
	go FindPeers(peerChan)
	go a.Serve()
	var peerList []Peer
	cnt := 0
	var netClient = &http.Client{
		Timeout: time.Second * 10,
	}
	for {
		select {
		case peer := <-peerChan:
			peerList = append(peerList, peer)
			break
		case <-time.Tick(time.Second * 3):
			if cnt%3 == 0 {
				log.Info("POSTING")
				msgs := s.GetAllMessages(id)
				enc, _ := json.Marshal(msgs)
				_, err := netClient.Post("http://52.116.40.230:8000/update", "application/json", bytes.NewReader(enc))
				if err != nil {
					log.Info("No internet")
				}
			}
			cnt++
			log.Info("SYNCING")
			var npl []Peer
			for _, p := range peerList {
				if p.UUID == id.String() {
					continue
				}
				oneCon := false
				for _, addr := range p.IPS {
					curString := addr.String()
					log.Debug(curString)
					resp, err := netClient.Get("http://" + curString + ":" + strconv.Itoa(p.Port) + "/messages")

					if err != nil {
						log.Error("Could not connect to peer", curString)
						log.Error(err)
					} else {
						defer resp.Body.Close()
						oneCon = true
						if resp.StatusCode == http.StatusOK {
							bodyBytes, err := ioutil.ReadAll(resp.Body)
							if err != nil {
								log.Fatal(err)
							}

							var encMessages []EncryptedMessage
							json.Unmarshal(bodyBytes, &encMessages)
							myMessages := s.GetAllMessages(id)
							var totMessages []EncryptedMessage
							for _, msg := range encMessages {
								found := false
								for _, msg2 := range myMessages {
									if string(msg.Body) == string(msg2.Body) {
										found = true
										break
									}
								}
								if !found {
									totMessages = append(totMessages, msg)
								}
							}
							if len(totMessages) == 0 {
								continue
							}
							fmt.Println("Got ", len(totMessages), " New Messages")
							s.SaveMessages(totMessages)
						}
					}
				}
				if oneCon {
					npl = append(npl, p)
				}
			}
			peerList = npl
		}
	}

}