new tsv

[task-reporter.git] / src / main / scala / jp / ymir / taskReporter / core / ReportSet.scala

diff --git a/src/main/scala/jp/ymir/taskReporter/core/ReportSet.scala b/src/main/scala/jp/ymir/taskReporter/core/ReportSet.scala
index f4392b76e41d7c3993d0e91ef664b1b073fbe4b9..66a5d6d08fdf90a79991b8a33adaff6ba2094eae 100644 (file)
--- a/src/main/scala/jp/ymir/taskReporter/core/ReportSet.scala
+++ b/src/main/scala/jp/ymir/taskReporter/core/ReportSet.scala
@@ -1,14 +1,90 @@
 package jp.ymir.taskReporter.core
-import java.io._;
+import java.io._
+import java.text.SimpleDateFormat
+import java.util.Calendar
+import javax.swing.table.AbstractTableModel
+import jp.ymir.taskReporter.core._
+import scala.collection.immutable._
+import scala.io._
 
-class ReportSet(private var _file: Option[File]) {
-  private var _dirty : Boolean = false
+class ReportSet(private var _file: Option[File])
+    extends AbstractTableModel {
 
-  def file: Option[File] = {
-    return _file
+  private var _reports : SortedMap[Calendar, Report] = TreeMap()
+  private var _dirty = false
+
+  _file.foreach { file => load(file) }
+
+  def load(file: File) {
+    import Task._
+
+    var reports = _reports.empty
+    val src     = Source.fromFile(file, "UTF-8")
+
+    TSV.decode(src.mkString).foreach { task =>
+      if (reports.isDefinedAt(task.date)) {
+        reports(task.date) += task
+      }
+      else {
+        val report = new Report(task.date)
+        report  += task
+        reports += (report.date -> report)
+      }
+    }
+
+/*
+    for (line <- src.getLines) {
+      if (!line.isEmpty) {
+        val task = new Task(line)
+        if (reports.isDefinedAt(task.date)) {
+          reports(task.date) += task
+        }
+        else {
+          val report = new Report(task.date)
+          report  += task
+          reports += (report.date -> report)
+        }
+      }
+    }
+ */
+
+    _file    = Some(file)
+    _reports = reports
+    _dirty   = false
+    fireTableDataChanged
+  }
+
+  def file  = _file
+  def dirty = _dirty
+
+  def apply(n: Int) : Report =
+    // THINKME: Any better way than this?
+    _reports.values.toIndexedSeq(n)
+
+  def save(file: File) {
+    _file = Some(file); save
+  }
+  def save {
+    // FIXME
+    _dirty = false
+  }
+
+  /* Method definitions for AbstractTableModel
+   */
+  def getColumnCount = 2
+  def getRowCount    = _reports.size
+
+  override def getColumnName(column: Int) : String = {
+    column match {
+      case 0 => "Date"
+      case 1 => "# Tasks"
+    }
   }
 
-  def dirty: Boolean = {
-    return _dirty
+  def getValueAt(row: Int, column: Int) : Object = {
+    column match {
+      case 0 => this(row).dateString
+      case 1 => this(row).size : Integer
+    }
   }
 }