]>
Commit | Line | Data |
---|---|---|
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 | /*}}}*/ |