package main

import (

	"../xmpp"

	"crypto/tls"

	"encoding/xml"

	"flag"

	"fmt"

	"log"

	"os"

	"strings"

)

type StdLogger struct {

}

func (s *StdLogger) Log(v ...interface{}) {

	log.Println(v...)

}

func (s *StdLogger) Logf(fmt string, v ...interface{}) {

	log.Printf(fmt, v...)

}

func init() {

	logger := &StdLogger{}

	// xmpp.Debug = logger

	xmpp.Info = logger

	xmpp.Warn = logger

}

// Demonstrate the API, and allow the user to interact with an XMPP

// server via the terminal.

func main() {

	var jid xmpp.JID

	flag.Var(&jid, "jid", "JID to log in as")

	var pw *string = flag.String("pw", "", "password")

	flag.Parse()

	if jid.Domain == "" || *pw == "" {

		flag.Usage()

		os.Exit(2)

	}

	tlsConf := tls.Config{InsecureSkipVerify: true}

	c, err := xmpp.NewClient(&jid, *pw, tlsConf, nil)

	if err != nil {

		log.Fatalf("NewClient(%v): %v", jid, err)

	}

	defer close(c.Send)

	go func(ch <-chan xmpp.Stanza) {

		for obj := range ch {

			fmt.Printf("s: %v\n", obj)

		}

		fmt.Println("done reading")

	}(c.Recv)

	err = c.StartSession(&xmpp.Presence{})

	if err != nil {

		log.Fatalf("StartSession: %v", err)

	}

	c.Roster.Update()

	roster := c.Roster.Get()

	fmt.Printf("%d roster entries:\n", len(roster))

	for i, entry := range roster {

		fmt.Printf("%d: %v\n", i, entry)

	}

	p := make([]byte, 1024)

	for {

		nr, _ := os.Stdin.Read(p)

		if nr == 0 {

			break

		}

		s := string(p)

		dec := xml.NewDecoder(strings.NewReader(s))

		t, err := dec.Token()

		if err != nil {

			fmt.Printf("token: %s\n", err)

			break

		}

		var se *xml.StartElement

		var ok bool

		if se, ok = t.(*xml.StartElement); !ok {

			fmt.Println("Couldn't find start element")

			break

		}

		var stan xmpp.Stanza

		switch se.Name.Local {

		case "iq":

			stan = &xmpp.Iq{}

		case "message":

			stan = &xmpp.Message{}

		case "presence":

			stan = &xmpp.Presence{}

		default:

			fmt.Println("Can't parse non-stanza.")

			continue

		}

		err = dec.Decode(stan)

		if err == nil {

			c.Send <- stan

		} else {

			fmt.Printf("Parse error: %v\n", err)

			break

		}

	}

	fmt.Println("done sending")

}