Chris@109: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@109: 
Chris@109: /*
Chris@109:     EasyMercurial
Chris@109: 
Chris@109:     Based on HgExplorer by Jari Korhonen
Chris@109:     Copyright (c) 2010 Jari Korhonen
Chris@244:     Copyright (c) 2011 Chris Cannam
Chris@244:     Copyright (c) 2011 Queen Mary, University of London
Chris@109: 
Chris@109:     This program is free software; you can redistribute it and/or
Chris@109:     modify it under the terms of the GNU General Public License as
Chris@109:     published by the Free Software Foundation; either version 2 of the
Chris@109:     License, or (at your option) any later version.  See the file
Chris@109:     COPYING included with this distribution for more information.
Chris@109: */
Chris@109: 
Chris@92: #include "filestates.h"
Chris@74: 
Chris@86: #include "debug.h"
Chris@86: 
Chris@74: #include <QMap>
Chris@74: 
Chris@94: FileStates::FileStates()
Chris@94: {
Chris@94: }
Chris@94: 
Chris@94: void FileStates::clearBuckets()
Chris@94: {
Chris@199:     m_clean.clear();
Chris@94:     m_modified.clear();
Chris@94:     m_added.clear();
Chris@94:     m_removed.clear();
Chris@94:     m_missing.clear();
Chris@159:     m_inConflict.clear();
Chris@94:     m_unknown.clear();
Chris@199:     m_ignored.clear();
Chris@94: }
Chris@94: 
Chris@94: FileStates::State FileStates::charToState(QChar c, bool *ok)
Chris@94: {
Chris@159:     // Note that InConflict does not correspond to a stat char -- it's
Chris@163:     // reported separately, by resolve --list, which shows U for
Chris@163:     // Unresolved -- stat reports files in conflict as M, which means
Chris@163:     // they will appear in more than one bin if we handle them
Chris@163:     // naively.  'u' is also used by stat as the command option for
Chris@163:     // Unknown, but the stat output uses ? for these so there's no
Chris@163:     // ambiguity in parsing.
Chris@159: 
Chris@94:     if (ok) *ok = true;
Chris@94:     if (c == 'M') return Modified;
Chris@94:     if (c == 'A') return Added;
Chris@94:     if (c == 'R') return Removed;
Chris@94:     if (c == '!') return Missing;
Chris@163:     if (c == 'U') return InConflict;
Chris@94:     if (c == '?') return Unknown;
Chris@94:     if (c == 'C') return Clean;
Chris@199:     if (c == 'I') return Ignored;
Chris@94:     if (ok) *ok = false;
Chris@94:     return Unknown;
Chris@94: }
Chris@94: 
Chris@94: QStringList *FileStates::stateToBucket(State s)
Chris@94: {
Chris@94:     switch (s) {
Chris@199:     case Clean: return &m_clean; // not implemented yet
Chris@94:     case Modified: return &m_modified;
Chris@94:     case Added: return &m_added;
Chris@94:     case Unknown: return &m_unknown;
Chris@94:     case Removed: return &m_removed;
Chris@94:     case Missing: return &m_missing;
Chris@159:     case InConflict: return &m_inConflict;
Chris@199:     case Ignored: return &m_ignored;
Chris@205: 
Chris@205:     default: return &m_clean;
Chris@94:     }
Chris@94: }
Chris@94: 
Chris@94: void FileStates::parseStates(QString text)
Chris@74: {
Chris@74:     text.replace("\r\n", "\n");
Chris@74: 
Chris@94:     clearBuckets();
Chris@163:     m_stateMap.clear();
Chris@74: 
Chris@74:     QStringList lines = text.split("\n", QString::SkipEmptyParts);
Chris@94: 
Chris@74:     foreach (QString line, lines) {
Chris@94: 
Chris@86:         if (line.length() < 3 || line[1] != ' ') {
Chris@86:             continue;
Chris@86:         }
Chris@94: 
Chris@94:         QChar c = line[0];
Chris@94:         bool ok = false;
Chris@94:         State s = charToState(c, &ok);
Chris@94:         if (!ok) continue;
Chris@94: 
Chris@74:         QString file = line.right(line.length() - 2);
Chris@163:         m_stateMap[file] = s;
Chris@163:     }
Chris@163: 
Chris@163:     foreach (QString file, m_stateMap.keys()) {
Chris@163:         QStringList *bucket = stateToBucket(m_stateMap[file]);
Chris@199:         if (bucket) bucket->push_back(file);
Chris@74:     }
Chris@86: 
Chris@199:     DEBUG << "FileStates: "
Chris@199:           << m_modified.size() << " modified, " << m_added.size()
Chris@159:           << " added, " << m_removed.size() << " removed, " << m_missing.size()
Chris@159:           << " missing, " << m_inConflict.size() << " in conflict, "
Chris@159:           << m_unknown.size() << " unknown" << endl;
Chris@74: }
Chris@92: 
Chris@94: QStringList FileStates::getFilesInState(State s) const
Chris@92: {
Chris@94:     QStringList *sl = const_cast<FileStates *>(this)->stateToBucket(s);
Chris@94:     if (sl) return *sl;
Chris@94:     else return QStringList();
Chris@92: }
Chris@92: 
Chris@94: FileStates::State FileStates::getStateOfFile(QString file) const
Chris@92: {
Chris@94:     if (m_stateMap.contains(file)) {
Chris@94:         return m_stateMap[file];
Chris@92:     }
Chris@92:     DEBUG << "FileStates: WARNING: getStateOfFile: file "
Chris@92:             << file << " is unknown to us: returning Unknown state, "
Chris@92:             << "but unknown to us is not supposed to be the same "
Chris@92:             << "thing as unknown state..."
Chris@92:             << endl;
Chris@92:     return Unknown;
Chris@92: }