cmdline/apt-cdrom.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: apt-cdrom.cc,v 1.45 2003/11/19 23:50:51 mdz Exp $
   4 /* ######################################################################
   5
   6    APT CDROM - Tool for handling APT's CDROM database.
   7
   8    Currently the only option is 'add' which will take the current CD
   9    in the drive and add it into the database.
  10
  11    ##################################################################### */
  12                                                                         /*}}}*/
  13 // Include Files                                                        /*{{{*/
  14 #include <apt-pkg/cmndline.h>
  15 #include <apt-pkg/error.h>
  16 #include <apt-pkg/init.h>
  17 #include <apt-pkg/fileutl.h>
  18 #include <apt-pkg/progress.h>
  19 #include <apt-pkg/cdromutl.h>
  20 #include <apt-pkg/strutl.h>
  21 #include <apt-pkg/acquire.h>
  22 #include <apt-pkg/acquire-item.h>
  23 #include <apt-pkg/cdrom.h>
  24 #include <config.h>
  25 #include <apti18n.h>
  26
  27 //#include "indexcopy.h"
  28
  29 #include <locale.h>
  30 #include <iostream>
  31 #include <fstream>
  32 #include <vector>
  33 #include <algorithm>
  34 #include <sys/stat.h>
  35 #include <fcntl.h>
  36 #include <dirent.h>
  37 #include <unistd.h>
  38 #include <stdio.h>
  39                                                                         /*}}}*/
  40
  41 using namespace std;
  42
  43 class pkgCdromTextStatus : public pkgCdromStatus                        /*{{{*/
  44 {
  45 protected:
  46    OpTextProgress Progress;
  47    void Prompt(const char *Text);
  48    string PromptLine(const char *Text);
  49    bool AskCdromName(string &name);
  50
  51 public:
  52    virtual void Update(string text, int current);
  53    virtual bool ChangeCdrom();
  54    virtual OpProgress* GetOpProgress();
  55 };
  56
  57 void pkgCdromTextStatus::Prompt(const char *Text)
  58 {
  59    char C;
  60    cout << Text << ' ' << flush;
  61    read(STDIN_FILENO,&C,1);
  62    if (C != '\n')
  63       cout << endl;
  64 }
  65
  66 string pkgCdromTextStatus::PromptLine(const char *Text)
  67 {
  68    cout << Text << ':' << endl;
  69
  70    string Res;
  71    getline(cin,Res);
  72    return Res;
  73 }
  74
  75 bool pkgCdromTextStatus::AskCdromName(string &name)
  76 {
  77    cout << _("Please provide a name for this Disc, such as 'Debian 2.1r1 Disk 1'") << flush;
  78    name = PromptLine("");
  79
  80    return true;
  81 }
  82
  83
  84 void pkgCdromTextStatus::Update(string text, int current)
  85 {
  86    if(text.size() > 0)
  87       cout << text << flush;
  88 }
  89
  90 bool pkgCdromTextStatus::ChangeCdrom()
  91 {
  92    Prompt(_("Please insert a Disc in the drive and press enter"));
  93    return true;
  94 }
  95
  96 OpProgress* pkgCdromTextStatus::GetOpProgress()
  97 {
  98    return &Progress;
  99 };
 100                                                                         /*}}}*/
 101 // DoAdd - Add a new CDROM                                              /*{{{*/
 102 // ---------------------------------------------------------------------
 103 /* This does the main add bit.. We show some status and things. The
 104    sequence is to mount/umount the CD, Ident it then scan it for package
 105    files and reduce that list. Then we copy over the package files and
 106    verify them. Then rewrite the database files */
 107 bool DoAdd(CommandLine &)
 108 {
 109    bool res = false;
 110    pkgCdromTextStatus log;
 111    pkgCdrom cdrom;
 112    res = cdrom.Add(&log);
 113    if(res)
 114       cout << _("Repeat this process for the rest of the CDs in your set.") << endl;
 115    return res;
 116 }
 117                                                                         /*}}}*/
 118 // DoIdent - Ident a CDROM                                              /*{{{*/
 119 // ---------------------------------------------------------------------
 120 /* */
 121 bool DoIdent(CommandLine &)
 122 {
 123    string ident;
 124    pkgCdromTextStatus log;
 125    pkgCdrom cdrom;
 126    return cdrom.Ident(ident, &log);
 127 }
 128                                                                         /*}}}*/
 129 // ShowHelp - Show the help screen                                      /*{{{*/
 130 // ---------------------------------------------------------------------
 131 /* */
 132 int ShowHelp()
 133 {
 134    ioprintf(cout,_("%s %s for %s compiled on %s %s\n"),PACKAGE,VERSION,
 135             COMMON_ARCH,__DATE__,__TIME__);
 136    if (_config->FindB("version") == true)
 137       return 0;
 138
 139    cout <<
 140       "Usage: apt-cdrom [options] command\n"
 141       "\n"
 142       "apt-cdrom is a tool to add CDROM's to APT's source list. The\n"
 143       "CDROM mount point and device information is taken from apt.conf\n"
 144       "and /etc/fstab.\n"
 145       "\n"
 146       "Commands:\n"
 147       "   add - Add a CDROM\n"
 148       "   ident - Report the identity of a CDROM\n"
 149       "\n"
 150       "Options:\n"
 151       "  -h   This help text\n"
 152       "  -d   CD-ROM mount point\n"
 153       "  -r   Rename a recognized CD-ROM\n"
 154       "  -m   No mounting\n"
 155       "  -f   Fast mode, don't check package files\n"
 156       "  -a   Thorough scan mode\n"
 157       "  -c=? Read this configuration file\n"
 158       "  -o=? Set an arbitrary configuration option, eg -o dir::cache=/tmp\n"
 159       "See fstab(5)\n";
 160    return 0;
 161 }
 162                                                                         /*}}}*/
 163 int main(int argc,const char *argv[])                                   /*{{{*/
 164 {
 165    CommandLine::Args Args[] = {
 166       {'h',"help","help",0},
 167       {'v',"version","version",0},
 168       {'d',"cdrom","Acquire::cdrom::mount",CommandLine::HasArg},
 169       {'r',"rename","APT::CDROM::Rename",0},
 170       {'m',"no-mount","APT::CDROM::NoMount",0},
 171       {'f',"fast","APT::CDROM::Fast",0},
 172       {'n',"just-print","APT::CDROM::NoAct",0},
 173       {'n',"recon","APT::CDROM::NoAct",0},
 174       {'n',"no-act","APT::CDROM::NoAct",0},
 175       {'a',"thorough","APT::CDROM::Thorough",0},
 176       {'c',"config-file",0,CommandLine::ConfigFile},
 177       {'o',"option",0,CommandLine::ArbItem},
 178       {0,0,0,0}};
 179    CommandLine::Dispatch Cmds[] = {
 180       {"add",&DoAdd},
 181       {"ident",&DoIdent},
 182       {0,0}};
 183
 184    // Set up gettext support
 185    setlocale(LC_ALL,"");
 186    textdomain(PACKAGE);
 187
 188    // Parse the command line and initialize the package library
 189    CommandLine CmdL(Args,_config);
 190    if (pkgInitConfig(*_config) == false ||
 191        CmdL.Parse(argc,argv) == false ||
 192        pkgInitSystem(*_config,_system) == false)
 193    {
 194       _error->DumpErrors();
 195       return 100;
 196    }
 197
 198    // See if the help should be shown
 199    if (_config->FindB("help") == true || _config->FindB("version") == true ||
 200        CmdL.FileSize() == 0)
 201       return ShowHelp();
 202
 203    // Deal with stdout not being a tty
 204    if (isatty(STDOUT_FILENO) && _config->FindI("quiet",0) < 1)
 205       _config->Set("quiet","1");
 206
 207    // Match the operation
 208    CmdL.DispatchArg(Cmds);
 209
 210    // Print any errors or warnings found during parsing
 211    if (_error->empty() == false)
 212    {
 213       bool Errors = _error->PendingError();
 214       _error->DumpErrors();
 215       return Errors == true?100:0;
 216    }
 217
 218    return 0;
 219 }
 220                                                                         /*}}}*/