package main

import (
    "fmt"
    "net"
    "os"
)

type Mail struct {
    clientName string
    from       string
    to         string
    data       string
    reciveData bool
}

func main() {

    listen, err := net.ListenTCP("tcp", &net.TCPAddr{net.ParseIP("127.0.0.1"), 25, ""})
    if err != nil {
        fmt.Println("監(jiān)聽端口失敗:", err.Error())
        return
    }
    defer listen.Close()
    recive(listen)

}

func recive(listen *net.TCPListener) {

    for {
        conn, err := listen.AcceptTCP() //接收連接
        defer conn.Close()
        if err != nil {
            fmt.Println("客戶端連接異常", err.Error()) //客戶端連接異常
            continue
        }
        conn.Write([]byte("220 GoMailSever "))
        fmt.Println("新客戶端", conn.RemoteAddr().String())
        go func() {

            tmpBuffer := make([]byte, 0)           //創(chuàng)建待處理字節(jié)切片
            readerChannel := make(chan []byte, 16) //創(chuàng)建channel通信管道
            _mail := &Mail{}
            go reader(readerChannel, conn, _mail) //channel信息接收端
            data := make([]byte, 1024)            //創(chuàng)建輸出緩沖切片

            for {
                i, err := conn.Read(data) //從Socket讀取數(shù)據(jù)
                if err != nil {
                    fmt.Println("讀取發(fā)生錯誤", err.Error()) //讀取失敗發(fā)生錯誤
                    return
                }
                tmpBuffer = DepackMail(append(tmpBuffer, data[:i]...), readerChannel) //用自定義協(xié)議處理字節(jié)

                //fmt.Println(data, string(data))
            }

        }()

    }
}
func reader(readerChannel chan []byte, conn *net.TCPConn, _mail *Mail) {

    for {
        select {
        case data := <-readerChannel:
            fmt.Println(string(data[0:4]))
            switch string(data[0:4]) {
            case "HELO":
                _mail.clientName = string(data[4+2:])
                conn.Write([]byte("250 Give Me Mail"))
                break
            case "MAIL":
                _mail.from = string(data[4+2:])
                conn.Write([]byte("250 Give I am fine"))
                break
            case "RCPT":
                _mail.to = string(data[4+2:])
                conn.Write([]byte("250 Give I am fine"))
                break
            case "DATA":
                conn.Write([]byte("354 Give I am fine"))
                break
            case "QUIT":
                conn.Write([]byte("221 ByeBye"))
                file, _ := os.Create("./mail/3.eml")
                file.WriteString(_mail.data)
                file.Close()

                fmt.Println(_mail)
                break
            default:
                _mail.data = _mail.data + string(data)
                fmt.Println(string(_mail.data[len(_mail.data)-1:]))
                if string(_mail.data[len(_mail.data)-1:]) == "." {
                    fmt.Println("111")
                    conn.Write([]byte("250 Give I am fine"))
                }

                break
            }

        }
    }
}
func DepackMail(buffer []byte, readerChannel chan []byte) []byte {
    length := len(buffer)

    var i int

    for i = 0; i < length; {
        if length < i+4 { //2位為尾標記
            break
        }
        var sIndex int
        for dump := i; dump < length-1; dump++ {
            if buffer[dump] == byte(13) && buffer[dump+1] == byte(10) {

                sIndex = dump
            }
        }
        if sIndex <= 0 {
            break
        }
        data := buffer[i:sIndex]

        readerChannel <- data //向管道發(fā)送數(shù)據(jù)
        i = sIndex + 2
    }

    if i == length { //全部處理完畢返回空包
        return make([]byte, 0)
    }
    return buffer[i:] //未處理完畢返回 剩余包
}