package ftp

import ftp.messages.FileMessage
import ftp.messages.LoginMessage
import ftp.messages.Message
import ftp.messages.TextMessage
import java.io.BufferedReader
import java.io.File
import java.io.PrintWriter
import java.net.Socket
import java.nio.file.Files
import java.nio.file.Paths
import kotlin.concurrent.thread

/**
 * Created by Florian Hofmair on 3/14/17.
 */

class FTPClient {
    lateinit var client: Client

    fun connect(ip: String, port: Int) {
        val c = Socket(ip, port)
        client = Client(c)

        thread {
            while (!client.conn.isClosed) {
                val msg = client.input.readObject() as Message

                when(msg) {
                    is TextMessage -> println(msg.text)
                    is FileMessage -> println(msg.path + "\n" + msg.content)
                    is LoginMessage -> println(msg.user + ":" + msg.pass)
                }
            }
        }
    }

    fun disconnect() {
        client.conn.close()
        client.output.close()
        client.input.close()
    }
}

fun main(args: Array<String>) {
    val client = FTPClient()
    client.connect("192.168.101.75", 6666)

    println("Username: ")
    val user = readLine()
    println("Password: ")
    val pass = readLine()

    client.client.sendLogin(user!!, pass!!)

    loop@ while(true) {
        val line = readLine()
        if(line != null) {
            when(line) {
                "quit" -> break@loop
                else -> {
                    if(line.startsWith("put")) {
                        val a = line.split(' ').drop(1)
                        if(a.size < 2)
                            continue@loop

                        if(Files.exists(Paths.get(a[1]))) {
                            client.client.sendFile(File(a[1]).readText(), a[2])
                            println("Uploading ${a[1]} to ${a[2]}")
                        } else
                            println("File ${a[1]} not found")
                    }
                    else {
                        client.client.sendMessage(line)
                    }
                }
            }
        }
    }
    client.disconnect()
}