sinksh: support for printing tables

1 files changed, 38 insertions, 0 deletions

diff --git a/sinksh/state.cpp b/sinksh/state.cpp
index 7fd3959..6fce21f 100644
--- a/sinksh/state.cpp
+++ b/sinksh/state.cpp
@@ -48,6 +48,7 @@ public:
     QEventLoop *event = 0;
     bool timing = false;
     QTextStream outStream;
+    QList<QStringList> table;
 };
 
 State::State() : d(new Private)
@@ -75,6 +76,43 @@ void State::printError(const QString &errorMessage, const QString &errorCode) co
     printLine("ERROR" + (errorCode.isEmpty() ? "" : " " + errorCode) + ": " + errorMessage);
 }
 
+void State::stageTableLine(const QStringList &line) const
+{
+    d->table << line;
+}
+
+void State::printTable(const QList<QStringList> &table) const
+{
+    //First let's find out the maximum size for each column depending on the content
+    QVector<int> columnSizes;
+    columnSizes.fill(0, 10);
+    for (const auto &row : table) {
+        for (int columnIndex = 0; columnIndex < row.size(); columnIndex++) {
+            if (columnSizes.size() <= columnIndex) {
+                columnSizes.append(0);
+            }
+            columnSizes[columnIndex] = qMax(columnSizes[columnIndex], row[columnIndex].size());
+        }
+    }
+    //And now print the table
+    for (const auto &row : table) {
+        for (int columnIndex = 0; columnIndex < row.size(); columnIndex++) {
+            if (columnIndex > 0) {
+                d->outStream << " | ";
+            }
+            d->outStream << row[columnIndex].leftJustified(columnSizes[columnIndex], ' ', true);
+        }
+        d->outStream << "\n";
+    }
+    d->outStream.flush();
+}
+
+void State::flushTable() const
+{
+    printTable(d->table);
+    d->table.clear();
+}
+
 void State::setDebugLevel(unsigned int level)
 {
     if (level < 7) {