Ejemplo n.º 1
0
func paint(user *dht.User) {

	// clear space
	for j := 1; j < 100; j++ {
		fmt.Println("")
	}

	// new messages?
	usersWithPendingMessages := make([]string, 0)
	for peer, _ := range user.MessageHistory {
		areNew, _ := user.AreNewMessagesFrom(peer)
		if areNew {
			usersWithPendingMessages = append(usersWithPendingMessages, peer)
		}
	}

	// print users with pending messages
	for _, peer := range usersWithPendingMessages {
		fmt.Printf("New message(s) from `%s`!\n", peer)
	}
	fmt.Printf("\n\n========================================\n")

	// are we current chatting?
	if user.Current != "" {
		fmt.Printf("Conversation with `%s`:\n\n", user.Current)

		newMessages := user.AllMessagesFromUser(user.Current)

		messages := make([]dht.SendMessageArgs, 0)
		for _, msg := range newMessages {
			messages = append(messages, *msg)
		}

		sortutil.AscByField(messages, "Timestamp")
		for i := 0; i < len(messages); i++ {
			msg := messages[i]
			fmt.Printf("%s> %s\n", msg.FromUsername, msg.Content)
		}
	}

	fmt.Printf("=========================================\n")
	fmt.Printf("me> ")
}
Ejemplo n.º 2
0
func startChat() {
	reader := bufio.NewReader(os.Stdin)

	// load the Peerchat banner
	content, err := ioutil.ReadFile("peerchat.txt")
	if err != nil {
		//Do something
	}
	fmt.Println(string(content))

	// get username
	fmt.Printf("Enter username: "******"Enter IP Address (xxx.xxx.xxx.xxx:yyyy): ")
	address := input(reader)

	// search for this user's history
	files, _ := filepath.Glob("/tmp/*.gob")
	userfile := ""
	user := new(dht.User)
	for _, file := range files {
		if file == dht.UsernameToPath(username) {
			userfile = file
		}
	}

	// did we find the file?
	if userfile != "" {
		// we found this user, load from file
		user = dht.Login(username, address)

	} else {
		// we did not find a matching user
		fmt.Printf("\n[*] Looks like you haven't logged in on this computer before! Would you like to create a new network, or join an existing one?\n")
		fmt.Printf("Join existing? Type (Y/N):")
		join := input(reader)

		if join == "Y" || join == "y" {
			// join existing network, ask for boostrap IP
			fmt.Printf("\n[*] To join an existing network, please enter an IP address/port of a friend (xxx.xxx.xxx.xxx:yyyy): ")
			boostrapAddress := input(reader)
			user = dht.RegisterAndLogin(username, address, boostrapAddress)

		} else {
			// creating a new network, simply create new user
			// use bogus address and start with empty routing table
			user = dht.RegisterAndLogin(username, address, "")

		}
	}

	// we're now logged in with a user
	fmt.Printf("Connecting to Peerchat")
	time.Sleep(300 * time.Millisecond)
	fmt.Printf(".")
	time.Sleep(300 * time.Millisecond)
	fmt.Printf(".")
	time.Sleep(300 * time.Millisecond)
	fmt.Printf(".\n")

	// prompt them to chat
	// fmt.Printf("Connected as user: %+v\n", user)

	// update loop
	go func() {
		notifications := user.GetNotificationsChannel()
		for {
			<-notifications
			paint(user)
		}
	}()

	// input loop
	peer := ""
	for {
		// State 1) get a user to chat with
		if peer == "" {
			fmt.Printf("User to chat with: ")
			peer = input(reader)
			fmt.Printf("Starting to talk to: %s\n", peer)
			user.UpdateCurrentPeer(peer)
			fmt.Printf("me> ")

			// State 2) continue chatting
		} else {

			text := input(reader)

			if text == "" {
				// do nothing
				fmt.Printf("%s> \n", peer)
				paint(user)

			} else if text[0] == 92 {
				// switching users to chat with
				peer = text[1:]
				fmt.Printf("Swtiching to talk to: %v\n", peer)
				user.UpdateCurrentPeer(peer)
				fmt.Printf("me> ")
				paint(user)

			} else if text == "exit" {
				// exit peerchat
				fmt.Printf("Exiting Peerchat!\n")
				user.Logoff()
				break

			} else {
				// send the message!
				user.SendMessage(user.Current, text)
				paint(user)
			}
		}
	}
}