Ominaisuus valmis?

Signed-off-by: laurimaaninka <lauri.maaninka@gmail.com>

src/main/java/fi/lpam/gui/elementit/KuljetaTableCell.java

@@ -3,49 +3,36 @@ package fi.lpam.gui.elementit;
 import fi.lpam.dataluokat.Asiakas;
 import fi.lpam.dataluokat.Kuljetus;
 import fi.lpam.gui.KuljetusListat;
-import javafx.collections.FXCollections;
-import javafx.collections.ObservableList;
 import javafx.geometry.Insets;
 import javafx.scene.control.Button;
 import javafx.scene.control.ContentDisplay;
 import javafx.scene.control.TableCell;
-import javafx.scene.control.TableView;
 import javafx.scene.layout.StackPane;
 
-import java.sql.Array;
-import java.util.ArrayList;
-import java.util.Collections;
 import java.util.HashMap;
 
 public class KuljetaTableCell extends TableCell<Kuljetus, Boolean> {
-    private final Button nappi = new Button("Vaihda");
     private final StackPane paddedNappi = new StackPane();
 
-    public KuljetaTableCell(final HashMap<Integer, Asiakas> asiakkaat, final TableView<Kuljetus> table) {
+    public KuljetaTableCell(HashMap<Integer, Asiakas> asiakkaat) {
         paddedNappi.setPadding(new Insets(3));
+        Button nappi = new Button("Muuta");
         paddedNappi.getChildren().add(nappi);
-        nappi.setOnAction(_ -> {
+        nappi.setOnMouseClicked(_ -> {
+            getTableView().getSelectionModel().select(getIndex());
             Kuljetus kuljetus = getTableRow().getItem();
-            System.out.println(kuljetus);
+
             if (kuljetus == null) return;
             if (kuljetus.kuljetetaan()) {
                 kuljetus.nollaa();
-                System.out.println("Nollattu");
-                System.out.println(kuljetus);
+            } else {
+                kuljetus.palautaMäärät(asiakkaat.get(kuljetus.getAsiakasID()));
             }
-            else {
-                Asiakas asiakas = asiakkaat.get(kuljetus.getAsiakasID());
-                System.out.println(asiakas);
-                kuljetus.palautaMäärät(asiakas);
-                System.out.println("Palautettu");
-            }
-            table.refresh();
+            getTableRow().setItem(new Kuljetus(kuljetus));
             KuljetusListat.tallennusStatus.setText("Tallennettu: Ei");
         });
     }
 
-
-    /** places an add button in the row only if the row is not empty. */
     @Override protected void updateItem(Boolean item, boolean empty) {
         super.updateItem(item, empty);
         if (!empty) {