コード例 #1
2
ファイル: ftpEntry.cpp プロジェクト: segrax/KiLLARMY
// Recursive Function
void _ftpEntry::removeFolderChildren( size_t folderID ) {
    Connection                     *sqlConnect = dbConnect();
    bool							isDone						= false;
	Query							serverQuery					= sqlConnect->query();
	StoreQueryResult				serverResults;
	StoreQueryResult::iterator		storeQueryIT;
    size_t                          ID;

    // Find all folders that parent us
    serverQuery << "select * from ServerFolders where parentID = "      << quote << folderID;
    serverQuery.execute();

    // 
    for(storeQueryIT = serverResults.begin(); storeQueryIT != serverResults.end(); storeQueryIT++ ) {
        
        // Remove all children from this parent
        ID = (size_t) (*storeQueryIT)["folderID"];
        removeFolderChildren( ID );
        
        // Now remove the parent
        serverQuery.str("");
	    serverQuery << "delete from ServerFolders where folderID = "	<< quote << ID;
        serverQuery.execute();
    }

    delete sqlConnect;
}
コード例 #2
0
ファイル: stdafx.cpp プロジェクト: segrax/KiLLARMY
// Find users with <ip>
vector<size_t> ipFind( database *database, string IP_ ) {
    socketTCP		socket("0.0.0.0");
    string			domain, ip, ipTmp;
	vector<size_t>	members;
	size_t			pos;
    // Resolve domain/ip
    ip      = socket.hostnameLookup( IP_ );	
    domain  = socket.ipLookup( IP_ );	

	if(!ip.size()) {

		// Remove after *
		size_t	pos = IP_.find("*");
		if(pos != string::npos) {
			
			IP_ = IP_.substr(0, pos);
		}

		ip = IP_;
	}

	// Get a list of similar ips
	StoreQueryResult			IPs = database->ipFind( ip );
	StoreQueryResult::iterator	ipIT;
	Row							IPRow;

	if( !IPs.size() )
		IPs = database->ipFind( domain );

	if(IPs.size() ) {

		// Add each matching member to the vector
		for( ipIT = IPs.begin(); ipIT != IPs.end(); ipIT++ ) {
			IPRow = *ipIT;
			members.push_back( (size_t) IPRow["memberID"] );
		}
		return members;
	}

	// Now get all IPs, resolve them all and try and match

	IPs = database->ipsRead(0);
	for( ipIT = IPs.begin(); ipIT != IPs.end(); ipIT++ ) {
		IPRow = *ipIT;
		
		ipTmp = (string) IPRow["ip"];
		if( (pos = ipTmp.find("*")) != string::npos) {
			ipTmp = ipTmp.substr(0, pos-1);
			if( ip.find( ipTmp) != string::npos )
				members.push_back( (size_t) IPRow["memberID"] );
		}

		ipTmp     = socket.hostnameLookup( ipTmp );

		if( ip == ipTmp )
			members.push_back( (size_t) IPRow["memberID"] );
	}

	return members;
}
コード例 #3
0
ファイル: stdafx.cpp プロジェクト: segrax/KiLLARMY
bool ipValidate( database *database, string IP, size_t memberID ) {
    socketTCP    socket("0.0.0.0");
    string       domain;

    // Resolve domain/ip
    IP      = socket.hostnameLookup(IP);		// Current Connect Client IP
    domain  = socket.ipLookup( IP );			// Resolve the clients IP

    // Get any IP/Host matches for this member's ID / Connected IP/Domain
    StoreQueryResult    IPs        = database->ipsRead( memberID, IP );
    StoreQueryResult    Domains    = database->ipsRead( memberID, domain );

    // Did we find any?
    if(!IPs.empty())
        return true;

    if(!Domains.empty())
        return true;

	if(!IP.size())
		return false;

    // Nope, so we need to get all IPs/Domains for this user from the database
    // Remove all wildcards and check the domain for matches
    Domains = database->ipsRead( memberID );
 
    StoreQueryResult::iterator   addressIT;
    string                       testName;
    string                       resolvedIP;
    string::iterator             sIT;
    Row                          row;

    for( addressIT = Domains.begin(); addressIT != Domains.end(); addressIT++ ) {
        row = *addressIT;

        // Get an address to test
        testName = (string) row["ip"];   

        // Resolve the name from the database, and try it against the connected IP
        resolvedIP = socket.hostnameLookup( testName );
		
        if( ! IP.compare( resolvedIP ) )
            return true;

        // Remove wildcards
        sIT = remove( testName.begin(), testName.end(), '*' );
        if(sIT != testName.end())
            testName.erase(sIT);

        // Check against the connected IP
        if( IP.find( testName ) != string::npos )
            return true;

        // Check against the connected domain name
        if( domain.find( testName ) != string::npos )
            return true;
    }

    return false;
}
コード例 #4
0
ファイル: Order.cpp プロジェクト: unixo/SENG
/**
   @brief Return the list of all orders of a given user
 
   @param[in] pp    An instance of User
 
   @return    A vector of Order
 */
