git.sur5r.net Git - bacula/bacula/blob - bacula/src/qt-console/restore/restore.cpp

   1 /*
   2    Bacula® - The Network Backup Solution
   3
   4    Copyright (C) 2007-2007 Free Software Foundation Europe e.V.
   5
   6    The main author of Bacula is Kern Sibbald, with contributions from
   7    many others, a complete list can be found in the file AUTHORS.
   8    This program is Free Software; you can redistribute it and/or
   9    modify it under the terms of version two of the GNU General Public
  10    License as published by the Free Software Foundation plus additions
  11    that are listed in the file LICENSE.
  12
  13    This program is distributed in the hope that it will be useful, but
  14    WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16    General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  21    02110-1301, USA.
  22
  23    Bacula® is a registered trademark of John Walker.
  24    The licensor of Bacula is the Free Software Foundation Europe
  25    (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
  26    Switzerland, email:ftf@fsfeurope.org.
  27 */
  28
  29 /*
  30  *   Version $Id$
  31  *
  32  *  Restore Class
  33  *
  34  *   Kern Sibbald, February MMVII
  35  *
  36  */
  37
  38 #include "bat.h"
  39 #include "restore.h"
  40
  41 restoreDialog::restoreDialog(Console *console)
  42 {
  43    m_console = console;
  44
  45    m_console->setEnabled(false);
  46    setupUi(this);
  47    connect(fileWidget, SIGNAL(itemDoubleClicked(QTreeWidgetItem*, int)),
  48            this, SLOT(fileDoubleClicked(QTreeWidgetItem *, int)));
  49    setFont(m_console->get_font());
  50    m_console->displayToPrompt();
  51    fillDirectory();
  52    this->show();
  53 }
  54
  55 /*
  56  * Fill the fileWidget box with the contents of the current directory
  57  */
  58 void restoreDialog::fillDirectory()
  59 {
  60    char cd_cmd[MAXSTRING];
  61    char modes[20], user[20], group[20], size[20], date[30];
  62    char marked[10];
  63    int pnl, fnl;
  64    POOLMEM *file = get_pool_memory(PM_FNAME);
  65    POOLMEM *path = get_pool_memory(PM_FNAME);
  66    QStringList titles;
  67
  68    titles << "Mark" << "File" << "Mode" << "User" << "Group" << "Size" << "Date";
  69    fileWidget->setHeaderLabels(titles);
  70
  71    char *dir = get_cwd();
  72    bsnprintf(cd_cmd, sizeof(cd_cmd), "cd \"%s\"\n", dir);
  73    Dmsg2(100, "dir=%s cmd=%s\n", dir, cd_cmd);
  74    m_console->write_dir(cd_cmd);
  75    m_console->discardToPrompt();
  76
  77    m_console->write_dir("dir");
  78    QList<QTreeWidgetItem *> items;
  79    QStringList item;
  80    while (m_console->read() > 0) {
  81       char *p = m_console->msg();
  82       char *l;
  83       strip_trailing_junk(p);
  84       if (*p == '$' || !*p) {
  85          continue;
  86       }
  87       l = p;
  88       skip_nonspaces(&p);             /* permissions */
  89       *p++ = 0;
  90       bstrncpy(modes, l, sizeof(modes));
  91       skip_spaces(&p);
  92       skip_nonspaces(&p);             /* link count */
  93       *p++ = 0;
  94       skip_spaces(&p);
  95       l = p;
  96       skip_nonspaces(&p);             /* user */
  97       *p++ = 0;
  98       skip_spaces(&p);
  99       bstrncpy(user, l, sizeof(user));
 100       l = p;
 101       skip_nonspaces(&p);             /* group */
 102       *p++ = 0;
 103       bstrncpy(group, l, sizeof(group));
 104       skip_spaces(&p);
 105       l = p;
 106       skip_nonspaces(&p);             /* size */
 107       *p++ = 0;
 108       bstrncpy(size, l, sizeof(size));
 109       skip_spaces(&p);
 110       l = p;
 111       skip_nonspaces(&p);             /* date/time */
 112       skip_spaces(&p);
 113       skip_nonspaces(&p);
 114       *p++ = 0;
 115       bstrncpy(date, l, sizeof(date));
 116       skip_spaces(&p);
 117       if (*p == '*') {
 118          bstrncpy(marked, "*", sizeof(marked));
 119          p++;
 120       } else {
 121          bstrncpy(marked, " ", sizeof(marked));
 122       }
 123       split_path_and_filename(p, &path, &pnl, &file, &fnl);
 124       item.clear();
 125       item << "" << file << modes << user << group << size << date;
 126       QTreeWidgetItem *ti = new QTreeWidgetItem((QTreeWidget *)0, item);
 127       ti->setTextAlignment(5, Qt::AlignRight); /* right align size */
 128       items.append(ti);
 129    }
 130    fileWidget->clear();
 131    fileWidget->insertTopLevelItems(0, items);
 132
 133    free_pool_memory(file);
 134    free_pool_memory(path);
 135 }
 136
 137 void restoreDialog::accept()
 138 {
 139    this->hide();
 140    m_console->write("done");
 141    delete this;
 142    m_console->setEnabled(true);
 143    mainWin->resetFocus();
 144 }
 145
 146
 147 void restoreDialog::reject()
 148 {
 149    this->hide();
 150    m_console->write("quit");
 151    mainWin->set_status("Canceled");
 152    delete this;
 153    m_console->setEnabled(true);
 154    mainWin->resetFocus();
 155 }
 156
 157 void restoreDialog::fileDoubleClicked(QTreeWidgetItem *item, int column)
 158 {
 159    char cmd[1000];
 160 //   printf("cwd=%s Text=%s column=%d\n", m_cwd.toUtf8().data(),
 161 //          item->text(1).toUtf8().data(), column);
 162    if (column == 0) {                 /* mark/unmark */
 163       if (item->text(0) == "*") {
 164          bsnprintf(cmd, sizeof(cmd), "unmark \"%s\"\n", item->text(1).toUtf8().data());
 165          item->setText(0, " ");
 166       } else {
 167          bsnprintf(cmd, sizeof(cmd), "mark \"%s\"\n", item->text(1).toUtf8().data());
 168          item->setText(0, "*");
 169       }
 170       m_console->write(cmd);
 171 //    printf("cmd=%s", cmd);
 172       m_console->displayToPrompt();
 173       return;
 174    }
 175 }
 176
 177 /*
 178  * Return cwd when in tree restore mode
 179  */
 180 char *restoreDialog::get_cwd()
 181 {
 182    int stat;
 183    m_console->write_dir(".pwd");
 184    Dmsg0(100, "send: .pwd\n");
 185    if ((stat = m_console->read()) > 0) {
 186       m_cwd = m_console->msg();
 187       Dmsg2(100, "cwd=%s msg=%s\n", m_cwd.toUtf8().data(), m_console->msg());
 188    } else {
 189       Dmsg1(000, "stat=%d\n", stat);
 190    }
 191    m_console->displayToPrompt();
 192    return m_cwd.toUtf8().data();
 193 }