File: importcsv.cc

package info (click to toggle)
signalbackup-tools 20250313.1-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 3,752 kB
  • sloc: cpp: 47,042; sh: 477; ansic: 399; ruby: 19; makefile: 3
file content (97 lines) | stat: -rw-r--r-- 2,963 bytes parent folder | download
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
/*
  Copyright (C) 2021-2024  Selwin van Dijk

  This file is part of signalbackup-tools.

  signalbackup-tools is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  signalbackup-tools 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with signalbackup-tools.  If not, see <https://www.gnu.org/licenses/>.
*/

#include "signalbackup.ih"

/*
 * Things to deal with:
 * get proper thread from id (phone number)
 * deal with type, in what way do we assume incoming/outgoing is specified in the csv
 * should imported messages be marked as received / read
 * should they be imported as secured or unsecured messages?
 */

bool SignalBackup::importCSV(std::string const &file, std::map<std::string, std::string> const &fieldmap)
{
  CSVReader csvfile(file);
  if (!csvfile.ok())
    return false;

  std::string statementstub("INSERT INTO sms SET (");

  int64_t idx_of_address = -1;
  //int64_t idx_of_type = -1;
  std::vector<unsigned int> date_indeces;

  // get columns to set
  for (unsigned int i = 0; i < csvfile.fields(); ++i)
  {
    std::string fieldname = csvfile.getFieldName(i);
    if (fieldmap.find(fieldname) != fieldmap.end())// (fieldmap.contains(fieldname))
      fieldname = fieldmap.at(fieldname);

    if (fieldname == d_sms_recipient_id)
      idx_of_address = i;
    else if (fieldname == "type")
      ;//idx_of_type = i;
    else if (fieldname.find("date") != std::string::npos) /// not sure what this does, and if it works as intended
      date_indeces.push_back(i);                          // with d_sms_date_received

    statementstub += fieldname + ',';
  }
  statementstub += "thread_id) VALUES (";

  // build statement from each row
  for (unsigned int msg = 0; msg < csvfile.rows(); ++msg)
  {
    std::string statement = statementstub;
    for (unsigned int f = 0; f < csvfile.fields(); ++f)
    {
      //if (f == idx_of_type)
      //  translate type?

      //if (date_indeces.contains(f))
      //{
      //  std::string date = csvfile.get(f, msg);
      //  if (date.find_first_not_of("0123456789") != std::string::npos)
      //    translate(date);
      //}

      statement += csvfile.get(f, msg) + ',';
    }

    // determine thread_id
    long long int tid = getThreadIdFromRecipient(csvfile.get(idx_of_address, msg));
    if (tid == -1)
    {
      Logger::error("Unable to determine thread_id for message.");
      return false;
    }
    statement += bepaald::toString(tid);



    statement += ')';

    if (!d_database.exec(statement))
      return false;

  }
  return true;
}