3 files changed, 279 insertions, 29 deletions
diff --git a/works/life/cpp-practicum/Record.cpp b/works/life/cpp-practicum/Record.cpp
index 4d7d4a9..bdee28b 100644
--- a/works/life/cpp-practicum/Record.cpp
+++ b/works/life/cpp-practicum/Record.cpp
@@ -1,10 +1,8 @@
 #include "Record.hpp"
 
-void Record::WriteTo(QFile file) {
-  file.open(QFile::ReadWrite | QFile::Text | QFile::Truncate);
-  QTextStream stream(&file);
-  stream.setCodec("UTF-8");
+#include <algorithm>
 
+void Record::WriteTo(QTextStream &stream) {
   stream << books_.size() << ' ' << vendors_.size() << '\n';
   for (const auto &book : books_) {
     stream << book << '\n';
@@ -14,11 +12,7 @@ void Record::WriteTo(QFile file) {
   }
 }
 
-void Record::ReadFrom(QFile file) {
-  file.open(QFile::ReadOnly | QFile::Text);
-  QTextStream stream(&file);
-  stream.setCodec("UTF-8");
-
+void Record::ReadFrom(QTextStream &stream) {
   books_.clear();
   vendors_.clear();
 
@@ -180,3 +174,170 @@ bool BookModel::removeRows(int row, int count, const QModelIndex &parent) {
   endRemoveRows();
   return true;
 }
+
+void BookModel::sort(int column, Qt::SortOrder order) {
+  if (column == 0) {
+    beginResetModel();
+    if (order == Qt::AscendingOrder) {
+      std::sort(record_->GetBooks().begin(), record_->GetBooks().end(),
+                [](const Book &left, const Book &right) {
+                  return left.GetIsbn() < right.GetIsbn();
+                });
+    } else {
+      std::sort(record_->GetBooks().begin(), record_->GetBooks().end(),
+                [](const Book &left, const Book &right) {
+                  return left.GetIsbn() > right.GetIsbn();
+                });
+    }
+    endResetModel();
+  } else if (column == 1) {
+    beginResetModel();
+    if (order == Qt::AscendingOrder) {
+      std::sort(record_->GetBooks().begin(), record_->GetBooks().end(),
+                [](const Book &left, const Book &right) {
+                  return left.GetTitle() < right.GetTitle();
+                });
+    } else {
+      std::sort(record_->GetBooks().begin(), record_->GetBooks().end(),
+                [](const Book &left, const Book &right) {
+                  return left.GetTitle() > right.GetTitle();
+                });
+    }
+    endResetModel();
+  }
+}
+
+int VendorModel::rowCount(const QModelIndex &parent) const {
+  if (parent.isValid())
+    return 0;
+  return static_cast<int>(record_->GetVendors().size());
+}
+
+int VendorModel::columnCount(const QModelIndex &parent) const {
+  if (parent.isValid())
+    return 0;
+  return 5;
+}
+
+QVariant VendorModel::headerData(int section, Qt::Orientation orientation,
+                                 int role) const {
+  if (role != Qt::DisplayRole)
+    return QVariant();
+  if (orientation == Qt::Horizontal) {
+    switch (section) {
+    case 0:
+      return QStringLiteral("编号");
+    case 1:
+      return QStringLiteral("名称");
+    case 2:
+      return QStringLiteral("类型");
+    case 3:
+      return QStringLiteral("地址");
+    case 4:
+      return QStringLiteral("电话");
+    default:
+      return QVariant();
+    }
+  }
+  return QVariant();
+}
+
+QVariant VendorModel::data(const QModelIndex &index, int role) const {
+  if (role != Qt::DisplayRole)
+    return QVariant();
+
+  if (!index.isValid())
+    return QVariant();
+
+  if (index.row() >= static_cast<int>(record_->GetVendors().size()) ||
+      index.row() < 0)
+    return QVariant();
+
+  int row = index.row();
+  const Vendor &vendor = record_->GetVendors()[row];
+
+  int col = index.column();
+  switch (col) {
+  case 0:
+    return vendor.GetId();
+  case 1:
+    return QString::fromStdU16String(vendor.GetName());
+  case 2:
+    return QString::fromStdU16String(vendor.GetType());
+  case 3:
+    return QString::fromStdU16String(vendor.GetAddress());
+  case 4:
+    return QString::fromStdU16String(vendor.GetPhone());
+  default:
+    return QVariant();
+  }
+}
+
+bool VendorModel::setData(const QModelIndex &index, const QVariant &value,
+                          int role) {
+  if (index.isValid() && role == Qt::EditRole) {
+    int row = index.row();
+    Vendor &vendor = record_->GetVendors()[row];
+
+    int col = index.column();
+    switch (col) {
+    case 0:
+      if (!value.canConvert<int>())
+        return false;
+      vendor.SetId(value.toInt());
+      return true;
+    case 1:
+      if (!value.canConvert<QString>())
+        return false;
+      vendor.SetName(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    case 2:
+      if (!value.canConvert<QString>())
+        return false;
+      vendor.SetType(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    case 3:
+      if (!value.canConvert<QString>())
+        return false;
+      vendor.SetAddress(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    case 4:
+      if (!value.canConvert<QString>())
+        return false;
+      vendor.SetPhone(value.toString().toStdU16String());
+      emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+      return true;
+    default:
+      return false;
+    }
+  }
+  return false;
+}
+
+Qt::ItemFlags VendorModel::flags(const QModelIndex &index) const {
+  if (!index.isValid())
+    return Qt::ItemIsEnabled;
+
+  return QAbstractTableModel::flags(index) | Qt::ItemIsEditable;
+}
+
+bool VendorModel::insertRows(int row, int count, const QModelIndex &parent) {
+  beginInsertRows(parent, row, row + count - 1);
+  for (int i = 0; i < count; i++) {
+    record_->GetVendors().insert(record_->GetVendors().cbegin() + row,
+                                 Vendor());
+  }
+  endInsertRows();
+  return true;
+}
+
+bool VendorModel::removeRows(int row, int count, const QModelIndex &parent) {
+  beginRemoveRows(parent, row, row + count - 1);
+  record_->GetVendors().erase(record_->GetVendors().cbegin() + row,
+                              record_->GetVendors().cbegin() + row + count);
+  endRemoveRows();
+  return true;
+}
diff --git a/works/life/cpp-practicum/Record.hpp b/works/life/cpp-practicum/Record.hpp
index 78b70ba..c379f04 100644
--- a/works/life/cpp-practicum/Record.hpp
+++ b/works/life/cpp-practicum/Record.hpp
@@ -5,7 +5,7 @@
 #include "Vendor.hpp"
 
 #include <QAbstractTableModel>
-#include <QFile>
+#include <QTextStream>
 #include <optional>
 #include <vector>
 
@@ -19,8 +19,8 @@ public:
   ~Record() = default;
 
 public:
-  void WriteTo(QFile file);
-  void ReadFrom(QFile file);
+  void WriteTo(QTextStream &stream);
+  void ReadFrom(QTextStream &stream);
 
   std::vector<Book> &GetBooks() { return books_; }
   std::vector<Vendor> &GetVendors() { return vendors_; }
@@ -47,7 +47,30 @@ public:
                   const QModelIndex &parent = QModelIndex()) override;
   bool removeRows(int row, int count,
                   const QModelIndex &parent = QModelIndex()) override;
+  void sort(int column, Qt::SortOrder order = Qt::AscendingOrder) override;
 
 private:
   Record *record_;
-};
-\ No newline at end of file
+};
+
+class VendorModel : public QAbstractTableModel {
+public:
+  explicit VendorModel(Record *record) : record_(record) {}
+
+  int rowCount(const QModelIndex &parent = QModelIndex()) const override;
+  int columnCount(const QModelIndex &parent = QModelIndex()) const override;
+  QVariant headerData(int section, Qt::Orientation orientation,
+                      int role = Qt::DisplayRole) const override;
+  QVariant data(const QModelIndex &index,
+                int role = Qt::DisplayRole) const override;
+  bool setData(const QModelIndex &index, const QVariant &value,
+               int role = Qt::EditRole) override;
+  Qt::ItemFlags flags(const QModelIndex &index) const override;
+  bool insertRows(int row, int count,
+                  const QModelIndex &parent = QModelIndex()) override;
+  bool removeRows(int row, int count,
+                  const QModelIndex &parent = QModelIndex()) override;
+
+private:
+  Record *record_;
+};
diff --git a/works/life/cpp-practicum/main.cpp b/works/life/cpp-practicum/main.cpp
index 9ecc308..399c82c 100644
--- a/works/life/cpp-practicum/main.cpp
+++ b/works/life/cpp-practicum/main.cpp
@@ -1,6 +1,8 @@
 #include "Record.hpp"
 
 #include <QApplication>
+#include <QDir>
+#include <QFile>
 #include <QHBoxLayout>
 #include <QHeaderView>
 #include <QMainWindow>
@@ -8,49 +10,113 @@
 #include <QTableView>
 #include <QVBoxLayout>
 #include <QWidget>
+#include <memory>
 
 int main(int argc, char *argv[]) {
   QApplication application(argc, argv);
 
+  Record record;
+
+  QDir app_dir(application.applicationDirPath());
+  QFile data_file = app_dir.filePath("data.txt");
+  std::unique_ptr<QTextStream> stream;
+  if (data_file.exists()) {
+    data_file.open(QFile::ReadWrite);
+    stream.reset(new QTextStream(&data_file));
+    stream->setCodec("UTF-8");
+    try {
+      record.ReadFrom(*stream);
+    } catch (...) {
+    }
+  } else {
+    data_file.open(QFile::ReadWrite);
+    stream.reset(new QTextStream(&data_file));
+    stream->setCodec("UTF-8");
+  }
+
   QWidget window;
   QVBoxLayout window_layout;
   window.setLayout(&window_layout);
 
-  QHBoxLayout top_bar;
-  window_layout.addLayout(&top_bar);
+  QHBoxLayout center_area;
+  window_layout.addLayout(&center_area);
 
-  QPushButton import_button(QStringLiteral("导入"));
-  QPushButton export_button(QStringLiteral("导出"));
-  QPushButton add_book_button(QStringLiteral("添加书"));
+  QVBoxLayout book_area;
+  QVBoxLayout vendor_area;
+  center_area.addLayout(&book_area);
+  center_area.addLayout(&vendor_area);
 
-  top_bar.addWidget(&import_button);
-  top_bar.addWidget(&export_button);
-  top_bar.addStretch(1);
-  top_bar.addWidget(&add_book_button);
+  QHBoxLayout book_top_area;
+  QHBoxLayout vendor_top_area;
+  book_area.addLayout(&book_top_area);
+  vendor_area.addLayout(&vendor_top_area);
 
-  QHBoxLayout center_area;
-  window_layout.addLayout(&center_area);
+  QPushButton book_add_button(QStringLiteral("添加"));
+  QPushButton book_remove_button(QStringLiteral("删除"));
+
+  book_top_area.addStretch(1);
+  book_top_area.addWidget(&book_add_button);
+  book_top_area.addWidget(&book_remove_button);
+
+  QPushButton vendor_add_button(QStringLiteral("添加"));
+  QPushButton vendor_remove_button(QStringLiteral("删除"));
+
+  vendor_top_area.addStretch(1);
+  vendor_top_area.addWidget(&vendor_add_button);
+  vendor_top_area.addWidget(&vendor_remove_button);
 
   QTableView book_view;
   book_view.verticalHeader()->hide();
   book_view.setSelectionBehavior(QAbstractItemView::SelectRows);
   book_view.setEditTriggers(QAbstractItemView::DoubleClicked);
   book_view.setSelectionMode(QAbstractItemView::SingleSelection);
+  book_view.setSortingEnabled(true);
+
   QTableView vendor_view;
+  vendor_view.verticalHeader()->hide();
+  vendor_view.setSelectionBehavior(QAbstractItemView::SelectRows);
+  vendor_view.setEditTriggers(QAbstractItemView::DoubleClicked);
+  vendor_view.setSelectionMode(QAbstractItemView::SingleSelection);
 
-  center_area.addWidget(&book_view);
-  center_area.addWidget(&vendor_view);
+  book_area.addWidget(&book_view, 1);
+  vendor_area.addWidget(&vendor_view, 1);
 
-  Record record;
   BookModel book_model(&record);
+  VendorModel vendor_model(&record);
 
   book_view.setModel(&book_model);
+  vendor_view.setModel(&vendor_model);
 
-  QObject::connect(&add_book_button, &QPushButton::clicked, [&book_model]() {
+  QObject::connect(&book_add_button, &QPushButton::clicked, [&book_model]() {
     book_model.insertRow(book_model.rowCount());
   });
 
+  QObject::connect(
+      &book_remove_button, &QPushButton::clicked, [&book_view, &book_model]() {
+        auto selected_rows = book_view.selectionModel()->selectedRows();
+        for (const auto &row : selected_rows) {
+          book_model.removeRow(row.row());
+        }
+      });
+
+  QObject::connect(
+      &vendor_add_button, &QPushButton::clicked,
+      [&vendor_model]() { vendor_model.insertRow(vendor_model.rowCount()); });
+
+  QObject::connect(&vendor_remove_button, &QPushButton::clicked,
+                   [&vendor_view, &vendor_model]() {
+                     auto selected_rows =
+                         vendor_view.selectionModel()->selectedRows();
+                     for (const auto &row : selected_rows) {
+                       vendor_model.removeRow(row.row());
+                     }
+                   });
+
   window.show();
 
-  return application.exec();
+  int result = application.exec();
+
+  record.WriteTo(*stream);
+
+  return result;
 }