vector<Order *> & Order::ordersForUser(User & pp)
{
    vector<Order *> *orders = new vector<Order *>;
    
    // get an instance of the database
    Database& db = Database::instance();
    
    try {
        // ask Database for a valid connection to mySQL
        Connection *conn = db.getConnection();
        
        // obtain an instance of mysqlpp::Query and init it
        Query q = conn->query();
        q << "SELECT * FROM orders WHERE uid = " << pp.uniqueID()
          << " ORDER BY oid, date";
        
        StoreQueryResult res = q.store();
        if (!res.empty()) {
            orders->reserve(res.num_rows());
            StoreQueryResult::const_iterator it;

            for (it = res.begin(); it != res.end(); it++){
                Row row = *it;
                orders->push_back(new Order(row));
            }
        }
    } catch (std::exception &e) {
        cerr << "an error occurred: " << e.what() << endl;
    }
    
    return *orders;
}
コード例 #5
0
ファイル: main.cpp プロジェクト: mffigueroa/crowdify
int
run()
{
	c.connect("crowdify", "localhost", "root", "root");


	create_test_data();

	stringstream qs;
	qs.clear();
	qs.str(string());
	qs << "SELECT * FROM clients;";
	Query q = c.query(qs.str());

	StoreQueryResult r = q.store();
	StoreQueryResult::iterator it;
	
	struct sockaddr_in sin;
	char *username, *password;
	for (it = r.begin(); it != r.end(); it++) {
		Row r = *it;
		run_client(r[0]);
	}

	print_query("select * from blocks;");
	print_query("select * from files;");
	print_query("select * from clients;");
	print_query("select * from filestoblocks;");
	print_query("select * from directoriestofiles;");
	print_query("select * from clientstodirectories;");

	return 0;
}
コード例 #6
0
ファイル: IRCBot.cpp プロジェクト: segrax/KiLLARMY
void IRCBot::actionMIA( ircChannel pChannel, irc_reply_data *hostd, string pParameters ) {
	size_t			count = 1;
	size_t			lastOnline;
	bool			active, showAll = false;

	if( pParameters == "all" )
		showAll = true;

	StoreQueryResult			members =	_db->membersReadSeen( true );
	StoreQueryResult::iterator	memberIT;
	Row							member;

	for(memberIT = members.begin(); memberIT != members.end(); memberIT++ ) {
		stringstream		irc;
		
		member = *memberIT;

		active = (bool) member["active"];
		if( !showAll && !active )
			continue;

		if( _smokingman->thepipeGet()->userOnline( ((size_t) member["memberID"] ) ) != userOFFLINE )
			continue;

		lastOnline = (size_t) member["lastOnline"];
		irc << count << ". " << ((string) member["nameUser"]) << " : " << timeCalc( time(0) - lastOnline );
		speak( "LastOn", irc, pChannel );

		++count;
	}

}
コード例 #7
0
ファイル: Order.cpp プロジェクト: unixo/SENG
/**
   @brief Returns the list of products (and requested quantity) of an order
 
   @return    A std::map where the key is product ID and the value the quantity
 */
