database.hpp

/*
 SPDX-FileCopyrightText: © 2014 Siemens AG
 Author: Daniele Fognini
 
 SPDX-License-Identifier: GPL-2.0-only
*/
 
#ifndef DATABASE_HPP
#define DATABASE_HPP
 
#include <string>
#include <vector>
 
#include "libfossdbmanagerclass.hpp"
#include "libfossAgentDatabaseHandler.hpp"
#include "cleanEntries.hpp"
 
#define MAX_TABLE_CREATION_RETRIES 5
 
class DatabaseEntry
{
public:
  DatabaseEntry();
 
  long agent_fk;                    
  long pfile_fk;                    
  std::string content;              
  std::string hash;                 
  std::string type;
  int copy_startbyte;               
  int copy_endbyte;                 
};
 
class CopyrightDatabaseHandler : public fo::AgentDatabaseHandler
{
public:
  CopyrightDatabaseHandler(fo::DbManager manager);
  CopyrightDatabaseHandler(const CopyrightDatabaseHandler&) = delete;
  CopyrightDatabaseHandler(CopyrightDatabaseHandler&& other) : fo::AgentDatabaseHandler(std::move(other)) {}; // = default
  CopyrightDatabaseHandler spawn() const;
 
  bool createTables() const;
  bool insertInDatabase(DatabaseEntry& entry) const;
  std::vector<unsigned long> queryFileIdsForUpload(int agentId, int uploadId, bool ignoreFilesWithMimeType);
 
private:
  typedef struct
  {
    const char* name;               
    const char* type;               
    const char* creationFlags;      
  } ColumnDef;
 
  static const ColumnDef columns[];
  static const ColumnDef columnsDecision[];
 
  bool createTableAgentFindings() const;
  bool createTableClearing() const;
  std::string getColumnListString(const ColumnDef in[], size_t size) const;
  std::string getColumnCreationString(const ColumnDef in[], size_t size) const;
};
 
#endif // DATABASE_HPP