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