FileIO.java

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;

import javafx.scene.control.Alert;
import javafx.scene.control.TextArea;
import javafx.scene.control.Alert.AlertType;
import javafx.scene.image.Image;
import javafx.stage.FileChooser;
import javafx.stage.Stage;

public class FileIO {

  private TextArea TextAreaUI;

  private File UserInputFile;

  private FileChooser Chooser;
  private Stage PrimaryStage;

  private boolean isreadOnly = false;

  public FileIO(TextArea tf, Stage primaryStage) {
    this.PrimaryStage = primaryStage;

    Chooser = new FileChooser();
    FileChooser.ExtensionFilter extFilter = new FileChooser.ExtensionFilter("Text Document (*.txt)", "*.txt");
    FileChooser.ExtensionFilter extFilter2 = new FileChooser.ExtensionFilter("All Files", "*.*");
    Chooser.getExtensionFilters().add(extFilter);
    Chooser.getExtensionFilters().add(extFilter2);

    TextAreaUI = tf;
  }

  public void OpenFile() {

    UserInputFile = Chooser.showOpenDialog(PrimaryStage);

    boolean isuserinputnontextfile;
    try {
      isuserinputnontextfile = IsBinaryFile(UserInputFile);
    } catch (IOException e1) {
      // เปิดไฟล์ไม่ได้ ให้จบการเปิดไฟล์เลย
      e1.printStackTrace();
      return;
    }
    StringBuilder texttoshow = new StringBuilder();
    if (isuserinputnontextfile) {
      // เปิดไฟล์ Binary ขึ้นมาแจ้งผู้ใช้ว่ามันเปิดไม่ได้
      Stage alertStage = new Stage();
      Alert alert = new Alert(AlertType.ERROR);
      alertStage = (Stage) alert.getDialogPane().getScene().getWindow();
      alert.setHeaderText(null);
      alert.setTitle("Error");
      alert.setContentText("This program can't open this file.");
      alertStage.getIcons().add(new Image(getClass().getResourceAsStream("Picture/Error.png")));
      alert.show();
      return;
    }
    try (
        FileReader Reader = new FileReader(UserInputFile, StandardCharsets.UTF_8);
        BufferedReader bf = new BufferedReader(Reader);) {

      String temp = bf.readLine();

      while (temp != null) {

        texttoshow.append(temp);
        texttoshow.append(System.lineSeparator());
        temp = bf.readLine();

      }

      TextAreaUI.setText(texttoshow.toString());

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

    if (UserInputFile.canWrite()) {
      // ไฟล์สามารถเปิดอ่านและเขียนได้

      isreadOnly = false;

    } else {
      // ไฟล์เปิดอ่านได้แต่เขียนไม่ได้
      TextAreaUI.setEditable(false);
      Alert alert = new Alert(AlertType.INFORMATION);
      Stage alertStage = new Stage();
      alertStage = (Stage) alert.getDialogPane().getScene().getWindow();
      alertStage.getIcons().add(new Image(getClass().getResourceAsStream("Picture/Information.png")));
      alert.setHeaderText(null);
      alert.setTitle("Information");
      alert.setContentText("This file is read only and it won't be able to saved.");
      alert.show();

      isreadOnly = true;
    }

    Main.setSavestage(false);
    ChangeTitle();
    // เมื่อ user กด open อีกรอบจะเป็น directory ที่อยู่ล่าสุด
    Chooser.setInitialDirectory(new File(UserInputFile.getParent()));

  }

  public short SaveFile() {
    // return 0 เมื่อ save สำเร็จ 1 เมื่อ Fail
    if (isreadOnly) {
      return 1;
    }
    if (UserInputFile == null) {
      // ไม่ได้เปิดไฟล์เด้งไป save as
      return SaveAsFile();
    } else {
      // ถ้าเปิดไฟล์อยู่ก็ให้เซฟทับ

      try (FileWriter fileWriter = new FileWriter(UserInputFile, StandardCharsets.UTF_8)) {

        fileWriter.write(TextAreaUI.getText());
        Main.setSavestage(false);
        return 0;

      } catch (Exception e) {
        return 1;
      }

    }

  }

  public short SaveAsFile() {
    // return 0 เมื่อ save สำเร็จ 1 เมื่อ Fail
    if (isreadOnly) {
      return 1;
    }
    File file = Chooser.showSaveDialog(PrimaryStage);

    if (file == null) {
      return 1;
    }
    if (!file.exists()) {
      try {
        file.createNewFile();
      } catch (IOException e) {
        e.printStackTrace();
      }
    }
    try (FileWriter fileWriter = new FileWriter(file, StandardCharsets.UTF_8);) {
      fileWriter.write(TextAreaUI.getText());
      UserInputFile = new File(file.getPath());
      Main.setSavestage(false);
    } catch (Exception e) {
      e.printStackTrace();
    }
    ChangeTitle();
    Chooser.setInitialDirectory(new File(UserInputFile.getParent()));
    return 0;

  }

  public boolean IsBinaryFile(File F) throws IOException {

    String type = Files.probeContentType(F.toPath());
    String Filename = F.getName();
    if (Filename.endsWith(".cpp") || Filename.endsWith(".asm")
        || Filename.endsWith(".rb") || Filename.endsWith(".kt")
        || Filename.endsWith(".rs") || Filename.endsWith(".cs")
        || Filename.endsWith(".go")) {
      return false;
    } else if (type == null) {
      // เป็นอะไรก็ไม่รู้ถือว่าเป็น Birnary เลย
      return true;
    } else if (type.startsWith("text")) {
      return false;
    } else {
      // ไม่ใช่ไฟล์ text
      return true;
    }

  }

  public void ChangeTitle() {
    if (UserInputFile != null) {
      PrimaryStage.setTitle(UserInputFile.getName() + " - Notepad--");
    }
  }

  public boolean IsFileEditable() {
    return isreadOnly;
  }

  public void setboolReadOnly(boolean in) {
    isreadOnly = in;
    TextAreaUI.setEditable(!in);
  }

}