84 lines
2.2 KiB
C++
84 lines
2.2 KiB
C++
|
|
|
||
|
|
#include<iostream>
|
||
|
|
#include<string>
|
||
|
|
#include<filesystem>
|
||
|
|
#include "sqlite3.h"
|
||
|
|
|
||
|
|
using std::cout;
|
||
|
|
using std::endl;
|
||
|
|
using std::filesystem::directory_iterator;
|
||
|
|
using std::filesystem::directory_entry;
|
||
|
|
|
||
|
|
void directoryList(std::string path){
|
||
|
|
std::filesystem::directory_iterator Start{path};
|
||
|
|
|
||
|
|
for (const auto& entry : Start){
|
||
|
|
std::cout << entry.path() << std::endl;
|
||
|
|
if( entry.is_directory() ){
|
||
|
|
for (const auto& subentry : std::filesystem::directory_iterator(entry.path())){
|
||
|
|
std::cout << subentry.path() << std::endl;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void insertEntry(sqlite3* pDB, std::filesystem::directory_entry direntry){
|
||
|
|
sqlite3_stmt *pstmt = NULL;
|
||
|
|
int rc = 0;
|
||
|
|
rc = sqlite3_prepare_v3(pDB, "INSERT INTO FSCHILD(child_dir) VALUES(?1);", -1, 0, &pstmt, NULL);
|
||
|
|
if( rc ){
|
||
|
|
std::cerr << sqlite3_errmsg(pDB) << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
sqlite3_bind_text(pstmt, 1, direntry.path().generic_string().c_str(), -1, SQLITE_TRANSIENT);
|
||
|
|
|
||
|
|
rc = sqlite3_step(pstmt);
|
||
|
|
if (rc != SQLITE_DONE){
|
||
|
|
std::cerr << sqlite3_errmsg(pDB) << std::endl;
|
||
|
|
}
|
||
|
|
sqlite3_finalize(pstmt);
|
||
|
|
}
|
||
|
|
|
||
|
|
void insertDirectoryList(sqlite3* pDB, std::string path){
|
||
|
|
|
||
|
|
std::filesystem::directory_iterator Start{path};
|
||
|
|
|
||
|
|
for (const auto& entry : Start){
|
||
|
|
std::cout << entry.path() << std::endl;
|
||
|
|
if( entry.is_directory() ){
|
||
|
|
std::filesystem::directory_iterator SubPath{entry.path()};
|
||
|
|
for (const auto & subentry : SubPath){
|
||
|
|
insertEntry(pDB, subentry);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static int callback(void* data, int colCount, char** colData, char** colName)
|
||
|
|
{
|
||
|
|
std::string val = colData[0];
|
||
|
|
if (val == "1") {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
int main()
|
||
|
|
{
|
||
|
|
sqlite3* pDB = NULL;
|
||
|
|
int rc = 0;
|
||
|
|
|
||
|
|
rc = sqlite3_open("fs-meta.db", &pDB);
|
||
|
|
|
||
|
|
std::string sql("create table FSCHILD(child_dir STRING PRIMARY KEY NOT NULL );");
|
||
|
|
std::string data("CALLBACK FUNCTION");
|
||
|
|
rc = sqlite3_exec(pDB, sql.c_str(), callback, (void *)data.c_str(), NULL);
|
||
|
|
|
||
|
|
if( rc ){
|
||
|
|
std::cerr << sqlite3_errmsg(pDB) << std::endl;
|
||
|
|
}else{
|
||
|
|
insertDirectoryList(pDB, "/var");
|
||
|
|
}
|
||
|
|
sqlite3_close(pDB);
|
||
|
|
|
||
|
|
}
|