1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// files.cpp
/**
* Copyright (C) 2008 10gen Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "stdafx.h"
#include "client/gridfs.h"
#include "client/dbclient.h"
#include "Tool.h"
#include <fstream>
#include <iostream>
#include <boost/program_options.hpp>
using namespace mongo;
namespace po = boost::program_options;
class Files : public Tool {
public:
Files() : Tool( "files" ){
add_options()
( "command" , po::value<string>() , "command (list|search|put|get)" )
( "file" , po::value<string>() , "filename for get|put" )
;
addPositionArg( "command" , 1 );
addPositionArg( "file" , 2 );
}
virtual void printExtraHelp( ostream & out ){
out << "\t list - list all files. takes an optional filename. the file has to start with the filename" << endl;
out << "\t search - search all files for something that contains the string" << endl;
}
void display( GridFS * grid , BSONObj obj ){
auto_ptr<DBClientCursor> c = grid->list( obj );
while ( c->more() ){
BSONObj obj = c->next();
cout
<< obj["filename"].str() << "\t"
<< (long)obj["length"].number()
<< endl;
}
}
int run(){
string cmd = getParam( "command" );
if ( cmd.size() == 0 ){
cerr << "need command" << endl;
return -1;
}
GridFS g( _conn , _db );
string filename = getParam( "file" );
if ( cmd == "list" ){
BSONObjBuilder b;
if ( filename.size() )
b.appendRegex( "filename" , ( (string)"^" + filename ).c_str() );
display( &g , b.obj() );
return 0;
}
if ( filename.size() == 0 ){
cerr << "need a filename" << endl;
return -1;
}
if ( cmd == "search" ){
BSONObjBuilder b;
b.appendRegex( "filename" , filename.c_str() );
display( &g , b.obj() );
return 0;
}
if ( cmd == "get" ){
GridFile f = g.findFile( filename );
if ( ! f.exists() ){
cerr << "file not found" << endl;
return -2;
}
string out = f.getFilename();
f.write( out );
cout << "done write to: " << out << endl;
return 0;
}
if ( cmd == "put" ){
g.storeFile( filename );
cout << "done!";
return 0;
}
cerr << "unknown command: " << cmd << endl;
return -1;
}
};
int main( int argc , char ** argv ) {
Files f;
return f.main( argc , argv );
}
|
