OLA TOINO

import javax.xml.crypto.Data;
        import java.awt.*;
        import java.io.Serializable;
        import java.net.*;
        import java.io.IOException;
        import java.sql.SQLOutput;
        import java.util.ArrayList;
        import java.util.Scanner;
        import java.io.*;
        import static java.lang.Thread.sleep;

class User implements Serializable {
    private String username, password, user_type;

    public User(String username, String password, String user_type) {
        this.username = username;
        this.password = password;
        this.user_type = user_type;
    }

    public String getUsername() {
        return username;
    }

    public String getPassword() {
        return password;
    }

    public String getUser_type() { return user_type; }

    public void setUser_type(String user_type) {
        this.user_type = user_type;
    }
}

class Music implements Serializable {
    private String artist, genre, album, name;
    private int duration;

    public Music(String artist, String genre, String album, String name, int duration) {
        this.artist = artist;
        this.genre = genre;
        this.album = album;
        this.name = name;
        this.duration = duration;
    }

    public String getArtist() {
        return artist;
    }

    public String getGenre() {
        return genre;
    }

    public String getAlbum_title() {
        return album;
    }

    public String getName() {
        return name;
    }

    public int getDuration() {
        return duration;
    }
}

class Album implements Serializable {
    private String album_name, description;
    private int rating; //rating médio
    ArrayList <Music> music_list = new ArrayList<>(); //TALVEZ PÔR PRIVATE
    ArrayList <Review> review_list = new ArrayList<>(); //TALVEZ PÔR PRIVATE

    public Album(String album_name, String description, int rating, ArrayList<Music> music_list, ArrayList<Review> review_list) {
        this.album_name = album_name;
        this.description = description;
        this.rating = rating;
        this.music_list = music_list;
        this.review_list = review_list;
    }

    public String getAlbum_name() {
        return album_name;
    }

    public String getDescription() {
        return description;
    }

    public int getRating() {
        return rating;
    }

    public ArrayList<Music> getMusic_list() {
        return music_list;
    }

    public ArrayList<Review> getReview_list() {
        return review_list;
    }

    public void setDescription(String description) {
        this.description = description;
    }

    public void setRating(int rating) {
        this.rating = rating;
    }
}

class Review implements Serializable { //AINDA TENHO QUE VER PRECISA DE SERIALIZABLE OU NÃO
    private String text;
    private int rating;

    public Review(String text, int rating) {
        this.text = text;
        this.rating = rating;
    }

    public String getText() {
        return text;
    }

    public int getRating() {
        return rating;
    }
}


class DatabaseConnection {
    //Estudar ficheiros de texto e de objetos -> DONE
    //Rever projeto de POO -> DONE
    //Ver como ir buscar a música a uma diretoria e colocar música nessa diretoria

    public DatabaseConnection() {
    }

    boolean username_match (String username){

        ArrayList <User> users = new ArrayList<>();

        //Abrir para a leitura do ficheiro de objetos
        try (ObjectInputStream oin = new ObjectInputStream (new FileInputStream("Users_obj.txt"))) {
            users = (ArrayList) oin.readObject();
        } catch (IOException e){
            System.out.println(e.getMessage());
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }

        for (User u: users){
            if (u.getUsername().equals(username)) {
                return true;
            }
        }

        return false;
    }

    boolean password_match (String username, String password){
        //Tenho que procurar por username e depois por password
        //Porque se procuro apenas pela password, pode encontrar a password certa, mas ser de outro utilizador

        ArrayList <User> users = new ArrayList<>();

        try (ObjectInputStream oin = new ObjectInputStream (new FileInputStream("Users_obj.txt"))) {
            users = (ArrayList) oin.readObject();
        } catch (IOException e){
            System.out.println(e.getMessage());
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }

        for (User u: users) {
            if (u.getUsername().equals(username)) {
                if (u.getPassword().equals(password)) {
                    return true;
                }
            }

        }
        return false;
    }

}


class MessageParsing {

    public String [] MessageParsing (String message){

        //Separar por ";" e saber o número de pares
        String [] pares  = message.split(";");
        String  [] campos = new String[pares.length] ;

        if (pares.length >= 3){ //NAO SEI SE PRECISO DESTA CONDICAO OU NAO, TENHO QUE TER ATENCAO POR CAUSA DA EXCEPTION
            for (int i = 0; i < pares.length; i++){
                String [] aux = pares[i].split("\\|");
                campos[i] = aux[1];
            }
        }

        //É melhor imprimir num ciclo que vai ate ao total de elementos do array, caso contrario opde criar exception
        //System.out.println ("Tipo da mensagem: " + campos[0] + " campo2: " + campos[1] + " campo3: " + campos[2]);

        return campos;
    }

}


class UDPMulticastProtocol implements Serializable {

