Added yp2::PlainFile class which reads Unix-style plain text files.

[metaproxy-moved-to-github.git] / src / plainfile.cpp

diff --git a/src/plainfile.cpp b/src/plainfile.cpp
new file mode 100644 (file)
index 0000000..07361aa
--- /dev/null
+++ b/src/plainfile.cpp
@@ -0,0 +1,110 @@
+/* $Id: plainfile.cpp,v 1.1 2006-01-25 11:27:19 adam Exp $
+   Copyright (c) 2005, Index Data.
+
+%LICENSE%
+ */
+
+#include "config.hpp"
+
+#include <cstdlib>
+#include <string>
+#include <iostream>
+#include <fstream>
+
+#include "util.hpp"
+
+#define PLAINFILE_MAX_LINE 256
+
+namespace yp2 {
+    class PlainFile::Rep {
+        friend class PlainFile;
+        Rep();
+        void close();
+        int lineno;
+        std::ifstream *fh;
+    };
+}
+
+yp2::PlainFile::Rep::Rep() : lineno(1)
+{
+    fh = 0;
+}
+
+yp2::PlainFile::PlainFile() : m_p(new Rep)
+{
+}
+
+void yp2::PlainFile::Rep::close()
+{
+    delete fh;
+    fh = 0;
+    lineno = 0;
+}
+
+yp2::PlainFile::~PlainFile()
+{
+    m_p->close();
+}
+
+bool yp2::PlainFile::open(const std::string &fname)
+{
+    m_p->close();
+
+    std::ifstream *new_file = new std::ifstream(fname.c_str());
+    if (! *new_file)
+    {
+        delete new_file;
+        return false;
+    }
+    m_p->fh = new_file;
+    return true;
+}
+
+bool yp2::PlainFile::getline(std::vector<std::string> &args)
+{
+    args.clear();
+
+    if (!m_p->fh)
+        return false;  // no file at all.
+
+    char line_cstr[PLAINFILE_MAX_LINE];
+    while (true)
+    {
+        if (m_p->fh->eof())
+        {
+            m_p->close();   // might as well close it now
+            return false;
+        }
+        
+        m_p->lineno++;
+        m_p->fh->getline(line_cstr, PLAINFILE_MAX_LINE-1);
+        char first = line_cstr[0];
+        if (first && !strchr("# \t", first))
+            break;
+        // comment or blank line.. read next.
+    }
+    const char *cp = line_cstr;
+    while (true)
+    {
+        // skip whitespace
+        while (*cp && strchr(" \t", *cp))
+            cp++;
+        if (*cp == '\0')
+            break;
+        const char *cp0 = cp;
+        while (*cp && !strchr(" \t", *cp))
+            cp++;
+        std::string arg(cp0, cp - cp0);
+        args.push_back(arg);
+    }
+    return true;
+}
+
+/*
+ * Local variables:
+ * c-basic-offset: 4
+ * indent-tabs-mode: nil
+ * c-file-style: "stroustrup"
+ * End:
+ * vim: shiftwidth=4 tabstop=8 expandtab
+ */