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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
|
#include "portfolio.h"
char* portfolio_file_path;
void portfolio_file_path_init(void) {
char* home = getenv("HOME");
char* path = malloc(strlen(home) + 30);
pointer_alloc_check(path);
sprintf(path, "%s/.tick_portfolio.json", home);
portfolio_file_path = path;
}
String* portfolio_ncurses_get_plaintext_string(char** password_ref) {
String* pString = file_get_string(portfolio_file_path);
if (pString != NULL && pString->len > 0) { // NULL if error reading file
char* password = NULL;
if (!rc4_ncurses_decrypt_string(pString, &password))
string_destroy(&pString); // Error decrypting
if (password_ref != NULL)
*password_ref = password; // Copy password if given reference
else free(password);
}
return pString;
}
void portfolio_modify_write(const char* symbol, double quantity_shares, double usd_spent,
int mod_option) {
if (quantity_shares < 0 || usd_spent < 0) // Negative numbers
RET_MSG("You must use positive values.")
if (mod_option != SET && quantity_shares == 0 && usd_spent == 0) // Adding or removing 0
RET_MSG("You cannot add or remove values of 0.")
FILE* fp = fopen(portfolio_file_path, "a"); // Creates empty file if portfolio doesn't exist
if (fp == NULL) // If file exists, but cannot be opened, usually because of permissions
RET_MSG("Error opening porfolio.")
fclose(fp);
char* password = NULL; // Store password for re-encryption
String* pString = portfolio_ncurses_get_plaintext_string(&password);
if (pString == NULL) // Decryption error
return;
// Perform modification
portfolio_modify_string(pString, symbol, quantity_shares, usd_spent, mod_option);
if (password != NULL) { // If need to re-encrypt
rc4_encode_string(pString, password);
puts("Re-encrypting portfolio...");
free(password);
}
string_write_file(pString, portfolio_file_path); // Write portfolio file
string_destroy(&pString);
}
int portfolio_modify_string(String* pString, const char* symbol, double quantity_shares,
double usd_spent, int mod_option) {
// Null string or negative numbers
if ((symbol != NULL && symbol[0] == '\0') || quantity_shares < 0 || usd_spent < 0)
return 1;
// Can't add or remove both values of 0
if (mod_option != SET && quantity_shares == 0 && usd_spent == 0)
return 1;
int status = 0;
Json* jobj;
if (pString->len == 0) // Create new array if empty file
jobj = json_object_new_array();
else jobj = json_tokener_parse(pString->data); // Otherwise parse
if (jobj == NULL)
RET_FALSE_MSG("Portfolio corrupted! Check JSON formatting.");
int index = portfolio_symbol_index(symbol, jobj);
if (index == -1) { // If security is not already contained in portfolio
if (mod_option == REMOVE) {// If trying to remove a security they don't own
status = 1;
GOTO_CLEAN_MSG("You don't have any of this security to remove")
}
if (strcmp("USD$", symbol) != 0) { // Check that the symbol is valid, except if it's USD
Info* data = api_info_init();
strcpy(data->symbol, symbol);
api_info_store_data_batch(data, CHECK);
if (data->api_provider == EMPTY) {// If NULL response from APIs, it's invalid
api_info_destroy(&data);
status = 1;
GOTO_CLEAN_MSG("Invalid symbol.")
}
api_info_destroy(&data);
}
Json* new_object = json_object_new_object(); // Creates new array index and adds values to it
json_object_array_add(jobj, new_object);
json_object_object_add(new_object, "Symbol", json_object_new_string(symbol));
json_object_object_add(new_object, "Shares", json_object_new_double(quantity_shares));
json_object_object_add(new_object, "USD_Spent", json_object_new_double(usd_spent));
printf("Added %lf %s bought for %lf to portfolio.\n", quantity_shares, symbol, usd_spent);
} else { //if already in portfolio
Json* current_index = json_object_array_get_idx(jobj, (size_t) index);
// Store values already in portfolio to modify.
double current_shares = json_object_get_double(json_object_object_get(current_index, "Shares"));
double current_spent = json_object_get_double(json_object_object_get(current_index, "USD_Spent"));
json_object_object_del(current_index, "Shares"); // Deletes the objects already there
json_object_object_del(current_index, "USD_Spent");
if (mod_option == SET) { // SET
current_shares = quantity_shares;
current_spent = usd_spent;
printf("Set amount of %s in portfolio to %lf bought for %lf.\n", symbol, quantity_shares,
usd_spent);
} else if (mod_option == ADD) { // ADD
current_shares += quantity_shares;
current_spent += usd_spent;
printf("Added %lf %s bought for %lf to portfolio.\n", quantity_shares, symbol, usd_spent);
} else { // REMOVE
current_shares -= quantity_shares;
current_spent -= usd_spent;
if (current_shares < 0 || current_spent < 0) {// If you try to remove more than you have
status = 1;
GOTO_CLEAN_MSG("You don't have enough of this security to remove.")
}
printf("Removed %lf %s bought for %lf to portfolio.\n", quantity_shares, symbol, usd_spent);
}
if (current_shares == 0 && usd_spent == 0) // Deletes index from portfolio if values are 0
json_object_array_del_idx(jobj, (size_t) index, 1);
else { // Adds computed values to index
json_object_object_add(current_index, "Shares", json_object_new_double(current_shares));
json_object_object_add(current_index, "USD_Spent", json_object_new_double(current_spent));
}
}
const char* new_portfolio_str = json_object_to_json_string(jobj);
pString->len = strlen(new_portfolio_str);
pString->data = realloc(pString->data, pString->len + 1); // Realloc string with size of new length
pointer_alloc_check(pString->data);
strcpy(pString->data, new_portfolio_str);
cleanup: // CLEANUP
json_object_put(jobj);
return status;
}
Info_Array* portfolio_info_array_init_from_portfolio_string(String* pString) {
if (pString->len == 0) // If empty portfolio file
RETNULL_MSG("Your portfolio is empty.")
Json* jobj = json_tokener_parse(pString->data);
size_t length = json_object_array_length(jobj);
if (length == 0) { // If empty array
json_object_put(jobj);
RETNULL_MSG("Your portfolio is empty.");
}
Info_Array* portfolio_data = api_info_array_init_from_length(length);
portfolio_data->totals->total_spent = 0;
for (size_t i = 0; i < portfolio_data->length; i++) {
strcpy(portfolio_data->array[i]->symbol,
json_object_get_string(json_object_object_get(json_object_array_get_idx(jobj, i),
"Symbol")));
portfolio_data->array[i]->amount = json_object_get_double(json_object_object_get(
json_object_array_get_idx(jobj, i), "Shares"));
portfolio_data->array[i]->total_spent = json_object_get_double(json_object_object_get(
json_object_array_get_idx(jobj, i), "USD_Spent"));
portfolio_data->totals->total_spent += portfolio_data->array[i]->total_spent;
}
json_object_put(jobj);
return portfolio_data;
}
void portfolio_sort(Info_Array* portfolio_data, int sort_option) {
if (portfolio_data->length == 1) // Can't sort only one security
return;
int loop_flag = 1;
double val1 = 0, val2 = 0;
Info* sec_data1, * sec_data2, * temp;
while (loop_flag) { // Bubble sort
loop_flag = 0;
for (size_t i = 0; i < portfolio_data->length - 1; i++) {
sec_data1 = portfolio_data->array[i];
sec_data2 = portfolio_data->array[i + 1];
if (sort_option == SORT_ALPHA || sort_option > SORT_PROFIT_30D_PERCENT) {
if (strcmp(sec_data1->symbol, sec_data2->symbol) > 0) { // Least to greatest
temp = portfolio_data->array[i]; // Swap
portfolio_data->array[i] = portfolio_data->array[i + 1];
portfolio_data->array[i + 1] = temp;
loop_flag = 1;
}
} else if (sort_option == SORT_VALUE) {
val1 = sec_data1->current_value;
val2 = sec_data2->current_value;
} else if (sort_option == SORT_SPENT) {
val1 = sec_data1->total_spent;
val2 = sec_data2->total_spent;
} else if (sort_option == SORT_PROFIT) {
val1 = sec_data1->profit_total;
val2 = sec_data2->profit_total;
} else if (sort_option == SORT_PROFIT_PERCENT) {
val1 = sec_data1->profit_total_percent;
val2 = sec_data2->profit_total_percent;
} else if (sort_option == SORT_PROFIT_24H) {
val1 = sec_data1->profit_last_close;
val2 = sec_data2->profit_last_close;
} else if (sort_option == SORT_PROFIT_24H_PERCENT) {
val1 = sec_data1->profit_last_close_percent;
val2 = sec_data2->profit_last_close_percent;
} else if (sort_option == SORT_PROFIT_7D) {
val1 = sec_data1->profit_7d;
val2 = sec_data2->profit_7d;
} else if (sort_option == SORT_PROFIT_7D_PERCENT) {
val1 = sec_data1->profit_7d_percent;
val2 = sec_data2->profit_7d_percent;
} else if (sort_option == SORT_PROFIT_30D) {
val1 = sec_data1->profit_30d;
val2 = sec_data2->profit_30d;
} else if (sort_option == SORT_PROFIT_30D_PERCENT) {
val1 = sec_data1->profit_30d_percent;
val2 = sec_data2->profit_30d_percent;
}
if (val1 < val2) { // Greatest to least
temp = portfolio_data->array[i]; // Swap
portfolio_data->array[i] = portfolio_data->array[i + 1];
portfolio_data->array[i + 1] = temp;
loop_flag = 1;
}
}
}
}
int portfolio_symbol_index(const char* symbol, const Json* jarray) {
for (size_t i = 0; i < json_object_array_length(jarray); i++) {
const char* string = json_object_get_string(
json_object_object_get(json_object_array_get_idx(jarray, (size_t) i), "Symbol"));
if (strcmp(string, symbol) == 0)
return (int) i;
}
return -1;
}
|