map<int, int>& Order::products()
{
    map<int, int> *prd = NULL;
    
    // get an instance of the database
    Database& db = Database::instance();
    
    try {
        // ask Database for a valid connection to mySQL
        Connection *conn = db.getConnection();
        
        // obtain an instance of mysqlpp::Query and init it
        Query q = conn->query();
        q << "SELECT * FROM order_details WHERE oid = "
        << valueForKey(KEY_ORD_OID);            
        
        StoreQueryResult res = q.store();
        if (!res.empty()) {
            prd = new map<int, int>;
            StoreQueryResult::const_iterator it;
            
            for (it = res.begin(); it != res.end(); it++){
                Row row = *it;
                int key = atoi(row["pid"]);
                int qty = atoi(row["qty"]);
                (*prd)[key] = qty;
            }
        }
    } catch (std::exception &e) {
        cerr << "an error occurred: " << e.what() << endl;
    }
    
    return *prd;
}
コード例 #8
0
ファイル: Mysql.cpp プロジェクト: minamotoshi/web
Row Mysql::getRow(std::string sql){
	Query query = _con.query(sql);
	Row row;
	StoreQueryResult res = query.store();
	if (res) {
		row = *res.begin();
	}
	return row;
}
コード例 #9
0
ファイル: main.cpp プロジェクト: mffigueroa/crowdify
int print_query(const char *q) {
	Query q2 = c.query(q);

	StoreQueryResult r = q2.store();
	StoreQueryResult::iterator it;
	
	for (it = r.begin(); it != r.end(); it++) {
		Row r = *it;
		cout << " " << r[0] << endl;
	}
	cout << endl;

	return 0;
}
コード例 #10
0
ファイル: ftpEntry.cpp プロジェクト: segrax/KiLLARMY
void _ftpEntry::checkDBFolders() {
	bool							isDone						= false;
	Query							serverQuery					= _sqlConnection->query();
	StoreQueryResult				serverResults;
	StoreQueryResult::iterator		storeQueryIT;
	vector<_ftpEntry*>::iterator	entryIT;

	// Nothing to do here...
	if( directorys.empty() )
		return;

	// Get a list of all the directorys that we parent
	serverQuery << "select * from ServerFolders where serverID = " << quote << _serverID;
	serverQuery << " AND parentID = "	<< quote << _folderID;

	// Do the query
	serverResults	= serverQuery.store();

	// None?
	if( !serverResults.size() )
		return;

	// We now have to check each DB result, against the results in directorys vector
	for(storeQueryIT = serverResults.begin(); storeQueryIT != serverResults.end(); storeQueryIT++ , isDone = false) {

		for( entryIT = directorys.begin(); entryIT != directorys.end(); entryIT++) {

			// Does this directory still exist?
			if((*entryIT)->getName() == (string) (*storeQueryIT)["name"]) {
				isDone = true;	// Yes it exists
				break;
			}	// parentID check
		}	// entryIT

		// Deleted?
		if(!isDone) {
			// Yes, remove the entry from the DB

            removeFolderChildren( (size_t) (*storeQueryIT)["folderID"] );
			serverQuery.str("");
			serverQuery << "delete from ServerFolders where folderID = "	<< quote << (*storeQueryIT)["folderID"];

			// bye bye :)
			serverQuery.execute();

		}	// !isDone

	}	// storeQueryIT

}	// checkDBFolders
コード例 #11
0
ファイル: IRCBot.cpp プロジェクト: segrax/KiLLARMY
bool IRCBot::testPrepare() {
	StoreQueryResult			 ircBots = _db->ircBotReadAll();
	StoreQueryResult::iterator	 ircIT;
	stringstream				 ircs;
	size_t						 sleepTime = 500, count = 0;
	Row							 irc, ircBot;
	_ircMii						*botMii;

	if(ircBots.empty())
		return false;

	// loop through and start each server test
	for( ircIT = ircBots.begin(); ircIT != ircBots.end(); ++ircIT, ++count ) {
		
		ircBot = *ircIT;
		irc = _db->ircRead( (size_t) ircBot["ircID"] );
		if(irc.empty())
			return false;

		botMii = _Miis->ircMiiLoad( (string) irc["nick"] );
		if(botMii) {
			botMii->ircBotRowLoad();
			botMii->silentSet();
			
			botTestAdd( botMii, 0 );
		} else {
			stringstream ircs;
			ircs << "Bot " << ( (string) irc["nick"] ) << " is not online.";
			speak("Test", ircs, _testChannel);
		}

		Sleep(sleepTime);

		// Every 4, sleep triple time
		if(count == 4) {
			Sleep(sleepTime*2);
			count=0;
		}
	}

	// Nothing to do?
	if(_testMiis.empty()) {
		stringstream ircs;
		ircs << "No bots to test.";
		speak("Test", ircs, _testChannel);
		return false;
	}

	return true;
}
コード例 #12
0
ファイル: db.cpp プロジェクト: i7sid/SkyNet-Dongle
int db::get_station(std::string mac) {
	StoreQueryResult res = q_get_station.store(mac.c_str(), "");

	// found station
	if (res.num_rows() > 0) {
		mysqlpp::StoreQueryResult::const_iterator it;
		mysqlpp::Row row = *(res.begin());
		return row[0];
	}

	cerr << mac << endl;
	// new station to db
	SimpleResult r_exp = q_insert_station.execute("0", "0", mac.c_str());
	return r_exp.insert_id();
}
コード例 #13
0
ファイル: IRCBot.cpp プロジェクト: segrax/KiLLARMY
void IRCBot::actionTotalTime( ircChannel pChannel, irc_reply_data *hostd, string pParameters ) {
	size_t			count = 1;
	time_t			timeOnline;
	bool			active, showAll = false;

	if( pParameters == "all" )
		showAll = true;

	StoreQueryResult			members =	_db->memberTimeOnGet( false );
	StoreQueryResult::iterator	memberIT;
	Row							member;

	for(memberIT = members.begin(); memberIT != members.end(); memberIT++ ) {
		
		member = *memberIT;

		active = (bool) member["active"];
		if( !showAll && !active )
			continue;

		timeOnline = (time_t) member["timeOnline"];

		/*
		if( _smokingman->thepipeGet()->userOnline( ((size_t) member["memberID"] ) ) != userOFFLINE ) {
			client *user = _smokingman->thepipeGet()->userClient(((size_t) member["memberID"] ) );

			time_t addTime = time(0) - user->connectedTimeGet();

			timeOnline += addTime;
		}*/

		if(timeOnline > 0) {
			stringstream		irc;
			irc << count << ". " << ((string) member["nameUser"]) << " : " << timeCalc( timeOnline );
			speak( "TotalTime", irc, pChannel );

			++count;
		}
	}

}
コード例 #14
0
ファイル: Product.cpp プロジェクト: unixo/SENG
/**
   @brief Returns a product with given ID
 
   @param[in] aPid    The requested product ID
 
   @return A pointer to the requested Product, NULL if not found
 */