    private String MULTICAST_ADDRESS = "224.3.2.1";
    private int PORT = 4444; //Porto de envio
    DatabaseConnection data = new DatabaseConnection();

    public UDPMulticastProtocol() {

    }

    void login (String username, String password, MulticastSocket socket, int server_id){
        System.out.println("O cliente deseja fazer login");
        System.out.println("Username inserido: " + username);
        System.out.println("Password inserida: " + password);
        //Temos que ir à base de dados checkar se o username existe e qual a password associada e se é igual à inserida ou não


        if (data.username_match(username)){
            if (data.password_match(username, password)){ //SUCESSO
                //System.out.println("GREAT SUCCESS");
                String rsp = "server_id|" + server_id + "; type|status;" + "; logged|on; " + "msg|Welcome to DropMusic";

                try{
                    byte[] buffer = rsp.getBytes();
                    InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
                    DatagramPacket packet = new DatagramPacket(buffer, buffer.length, InetAddress.getByName(MULTICAST_ADDRESS), PORT); //mudei aqui
                    socket.send(packet);

                } catch (IOException e) {
                    e.printStackTrace();
                }

            }
            else { //PASSWORD ERRADA
                //System.out.println("PASSWORD ERRADA");
                String rsp = "server_id|" + server_id + "; type|status;" + "; logged|failed; " + "msg|Password incorreta";

                try{
                    byte[] buffer = rsp.getBytes();
                    InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
                    DatagramPacket packet = new DatagramPacket(buffer, buffer.length, InetAddress.getByName(MULTICAST_ADDRESS), PORT); //mudei aqui
                    socket.send(packet);

                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        else{ //UTILIZADOR NAO EXISTE
            //System.out.println("Utilizador nao existe");
            String rsp = "server_id|" + server_id + "; type|status;" + "; logged|failed; " + "msg|Utilizador nao existe";

            try{
                byte[] buffer = rsp.getBytes();
                InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
                DatagramPacket packet = new DatagramPacket(buffer, buffer.length, InetAddress.getByName(MULTICAST_ADDRESS), PORT); //mudei aqui
                socket.send(packet);

            } catch (IOException e) {
                e.printStackTrace();
            }
        }


    }

    void register (String username, String password, MulticastSocket socket, int server_id) throws IOException {

        ArrayList <User> users = new ArrayList<>(); //ArrayList que contém todos os utilizadores
        //Abrir para a leitura do ficheiro de objetos
        try (ObjectInputStream oin = new ObjectInputStream (new FileInputStream("Users_obj.txt"))) {
            users = (ArrayList) oin.readObject();
        } catch (IOException e){
            System.out.println(e.getMessage());
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }

        System.out.println("O cliente deseja registar-se");
        System.out.println("Username inserido: " + username);
        System.out.println("Password inserida: " + password);

        //Temos que ver se existe o username ou não
        if (!data.username_match(username)) {

            //Abrir para escrita no ficheiro de objetos
            if (users.isEmpty()){ //Se der mal, vou ver se o tamanho e zero
                User u = new User(username, password, "admin");
                users.add(u);
            }
            else{
                User u = new User(username, password, "normal");
                users.add(u);
            }

            try {
                ObjectOutputStream oS = new ObjectOutputStream(new FileOutputStream("Users_obj.txt"));
                oS.writeObject(users);
                oS.close();

            } catch (IOException e) {
                System.out.print("ERRO");
                System.out.printf("Ocorreu a exceçao %s ao escrever no ficheiro\n", e);
            }

            String rsp = "server_id|" + server_id + "; type|status;" + "; register|successful; " + "msg|Welcome to DropMusic";
            try{

                byte[] buffer = rsp.getBytes();

                InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
                DatagramPacket packet = new DatagramPacket(buffer, buffer.length, InetAddress.getByName(MULTICAST_ADDRESS), PORT); //mudei aqui
                socket.send(packet);

            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        else {
            String rsp = "server_id|" + server_id + "; type|status;" + "; register|unsuccessful; " + "msg|Esse username ja existe";

            try{
                byte[] buffer = rsp.getBytes();

                InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
                DatagramPacket packet = new DatagramPacket(buffer, buffer.length, InetAddress.getByName(MULTICAST_ADDRESS), PORT); //mudei aqui
                socket.send(packet);

            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        //SÓ PARA VER QUEM TENHO REGISTADO
        for (User uss: users){
            System.out.println(uss.getUsername() + " " + uss.getPassword() + " -> " + uss.getUser_type());
        }


    }

    void search_music (String choice, String name){
        System.out.println("O cliente deseja procurar musica");

        //1. Tenho que ir ver se a choice é all/album/music/artist/genre


        //2. Consoante a choice, apresentar o pretendido


    }

    void make_editor (String username, String novo_editor, MulticastSocket socket, int server_id){ //TESTAR
        System.out.println("O cliente deseja tornar o utilizador " + novo_editor + " editor");

        ArrayList <User> users = new ArrayList<>();

        try (ObjectInputStream oin = new ObjectInputStream (new FileInputStream("Users_obj.txt"))) {
            users = (ArrayList) oin.readObject();
        } catch (IOException e){
            System.out.println(e.getMessage());
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }

        //Verificar se o utilizador que está a querer tornar o outro num editor é também um editor
        if (data.username_match(username)) { //SUCESSO
            for (User u : users) {
                if (u.getUsername().equals(username)) {
                    if ((u.getUser_type().equals("editor")) || (u.getUser_type().equals("admin"))) {
                        System.out.println("Tem permissao para tornar o utilizador " + novo_editor + " editor");
                        for (User us : users) {
                            if (us.getUsername().equals(novo_editor)) {
                                us.setUser_type("editor");
                                //ENVIAR RESPOSTA PARA O RMI SERVER
                                //String rsp = "server_id|" + server_id + "; type|status;" + "; logged|on; " + "msg|Welcome to DropMusic";
                                String rsp = "AGORA ÉS EDITOR CARALHO!!!!!!!!!!";
                                try{
                                    byte[] buffer = rsp.getBytes();
                                    InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
                                    DatagramPacket packet = new DatagramPacket(buffer, buffer.length, InetAddress.getByName(MULTICAST_ADDRESS), PORT); //mudei aqui
                                    socket.send(packet);

                                } catch (IOException e) {
                                    e.printStackTrace();
                                }

                            } else {
                                System.out.println("Utilizador nao existe");
                                //ENVIAR RESPOSTA PARA O RMI SERVER
                            }
                        }
                    }
                }
            }
        }
        else {
            System.out.println("ESSE MANO NAO CONSTA DA BASE DE DADOS");
        }


    }

    void manage_data (String username, String choice, String name) { //TESTAR
        System.out.println("O cliente pretende gerir artistas, álbuns e músicas");

        //1. Primeiro tenho que ver se o utilizador que quer editar a info é editor
        ArrayList<User> users = new ArrayList<>();

        try (ObjectInputStream oin = new ObjectInputStream(new FileInputStream("Users_obj.txt"))) {
            users = (ArrayList) oin.readObject();
        } catch (IOException e) {
            System.out.println(e.getMessage());
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }

        if (data.username_match(username)) { //SUCESSO
            for (User u : users) {
                if (u.getUsername().equals(username)) {
                    if (u.getUser_type().equals("editor")) {
                        System.out.println("Tem permissao para editar");

                        //2. Tenho que ir ver no campo da mensagem, se é all, music, album ou review
                        switch (choice) {
                            case "artist":
                                String artist_name = name;


                                break;

                            case "album":
                                String album_title = name;

                                break;

                            case "music":
                                String music_name = name;

                                break;

                        }
                        //3. De seguida, abro esse álbum para escrita e vou editar a info que lá está


                        //4. Por fim envio a resposta ao utilizador

                    }
                }
            }
        }
    }

    void search_info (){
        System.out.println("O cliente pretende consultar detalhes sobre álbum e sobre artista");

        //1. Ver se a choice é album ou artista


        //2.



    }

    void album_review (String album_title, String text, String rating){
        System.out.println("O cliente pretende escrever uma crítica a um álbum");

        //PRIMEIRO AINDA VOU TER QUE VERIFICAR SE O CLIENTE É DO TIPO EDITOR OU NÃO


        ArrayList <Album> album_list = new ArrayList<>();

        for (Album a: album_list){
            if (a.getAlbum_name().equals(album_title)){
                a.setDescription(text);
                a.setRating(Integer.parseInt(rating));
            }
            else{ //ÁLBUM NÃO CONSTA DA BASE DE DADOS

            }
        }

        //DEPOIS VOU TER QUE ENVIAR A RESPOSTA A DIZER QUE A REVIEW FOI ESCRITA


    }

    void download (){
        System.out.println("O cliente quer fazer download de uma música");
    }

    void upload (){
        System.out.println("O cliente quer fazer upload de uma música");
    }

    void share_music (){
        System.out.println("O cliente pretende partilhar música com outros utilizadores");
    }

}


class AnswerRequests extends Thread {

    //private String MULTICAST_ADDRESS = "224.3.2.1";
    //private int PORT = 4444; //Porto de envio
    private String message;
    private String ip;
    private int porto;
    private MulticastSocket socket;
    private int server_id;

    MessageParsing parsing = new MessageParsing();

    public AnswerRequests(String message, String ip, int porto, MulticastSocket socket, int server_id) {
        this.message = message;
        this.ip = ip;
        this.porto = porto;
        this.socket = socket;
        this.server_id = server_id;
    }

    public void run(){

        UDPMulticastProtocol prot = new UDPMulticastProtocol(); //Ver se é aqui que devo chamar o construtor ou não
        System.out.println("Mensagem do cliente (IP: " + ip + ";Porto: " + porto + ")" + ": " + message);

        //Enviar resposta para o cliente
        //MulticastSocket socket = null;

        //Parsing da mensagem
        String [] pares = parsing.MessageParsing(message);
        String type = pares[1];

        // De acordo com o tipo de argumento no campo "type" vamos definir a operacao pretendida e vai ser invocado o respetivo método
        //Trata do pedido para depois enviar a resposta para o cliente
        switch (type) {
            case "login":
                prot.login(pares[2], pares[3], socket, server_id);
                break;
            case "search music":
                prot.search_music(pares[2], pares[3]);
                break;
            case "make editor": //RESOLVER O PROBLEMA: COMO É QUE HEI DE SABER QUEM É QUE ESTÁ A QUERER TORNAR ALGUÉM EDITOR, PARA SABER SE TAMBÉM É EDITOR
                String username = pares[2];
                String new_editor = pares[3];
                prot.make_editor(username, new_editor, socket, server_id); //EDITOR
                break;
            case "register":
                try {
                    prot.register(pares[2], pares[3], socket, server_id);
                } catch (IOException e) {
                    e.printStackTrace();
                }
                break;
            case "edit info":
                String user = pares[2];
                String search = pares[3];
                String msg = pares[4];
                prot.manage_data(user, search, msg); //EDITOR
                break;
            case "search info":
                prot.search_info();
                break;
            case "review":
                prot.album_review(pares[2], pares[3], pares[4]); //ANY USER
                break;
            case "share music": //DEPOIS
                prot.share_music();
                break;
            case "download": //DEPOIS
                prot.download();
                break;
            case "upload":  //DEPOIS
                prot.upload();
                break;
        }

    }

}

public class MulticastServer extends Thread {

    private String MULTICAST_ADDRESS = "224.3.2.1";
    private int PORT = 4321; //Porto de recepção
    private int BUFFER_SIZE = 4096;
    private static int SERVER_ID;

    MessageParsing parsemsg = new MessageParsing();

    public static void main(String[] args) {
        SERVER_ID = Integer.parseInt(args[0]); //ID DO SERVIDOR
        MulticastServer server = new MulticastServer();
        server.start(); //Ao invocar o start da thread, estamos a chamar o metodo run()
    }

    public MulticastServer() {
        //super ("Server Multicast #" + (long) (Math.random() * 100));
        super ("Server Multicast #" + SERVER_ID);
    }

    public void run(){
        MulticastSocket socket = null;

        System.out.println(this.getName() + " running..."); //Vai buscar o que está dentro do método do construtor

        try {
            socket = new MulticastSocket(PORT); //cria socket e dá bind
            InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
            socket.joinGroup(group); //dá join ao grupo multicast


            while (true) {
                //Recepcao
                byte[] inBuffer = new byte[BUFFER_SIZE];
                DatagramPacket msgIn = new DatagramPacket(inBuffer, inBuffer.length);
                socket.receive(msgIn); //Recebe o datagrama UDP

                String clientIP = msgIn.getAddress().getHostAddress(); //Endereço IP do cliente que enviou a mensagem
                int clientport = msgIn.getPort(); //Porto do cliente

                String message = new String(msgIn.getData(), 0, msgIn.getLength());
                String [] fields = parsemsg.MessageParsing(message);

                if (Integer.parseInt(fields[0]) == SERVER_ID){
                    System.out.println("SO VOU RESPONDER EU -> SERVER " + SERVER_ID);

                    //Vai buscar a informação da mensagem e lança a thread para tratar do pedido
                    AnswerRequests work = new AnswerRequests(message, clientIP , clientport, socket, SERVER_ID); //Também vou passar a socket por parâmetro
                    work.start();
                }
                else{
                    //Se for um registo, mudança de informação de álbuns ou cenas que têm que ser replicadas
                    //Ou seja, tem que ir para as 3 bases de dados e vai ser este "else" que vai tratar disso
                    System.out.println("VOU ESTAR QUIETO -> SERVER " + SERVER_ID);

                }

            }

        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            socket.close(); //Fechar a socket
        }
    }

}

/*
try{
            socket = new MulticastSocket(); //Não é necessário dar bind porque está apenas a enviar
            socket.setTimeToLive(1);

            Scanner keyboardScaner = new Scanner(System.in);
            while (true){
                String keyboard = keyboardScaner.nextLine();
                byte[] buffer = keyboard.getBytes();

                InetAddress group = InetAddress.getByName(MULTICAST_ADDRESS);
                DatagramPacket packet = new DatagramPacket(buffer, buffer.length, InetAddress.getByName(MULTICAST_ADDRESS), PORT); //mudei aqui
                socket.send(packet);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            socket.close();
        }
 */