package train.client;

import train.mvc.*;
import train.parameter.*;
import java.util.*;
import javax.swing.table.*;
import train.command.*;
/**
 * <p>Title: </p>
 *
 * <p>Description: </p>
 *
 * <p>Copyright: Copyright (c) 2006</p>
 *
 * <p>Company: 沈陽化工學(xué)院計(jì)算機(jī)</p>
 *
 * @author 彭勝勇 胡林
 * @version 1.0
 */
public class ModelNetImpl extends AbstractTableModel implements Model {
    Client client;
    View view;
    List list;
    TableModelTrain tableModelTrain;
    public ModelNetImpl() {
        this.client = new Client(CommonConstant.localhost,
                                 CommonConstant.localport);
        client.connect();
    }

    public void addViewListener(View view, TableModelTrain tableModelTrain) {
        this.view = view;
        this.tableModelTrain = tableModelTrain;
    }



    public void deleteTrain(Train train) {
        DeleteTrainCommand deleteTrainCommand = new DeleteTrainCommand(train);
        client.write(deleteTrainCommand);
        deleteTrainCommand = (DeleteTrainCommand) client.read();
        if (!deleteTrainCommand.fail) {
            this.refreshTrainTable();
        } else {
            try {
                throw new Exception(" Delete " + train + " fail!");
            } catch (Exception ex) {
                ex.printStackTrace();
            }
        }

    }


    public void insertTrain(Train train) {
        InsertTrainCommand insertTrainCommand = new InsertTrainCommand(train);
        client.write(insertTrainCommand);
        insertTrainCommand = (InsertTrainCommand) client.read();
        if (!insertTrainCommand.fail) {
            this.refreshTrainTable();
        } else {
            try {
                throw new Exception(" Insert " + train + " fail!");
            } catch (Exception ex) {
                ex.printStackTrace();
            }
        }

    }



    public void updateTrain(Train train) {
        UpdateTrainCommand updateTrainCommand = new UpdateTrainCommand(train);
        client.write(updateTrainCommand);
        updateTrainCommand = (UpdateTrainCommand) client.read();
        if (!updateTrainCommand.fail) {
            this.refreshTrainTable();
        } else {
            try {
                throw new Exception(" Delete " + train + " fail!");
            } catch (Exception ex) {
                ex.printStackTrace();
            }
        }

    }


    public List getTrain() {
        GetTrainCommand getTrainCommand = new GetTrainCommand();
        client.write(getTrainCommand);
        Object obj = client.read();
        getTrainCommand = (GetTrainCommand) obj;
        return getTrainCommand.getList();
    }
    public List getStation() {
         GetStationCommand getStationCommand = new GetStationCommand();
         client.write(getStationCommand);
         Object obj = client.read();
         getStationCommand = (GetStationCommand) obj;
         return getStationCommand.getList();
     }



    public List queryTrainWiID(String trainID) {
        QueryTrainWiIDCommand  queryTrainWiIDCommand = new QueryTrainWiIDCommand(trainID);
        client.write(queryTrainWiIDCommand);
        Object obj = client.read();
        queryTrainWiIDCommand = (QueryTrainWiIDCommand) obj;
        return queryTrainWiIDCommand.getTrainList();
    }
    public List queryStation(String staIndex) {
        QueryStationCommand  queryStationCommand = new QueryStationCommand(staIndex);
        client.write(queryStationCommand);
        Object obj = client.read();
        queryStationCommand = (QueryStationCommand) obj;
        return queryStationCommand.getStationList();
    }

    public List queryTrainWiSta(String fromSta,String endSta) {
        QueryTrainWiStaCommand queryTrainWiStaCommand = new QueryTrainWiStaCommand(fromSta,endSta);
        client.write(queryTrainWiStaCommand);
        Object obj = client.read();
        queryTrainWiStaCommand = (QueryTrainWiStaCommand) obj;
        return queryTrainWiStaCommand.getTrainList();
    }


    public int getColumnCount() {
        return 0;
    }

    public int getRowCount() {
        return 0;
    }

    public Object getValueAt(int rowIndex, int columnIndex) {
        return null;
    }


    public void refreshTrainTable() {
        this.view.refreshTableTrainChange();
    }

    public int getIndex(Object obj, int objType) {
        int index = 0;
        switch (objType) {
        case 0: //CommonConstant.objUser
            Train train = (Train) obj;
            list = this.getTrain();
            /*for (; index < list.size(); index++) {
                User userTest = (User) list.get(index);
                if (userTest.getUserID().equals(user.getUserID())) {
                    return index;
                }
            }*/
            break;
        default:
            index = -1;
            break;
        }
        return index;

    }
}