// Pop3 mailcheck - check the number of new mails in your maildrop
// - TODO: use windows.forms, use #develop
// (c) Anzuhan

using System;
using System.IO;
using System.Net;
using System.Net.Sockets;
using System.Threading;
using System.Text;
using System.Text.RegularExpressions;

namespace MailChecker
{
    class MailCount
    {
        private static string server;
        private static int port;
        private static string username;
        private static string password;

        // socket stuff
        private static TcpClient pop3;
        private static NetworkStream netstream;
        private static StreamReader reader;
        private static StreamWriter writer;

        public MailCount(string s, int p, string u, string pass) {
            server = s;
            port = p;
            username = u;
            password = pass;

            this.Connect();
        }

        // connect to server, and then authenticate
        private void Connect() {
            try {
                pop3 = new TcpClient(server, port);
                netstream = pop3.GetStream();

                reader = new StreamReader(netstream);
                writer = new StreamWriter(netstream);
            }
            catch (Exception e) {
                Console.WriteLine("Error: " + e.ToString());
            }
        }

        // send password to server
        public void GetMailCount() {
            string linedata;
            Regex re, re2;
            Match m, m2;

            try {
                writer.WriteLine("USER " + username);
                writer.Flush();

                writer.WriteLine("PASS " + password);
                writer.Flush();

                                // just a small delay
                Thread.Sleep(1000);

                writer.WriteLine("STAT");
                writer.Flush();

                while (true) {

                    // check if it was a correct STAT reply
                    // such as +OK 6 320
                    while ((linedata = reader.ReadLine()) != null) {
                        re = new Regex(@"^\+OK\s(\d+?)\s(\d+?)$"); // +OK x y
                        re2 = new Regex(@"^-ERR\s(.*)$"); // -ERR error

                        m = re.Match(linedata);
                        m2 = re2.Match(linedata);

                        // should it match, print the number of new mails and disconnect
                        if (m.Success) {
                            Console.WriteLine("You have " + m.Groups[1].Captures[0].Value + " new mails!");
                            break;
                        }

                        // did we get an error?
                        else if (m2.Success) {
                            Console.WriteLine("Error: " + m2.Groups[1].Captures[0].Value);
                            break;
                        }
                    }

                    reader.Close();
                    writer.Close();
                    pop3.Close();
                    break;
                }
            }
            catch (Exception e) {
                Console.WriteLine("Error: " + e.ToString());
            }
        }
    }

    // a test class
    class GetMail
    {
        static void Main(string[] args) {
            MailCount mails = new MailCount("server.com", 110, "someuser", "somepassword");
            mails.GetMailCount();
        }
    }
}