Product * Product::productByID(int aPid)
{
    // get an instance of the database
    Database &db = Database::instance();
    
    // ask Database for a valid connection to mySQL
    Connection *conn = db.getConnection();
    
    // obtain an instance of mysqlpp::Query and init it
    Query q = conn->query();
    q << SQL_PRODUCT_PROXY << aPid;    
    StoreQueryResult res = q.store();
    if (!res.empty()) {
        StoreQueryResult::const_iterator it = res.begin();
        Row row = *it;
        
        return new Product(row);
    }
    
    return NULL;
}
コード例 #15
0
ファイル: Product.cpp プロジェクト: unixo/SENG
/**
   @brief Obtain a real instance of the Product
 
   Concrete creation of an instance of Product class; when 
   client programmer tries to access to some ProductProxy methods, 
   such as getPrice(), a call to this method ensures that the product 
   is instantiated only we really needed.
 
   @return A pointer to class Product
 */
Product *ProductProxy::getProduct() throw (InvalidArgument)
{
    if (!_theProduct) {
        // get an instance of the database
        Database &db = Database::instance();
        
        // ask Database for a valid connection to mySQL and obtain a Query
        Query q = db.getConnection()->query();
        
        // obtain an instance of mysqlpp::Query and init it
        q << SQL_PRODUCT_PROXY << _pid;
        StoreQueryResult res = q.store();        
        
        if (res.empty())
            throw InvalidArgument("PID");
        
        StoreQueryResult::const_iterator it = res.begin();
        Row row = *it;
        
        _theProduct = new Product(row);
    }
    
    return _theProduct;
}
コード例 #16
0
ファイル: GetInventory.cpp プロジェクト: ElMoustache/GoGo
static vector<Item> handle_get_inventory(const StoreQueryResult& result)
{
	typedef mysqlpp::StoreQueryResult::const_iterator ResultIt;

	vector<Item> items(16);

	for(ResultIt i = result.begin(), end = result.end(); i != end; ++i)
	{
		const Row& row = *i;
		Item item;

		item.cid = row["id"];
		item.itemid = row["itemid"];

		if(row["rental"])
		{
			// TODO(Jacob): Handle item.RentHour
		}

		items.push_back(item);
	}

	return items;
}
コード例 #17
0
ファイル: IRCBot.cpp プロジェクト: segrax/KiLLARMY
void IRCBot::actionOnline( ircChannel pChannel, irc_reply_data *hostd, string pParameters ) {
	vector<size_t>					 doneMembers;
	vector<time_t>					 times;
	vector<time_t>::iterator		 timesIT;
	stringstream					 irc;

	StoreQueryResult				 members = _db->membersRead(0);
	StoreQueryResult::iterator		 memberIT;
	Row								 member;
	
	time_t							 timeOnline;
	client							*userClient = 0;
	size_t							 memberID = 0;

	for(memberIT = members.begin(); memberIT != members.end(); memberIT++ ) {
	
		member = *memberIT;
		memberID = (size_t) member["memberID"];

		//  Get the client
		userClient = _smokingman->thepipeGet()->userClient( memberID );
		if(!userClient)
			continue;
		// Calc the time online
		timeOnline = userClient->connectedTimeGet();

		times.push_back( timeOnline );
		userClient = 0;
	}

	// Sort the times
	sort( times.begin(), times.end() );

	size_t							 count = 1;

	// Output the result
	for(timesIT = times.begin(); timesIT != times.end(); timesIT++ ) {
		timeOnline = *timesIT;

		for(memberIT = members.begin(); memberIT != members.end(); memberIT++ ) {
			member = *memberIT;
			memberID = (size_t) member["memberID"];

			userClient = _smokingman->thepipeGet()->userClient( memberID );
			if(!userClient)
				continue;

			if( find( doneMembers.begin(), doneMembers.end(), memberID ) != doneMembers.end() )
				continue;

			if( timeOnline == userClient->connectedTimeGet() ) {
				doneMembers.push_back( memberID );
				irc << count << ". " << userClient->moduleGet<moduleAuth>(modAUTH)->memberUserNameGet() << " : " << timeCalc( time(0) - timeOnline );
				speak( "Online", irc, pChannel );
				irc.str("");
				++count;
				break;
			}
				
		}
	}
}
コード例 #18
0
ファイル: main.cpp プロジェクト: mffigueroa/crowdify
int
run_client(int cid)
{
	stringstream qs;
	struct sockaddr_in sin;
	char *username, *password;
	StoreQueryResult r;
	StoreQueryResult::iterator it;
	try {
		qs << "SELECT * FROM clients WHERE cid=" << cid << ";";
		Query q = c.query(qs.str());

		r = q.store();
		
		for (it = r.begin(); it != r.end(); it++) {
			Row r = *it;
			cout << " " << r[0] << " " << r[1] << " " << r[2] << " " << r[3] << endl;
			sin.sin_addr.s_addr = r[1];
			username = strdup(r[2]);
			password = strdup(r[3]);
		}
	}
	catch (const mysqlpp::BadQuery& er) {
		cerr << "no clients found: " << er.what() << endl;
		return 1;
	}

	libssh2_init(0);

	int sock = socket(AF_INET, SOCK_STREAM, 0);

	sin.sin_family = AF_INET;
	sin.sin_port = htons(22);

	if (connect(sock, (struct sockaddr*)(&sin), sizeof(struct sockaddr_in)) != 0) {
		cerr << "failed to connect" << endl;
		return 1;
	}

	ssh_session = libssh2_session_init();

	libssh2_session_set_blocking(ssh_session, 1);

	if (libssh2_session_handshake(ssh_session, sock)) {
		cerr << "handshake failed" << endl;
		return 1;
	}

	if (libssh2_userauth_password(ssh_session, username, password)) {
		cerr << "couldn't authenticate user" << endl;
		return 1;
	}

	Query q = c.query("CREATE TABLE IF NOT EXISTS directories (did INT AUTO_INCREMENT PRIMARY KEY NOT NULL, path VARCHAR(4096), uid SMALLINT, gid SMALLINT, perm SMALLINT);");
	q.execute();

	q = c.query("CREATE TABLE IF NOT EXISTS clientstodirectories (cid INT, did INT);");
	q.execute();


	try {
		qs.clear();
		qs.str(string());
		qs << "SELECT * FROM clientstodirectories WHERE cid=" << cid << ";";
		cerr << qs.str() << endl;
		q = c.query(qs.str());
		
		r = q.store();

		for (it = r.begin(); it != r.end(); it++) {
			Row r = *it;
			ssh_ftw(r[1]);
		}
	}
	catch (const mysqlpp::BadQuery& er) {
		cerr << "no clients found: " << er.what() << endl;
		return 1;
	}

	libssh2_session_disconnect(ssh_session, "bye");

	libssh2_session_free(ssh_session);

	libssh2_exit();

	return 0;
}
コード例 #19
0
ファイル: main.cpp プロジェクト: mffigueroa/crowdify
int ssh_ftw(int did) {
	stringstream qs;

	qs.clear();
	qs.str(string());
	qs << "SELECT * FROM directories WHERE did=" << did << ";";
	Query q = c.query(qs.str());
	cerr << qs.str() << endl;

	StoreQueryResult r;
	StoreQueryResult::iterator it;
	
	r = q.store();

	char *dirpath;

	cerr << "Opening: " << dirpath << endl;
	for (it = r.begin(); it != r.end(); it++) {
		Row r = *it;
		dirpath = strdup(r[1]);
	}

	sftp_session = libssh2_sftp_init(ssh_session);
	if (sftp_session == NULL) {
		cerr << "couldn't init sftp" << endl;
		return 1;
	}
	LIBSSH2_SFTP_HANDLE *sftp_dir = libssh2_sftp_opendir(sftp_session, dirpath);
	if (sftp_dir == NULL) {
		cerr << "couldn't open dir" << endl;
		return 1;
	}

	LIBSSH2_SFTP_ATTRIBUTES sftp_attr;

	char *name = new char[512];
	char *mem = new char[512];

	q = c.query("CREATE TABLE IF NOT EXISTS directoriestofiles (did INT, fid INT);");
	q.execute();

	int rc = 0;
	while ((rc = libssh2_sftp_readdir_ex(sftp_dir, mem, 512, name, 512, &sftp_attr)) > 0) {
		cerr << name << ": " << hex << sftp_attr.permissions << endl;
		if ((sftp_attr.permissions & 0100000L) != 0) {
			cout << name << endl;

			Query q = c.query("CREATE TABLE IF NOT EXISTS files (fid INT AUTO_INCREMENT PRIMARY KEY NOT NULL, name VARCHAR(256), ch BIGINT, uid SMALLINT, gid SMALLINT, perm SMALLINT, size BIGINT)");

			q.execute();

			std::stringstream fs;

			fs << "INSERT INTO files VALUES(NULL,\"" << mem << "\"," << 1 << "," << (short int) sftp_attr.uid << "," << (short int) sftp_attr.gid << "," << (short int) sftp_attr.permissions << "," << (long int) sftp_attr.filesize << ");";

			cerr << "query: " << fs.str() << endl;

			q = c.query(fs.str());

			int fid = q.execute().insert_id();
			
			stringstream qs;
			qs << "INSERT INTO directoriestofiles VALUES(" << did << "," << fid << ");";
			cerr << qs.str() << endl;
			q.execute(qs.str());

			stringstream full_name;
			full_name << dirpath << "/" << mem;

			insert_blocks(fid, full_name.str().c_str());
		}
	}

	libssh2_sftp_closedir(sftp_dir);
	libssh2_sftp_shutdown(sftp_session);
	return 0;
}
コード例 #20
0
ファイル: IRCBot.cpp プロジェクト: segrax/KiLLARMY
void IRCBot::actionSearch( ircChannel pChannel, irc_reply_data *hostd, string pParameters ) {
	StoreQueryResult				 serverResults;
	StoreQueryResult::iterator		 resultIT;
	bool							 modeFiles = false;
	stringstream					 irc;
	string							 serverShort;
	if(readServerName(pParameters.substr(0, pParameters.find(" "))) != 0)
		serverShort = parameterGet(pParameters);

	string							 searchTerm = parameterGet(pParameters);
	string							 resultName, next;

	Row								 server, search, folder;

	size_t searchServerID;
	size_t serverID	= readServerName( serverShort );
	size_t folderID = 0, count = 0, parentID = 0;

	searchServerID = serverID;
	if(!searchTerm.size()) {
		speak("search", "syntax wrong, usage", pChannel);
		speak("search", "folder: !search <term>", pChannel);
		speak("search", "files:  !search <term> file", pChannel);
		speak("search", "serverSpecific: !search <server shortname> <term>", pChannel);

		return;
	}

	while(pParameters.size()) {
		next = parameterGet(pParameters);

		if(next[0] == 'f' && pParameters.size() == 0) {
			modeFiles = true;
		} else {
			searchTerm.append(" ");
			searchTerm.append(next);
		}
	}

	if(modeFiles) {
		serverResults = _db->searchFiles( searchTerm );
	} else 
		serverResults = _db->searchFolders( searchTerm, readServerName(serverShort.c_str()) );
	
	if(!serverResults.size()) {
		irc << "No results for query '" << searchTerm << "'";
		speak("search", irc, pChannel);
		return;
	}

	// Have results to parse
	for( resultIT = serverResults.begin(); resultIT != serverResults.end(); resultIT++ ) {
		irc.str("");
		search = *resultIT;

		if(count > 10)
			break;

		if(modeFiles) {
			folderID	= (size_t) search["folderID"];
			folder		= _db->folderRead( folderID );

			serverID = (size_t) folder["serverID"];
			if(searchServerID && serverID != searchServerID)
				continue;

			server = _db->serverRead( serverID );
			serverShort = (string) server["nameShort"];
			resultName  = (string) search["name"];

			irc << "[" << serverShort << "]: " << resultName << " in ";

			parentID = (size_t) folder["parentID"];


		} else {
			serverID = (size_t) search["serverID"];
			server =  _db->serverRead( serverID );
			serverShort = (string) server["nameShort"];
			resultName = (string) search["name"];
			parentID = (size_t) search["parentID"];

			folder = search;
			irc << "[" << serverShort << "]: " << resultName << " in ";
		}

		if(parentID) {
			search = _db->folderRead( parentID );
				if(!search.empty())
					if((string) search["name"] != "/")
						irc << ((string) search["name"]);

		}
		if(modeFiles)
			irc << "/" << ((string) folder["name"]);

		speak("search", irc, pChannel);
		count++;
	}

}
コード例 #21
0
ファイル: view_mng.cpp プロジェクト: alpha2z/chaos
int CViewMng::init(Connection *conn)
{
	CoCycle *cycle = g_cycle;

	if(conn == NULL)	return -1;

	mDBConn = conn;

	StoreQueryResult::const_iterator it;

	Row row ;

	Query query = mDBConn->query();
	
	query << "select id, ltrim(rtrim(name)), ltrim(rtrim(remark)), "
		<< "interruptable, type from t_view_set";
	StoreQueryResult res = query.store();

	if(res.empty()) return -2;

	static char buf[2048];
	CodeConverter cc = CodeConverter("utf-8","gb2312");

	for (it = res.begin(); it != res.end(); ++it) {
		row = *it;

		ViewSet *viewSet = cycle->mPool->newObj<ViewSet>();
		assert(viewSet != NULL);

		viewSet->id = row[0];

		memset(buf, 0, sizeof(buf));
		cc.convert((char *)row[1].c_str(), row[1].length(), buf, sizeof(buf));

		viewSet->name = buf;

		memset(buf, 0, sizeof(buf));
		cc.convert((char *)row[2].c_str(), row[2].length(), buf, sizeof(buf));

		viewSet->remark = buf;

		int interruptable = row[3];
		if(interruptable == 0)		viewSet->interruptable = true;
		else viewSet->interruptable = false;

		viewSet->type = row[4];
		insertViewSet(viewSet);
	}

	query << "select a.id, ltrim(rtrim(a.name)), ltrim(rtrim(a.content)), "
		<< " ltrim(rtrim(b.name)) from t_view as a " 
		<< "left join t_view_set as b on a.view_set_id = b.id";
	res = query.store();


	if(res.empty())	return -3;

	for (it = res.begin(); it != res.end(); ++it) {
		row = *it;

		View *view = cycle->mPool->newObj<View>();
		assert(view != NULL);

		view->id = row[0];

		memset(buf, 0, sizeof(buf));
		cc.convert((char *)row[1].c_str(), row[1].length(), buf, sizeof(buf));

		view->name = buf;

		memset(buf, 0, sizeof(buf));
		cc.convert((char *)row[2].c_str(), row[2].length(), buf, sizeof(buf));

		view->content = buf;

		insertView(row[3].c_str(), view);
	}
	
	query << "select a.id, ltrim(rtrim(a.name)), ltrim(rtrim(a.title)), "
		<< " ltrim(rtrim(a.action)), a.type, ltrim(rtrim(b.name)) "
		<<"from t_click_action as a left join t_view_set as b "
		<< "on a.view_set_id = b.id";
	res = query.store();
	
	if(res.empty())	return -4;

	for (it = res.begin(); it != res.end(); ++it) {
		row = *it;

		ClickAction *clickAction = cycle->mPool->newObj<ClickAction>();
		assert(clickAction != NULL);

		clickAction->id = row[0];

		memset(buf, 0, sizeof(buf));
		cc.convert((char *)row[1].c_str(), row[1].length(), buf, sizeof(buf));

		clickAction->name = buf;

		memset(buf, 0, sizeof(buf));
		cc.convert((char *)row[2].c_str(), row[2].length(), buf, sizeof(buf));

		clickAction->title = buf;

		memset(buf, 0, sizeof(buf));
		cc.convert((char *)row[3].c_str(), row[3].length(), buf, sizeof(buf));

		clickAction->action = buf;

		clickAction->type = row[4];

		insertClickAction(row[5].c_str(), clickAction);
	}

	return 0;
}