curseSplitOutput.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846
  1. /**
  2. * curseSplitOutput.cpp for jsonstroller
  3. *
  4. * Author: isundil <isundill@gmail.com>
  5. **/
  6. #include <sys/ioctl.h>
  7. #include <unistd.h>
  8. #include <signal.h>
  9. #include <curses.h>
  10. #include "searchPattern.hh"
  11. #include "curseSplitOutput.hh"
  12. #include "jsonObject.hh"
  13. #include "jsonArray.hh"
  14. #include "jsonPrimitive.hh"
  15. #include "levenshteinMatrice.hpp"
  16. template<class T> const T &list_at(const std::list<T> &l, unsigned int pos)
  17. {
  18. typename std::list<T>::const_iterator it = l.cbegin();
  19. std::advance(it, pos);
  20. return *it;
  21. }
  22. CurseSplitOutput::CurseSplitOutput(const Params &p): CurseOutput(p)
  23. {
  24. diffMatrice = nullptr;
  25. init();
  26. }
  27. CurseSplitOutput::~CurseSplitOutput()
  28. {
  29. shutdown();
  30. if (diffMatrice)
  31. {
  32. delete diffMatrice;
  33. diffMatrice = nullptr;
  34. }
  35. }
  36. void CurseSplitOutput::run(const std::deque<std::string> &inputName, const std::deque<JSonElement*> &roots)
  37. {
  38. nbInputs = inputName.size();
  39. const t_Cursor screenSize = getScreenSize();
  40. selectedWin = 0;
  41. destroyAllSubWin();
  42. subWindows.clear();
  43. for (size_t i =0; i < nbInputs; ++i)
  44. {
  45. t_subWindow subwin;
  46. subwin.fileName = inputName.at(i);
  47. subwin.selection = subwin.lastSelection = subwin.root = roots.at(i);
  48. subwin.select_up = subwin.select_down = nullptr;
  49. subwin.innerWin = subwin.outerWin = nullptr;
  50. subwin.scrollTop = 0;
  51. subwin.outerWin = newwin(screenSize.second +2, screenSize.first, 0, i * (screenSize.first -1));
  52. subwin.innerWin = newwin(screenSize.second, screenSize.first -2, 1, i * (screenSize.first -1) +1);
  53. keypad(subwin.outerWin, true);
  54. subWindows.push_back(subwin);
  55. box(subwin.outerWin, 0, 0);
  56. wrefresh(subwin.outerWin);
  57. }
  58. computeDiff();
  59. loop(subWindows.at(0).outerWin);
  60. }
  61. void CurseSplitOutput::computeDiff()
  62. {
  63. //TODO diffMatrice should be LevenshteinMatrice_base[nbInputs -1]
  64. //And we should iterate such as diffMatrice[n] = diff(n, n+1) ?
  65. if (diffMatrice)
  66. delete diffMatrice;
  67. LevenshteinMatrice_base::Builder builder;
  68. if (nbInputs == 2)
  69. diffMatrice = builder.build(subWindows.at(0).root, subWindows.at(1).root);
  70. else if (nbInputs == 3)
  71. throw std::runtime_error("3-input diff not implemented");
  72. }
  73. inputResult CurseSplitOutput::selectUp()
  74. {
  75. setSelection(subWindows.at(selectedWin).select_up);
  76. return inputResult::redraw;
  77. }
  78. inputResult CurseSplitOutput::selectDown()
  79. {
  80. const JSonElement *newSelection = subWindows.at(selectedWin).select_down;
  81. size_t i = 0;
  82. for (t_subWindow &w : subWindows)
  83. {
  84. if (w.selectIsLast)
  85. w.scrollTop += 2;
  86. if (i == selectedWin)
  87. w.selection = w.lastSelection = newSelection;
  88. else
  89. {
  90. w.selection = diffMatrice->getEquivalence(newSelection);
  91. if (w.selection)
  92. w.lastSelection = w.selection;
  93. }
  94. ++i;
  95. }
  96. return inputResult::redraw;
  97. }
  98. void CurseSplitOutput::setSelection(const JSonElement *selection)
  99. {
  100. size_t i =0;
  101. for (t_subWindow &w : subWindows)
  102. {
  103. if (i == selectedWin)
  104. w.selection = w.lastSelection = selection;
  105. else
  106. {
  107. w.selection = diffMatrice->getEquivalence(selection);
  108. if (w.selection)
  109. w.lastSelection = w.selection;
  110. }
  111. ++i;
  112. }
  113. }
  114. inputResult CurseSplitOutput::selectPUp()
  115. {
  116. const JSonElement *_selection = subWindows.at(selectedWin).selection;
  117. const JSonElement *nextSelection = _selection->findPrev();
  118. if (nextSelection == nullptr)
  119. {
  120. const JSonElement *parent = _selection->getParent();
  121. if (parent && dynamic_cast<const JSonContainer*>(parent))
  122. {
  123. nextSelection = _selection = parent;
  124. if (_selection->getParent() && dynamic_cast<const JSonObjectEntry*> (_selection->getParent()))
  125. nextSelection = _selection->getParent();
  126. }
  127. else
  128. return inputResult::nextInput;
  129. }
  130. setSelection(nextSelection);
  131. return inputResult::redraw;
  132. }
  133. inputResult CurseSplitOutput::selectPDown()
  134. {
  135. const JSonElement *brother = subWindows.at(selectedWin).selection->findNext();
  136. if (brother)
  137. {
  138. setSelection(brother);
  139. return inputResult::redraw;
  140. }
  141. return inputResult::nextInput;
  142. }
  143. inputResult CurseSplitOutput::expandSelection()
  144. {
  145. const JSonElement *_selection = subWindows.at(selectedWin).selection;
  146. if (dynamic_cast<const JSonObjectEntry*>(_selection))
  147. _selection = **((const JSonObjectEntry*)_selection);
  148. if (!dynamic_cast<const JSonContainer*>(_selection))
  149. return inputResult::nextInput;
  150. if (collapsed.erase((const JSonContainer *)_selection))
  151. {
  152. _selection = diffMatrice->getEquivalence(_selection);
  153. collapsed.erase((const JSonContainer *)_selection);
  154. return inputResult::redraw;
  155. }
  156. if (!((const JSonContainer*)_selection)->size())
  157. return inputResult::nextInput;
  158. selectDown();
  159. return inputResult::redraw;
  160. }
  161. inputResult CurseSplitOutput::collapseSelection()
  162. {
  163. const JSonElement *_selection = subWindows.at(selectedWin).selection;
  164. if (dynamic_cast<const JSonObjectEntry*>(_selection))
  165. _selection = **((const JSonObjectEntry*)_selection);
  166. if (_selection->getParent() && (!dynamic_cast<const JSonContainer*>(_selection)
  167. || collapsed.find((const JSonContainer *)_selection) != collapsed.end()
  168. || (dynamic_cast<const JSonContainer*>(_selection) && ((const JSonContainer*)_selection)->size() == 0)))
  169. {
  170. _selection = subWindows.at(selectedWin).selection->getParent();
  171. if (_selection->getParent() && dynamic_cast<const JSonObjectEntry*>(_selection->getParent()))
  172. setSelection(_selection->getParent());
  173. else
  174. setSelection(_selection);
  175. }
  176. else
  177. {
  178. collapsed.insert((const JSonContainer *)_selection);
  179. _selection = diffMatrice->getEquivalence(_selection);
  180. if (_selection)
  181. collapsed.insert((const JSonContainer *)_selection);
  182. }
  183. return inputResult::redraw;
  184. }
  185. inputResult CurseSplitOutput::initSearch()
  186. {
  187. const SearchPattern *searchPattern = inputSearch();
  188. if (!searchPattern)
  189. return inputResult::redraw;
  190. for (t_subWindow &s : subWindows)
  191. s.searchResults.clear();
  192. if (searchPattern->isEmpty())
  193. return inputResult::redraw;
  194. search(*searchPattern);
  195. delete searchPattern;
  196. return nextResult();
  197. }
  198. inputResult CurseSplitOutput::nextResult()
  199. {
  200. if (subWindows.at(selectedWin).searchResults.empty())
  201. CurseOutput::redraw("Pattern not found");
  202. else if (jumpToNextSearch())
  203. return inputResult::redraw;
  204. return inputResult::nextInput;
  205. }
  206. inputResult CurseSplitOutput::changeWindow(char d, bool c)
  207. {
  208. if ((selectedWin +d < 0 || selectedWin +d >= nbInputs) && !c)
  209. return inputResult::nextInput;
  210. selectedWin = (selectedWin +d) % nbInputs;
  211. t_subWindow &w = subWindows.at(selectedWin);
  212. //Restore selection
  213. if (!w.selection)
  214. {
  215. size_t i =0;
  216. for (t_subWindow &it : subWindows)
  217. {
  218. if (i == selectedWin)
  219. it.selection = it.lastSelection;
  220. else
  221. {
  222. it.selection = diffMatrice->getEquivalence(w.lastSelection);
  223. if (it.selection)
  224. it.lastSelection = it.selection;
  225. }
  226. ++i;
  227. }
  228. }
  229. return inputResult::redrawAll;
  230. }
  231. void CurseSplitOutput::checkSelection(const JSonElement *item)
  232. {
  233. t_subWindow &w = subWindows.at(workingWin);
  234. if (!w.selectFound)
  235. {
  236. if (w.lastSelection == item)
  237. {
  238. if (w.cursor.second < w.scrollTop) //Selection is above vp, move scroll pos to selection and start drawing
  239. w.scrollTop = w.cursor.second;
  240. w.selectFound = true;
  241. }
  242. else if (!item->getParent() || !dynamic_cast<const JSonObjectEntry*>(item->getParent()))
  243. w.select_up = item;
  244. }
  245. else if (!w.select_down)
  246. {
  247. const JSonElement *parent = item->getParent();
  248. if (!dynamic_cast<const JSonContainer*>(item) &&
  249. parent &&
  250. w.lastSelection != parent &&
  251. dynamic_cast<const JSonObjectEntry*>(parent))
  252. item = parent;
  253. if (!item->getParent() || !dynamic_cast<const JSonObjectEntry*>(item->getParent()))
  254. w.select_down = item;
  255. }
  256. }
  257. bool CurseSplitOutput::jumpToNextSearch(const JSonElement *current, bool &selectFound)
  258. {
  259. const JSonContainer *container = dynamic_cast<const JSonContainer *> (current);
  260. const JSonObjectEntry *objEntry = dynamic_cast<const JSonObjectEntry *> (current);
  261. if (subWindows.at(selectedWin).lastSelection == current)
  262. selectFound = true;
  263. if (container)
  264. {
  265. if (!container->empty())
  266. for (const JSonElement *it : *container)
  267. if (jumpToNextSearch(it, selectFound))
  268. return true;
  269. }
  270. else
  271. {
  272. if (current &&
  273. std::find(subWindows.at(selectedWin).searchResults.cbegin(),
  274. subWindows.at(selectedWin).searchResults.cend(),
  275. current) != subWindows.at(selectedWin).searchResults.cend() &&
  276. current != subWindows.at(selectedWin).selection &&
  277. selectFound)
  278. {
  279. subWindows.at(selectedWin).lastSelection = current;
  280. return true;
  281. }
  282. if (objEntry)
  283. if (jumpToNextSearch(**objEntry, selectFound))
  284. return true;
  285. }
  286. return false;
  287. }
  288. unsigned int CurseSplitOutput::search(const SearchPattern &searchPattern)
  289. {
  290. unsigned int result =0;
  291. for (t_subWindow &w : subWindows)
  292. result += search(searchPattern, w.root);
  293. return result;
  294. }
  295. unsigned int CurseSplitOutput::search(const SearchPattern &searchPattern, const JSonElement *current)
  296. {
  297. const JSonContainer *container = dynamic_cast<const JSonContainer *> (current);
  298. const JSonObjectEntry *objEntry = dynamic_cast<const JSonObjectEntry *> (current);
  299. if (container)
  300. {
  301. if (!container->empty())
  302. for (const JSonElement *it : *container)
  303. search(searchPattern, it);
  304. }
  305. else
  306. {
  307. if (current && current->match(searchPattern))
  308. {
  309. if (current->getParent() && dynamic_cast<const JSonObjectEntry*>(current->getParent()))
  310. subWindows.at(workingWin).searchResults.push_back(current->getParent());
  311. else
  312. subWindows.at(workingWin).searchResults.push_back(current);
  313. }
  314. if (objEntry)
  315. search(searchPattern, **objEntry);
  316. }
  317. return subWindows.at(workingWin).searchResults.size();
  318. }
  319. bool CurseSplitOutput::jumpToNextSearch()
  320. {
  321. bool selectFound = false;
  322. bool res = jumpToNextSearch(subWindows.at(selectedWin).root, selectFound);
  323. if (!res)
  324. {
  325. subWindows.at(selectedWin).lastSelection = *(subWindows.at(selectedWin).searchResults.cbegin());
  326. unfold(subWindows.at(selectedWin).selection);
  327. CurseOutput::redraw("Search hit BOTTOM, continuing at TOP");
  328. return false;
  329. }
  330. unfold(subWindows.at(selectedWin).selection);
  331. return true;
  332. }
  333. const Optional<bool> CurseSplitOutput::redrawOneItemToWorkingWin(t_subWindow &w, const t_Cursor &screenSize)
  334. {
  335. bool result;
  336. try {
  337. if (w.parentsIterators.empty())
  338. result = redraw(w, screenSize, w.root);
  339. else
  340. result = redraw(w, screenSize, w.parentsIterators.top());
  341. }
  342. catch (SelectionOutOfRange &e)
  343. {
  344. return Optional<bool>::empty;
  345. }
  346. if (!result || w.parentsIterators.empty())
  347. {
  348. if (!result)
  349. {
  350. if (!w.selectFound)
  351. {
  352. w.scrollTop++;
  353. return Optional<bool>::empty;
  354. }
  355. if (!w.select_down)
  356. w.selectIsLast = true;
  357. }
  358. if (!w.select_down)
  359. {
  360. const JSonContainer *pselect = dynamic_cast<const JSonContainer*>(w.selection);
  361. if (pselect && !pselect->empty() && collapsed.find(pselect) == collapsed.cend())
  362. w.select_down = *(pselect->cbegin());
  363. else
  364. {
  365. const JSonElement *next = w.lastSelection->findNext();
  366. w.select_down = next ? next : w.lastSelection;
  367. }
  368. }
  369. if (!w.select_up)
  370. w.select_up = subWindows.at(workingWin).lastSelection;
  371. return Optional<bool>::of(true);
  372. }
  373. return Optional<bool>::of(false);
  374. }
  375. bool CurseSplitOutput::redraw()
  376. {
  377. const t_Cursor screenSize = getScreenSize();
  378. short writingDone = (1 << nbInputs) -1;
  379. workingWin = 0;
  380. for (t_subWindow &w : subWindows)
  381. {
  382. w.cursor = t_Cursor(2, 1);
  383. w.select_up = w.select_down = nullptr;
  384. w.selectFound = w.selectIsLast = false;
  385. wclear(w.innerWin);
  386. writeTopLine(w.fileName,
  387. workingWin == selectedWin ? OutputFlag::SPECIAL_ACTIVEINPUTNAME : OutputFlag::SPECIAL_INPUTNAME);
  388. w.parentsIterators = std::stack<std::pair<int, JSonContainer*> >();
  389. ++workingWin;
  390. }
  391. while (writingDone)
  392. {
  393. // Display Gap (--)
  394. bool restart = false;
  395. workingWin = 0;
  396. for (t_subWindow &w : subWindows)
  397. {
  398. if ((writingDone & (1 << workingWin)) &&
  399. ((!w.parentsIterators.empty() && isAdded(w.parentsIterators.top())) ||
  400. (w.parentsIterators.empty() && isAdded(w.root))))
  401. {
  402. const unsigned int startY = w.cursor.second;
  403. do
  404. {
  405. const Optional<bool> wrote = redrawOneItemToWorkingWin(w, screenSize);
  406. if (wrote.absent())
  407. return false;
  408. if (wrote.get())
  409. {
  410. writingDone &= ~(1 << workingWin);
  411. break;
  412. }
  413. } while (isAdded(w.parentsIterators.top()));
  414. const unsigned int diffY = w.cursor.second - startY;
  415. unsigned int i = 0;
  416. for (t_subWindow &wi: subWindows)
  417. if (i++ != workingWin)
  418. for (unsigned int j = 0; j < diffY; ++j)
  419. displayDiffOp(wi.innerWin, (wi.cursor.second)++, eLevenshteinOperator::rem);
  420. restart = true;
  421. break;
  422. }
  423. ++workingWin;
  424. }
  425. if (restart)
  426. continue;
  427. // Actual display
  428. workingWin = 0;
  429. for (t_subWindow &w : subWindows)
  430. {
  431. if ((writingDone & (1 << workingWin)))
  432. {
  433. const Optional<bool> wrote = redrawOneItemToWorkingWin(w, screenSize);
  434. if (wrote.absent())
  435. return false;
  436. if (wrote.get())
  437. writingDone &= ~(1 << workingWin);
  438. }
  439. ++workingWin;
  440. }
  441. }
  442. for (t_subWindow &w : subWindows)
  443. wrefresh(w.innerWin);
  444. return true;
  445. }
  446. bool CurseSplitOutput::writeContainer(const t_Cursor &maxSize, JSonContainer *item, bool opening)
  447. {
  448. char childDelimiter[2];
  449. t_subWindow &w = subWindows.at(workingWin);
  450. if (dynamic_cast<const JSonObject *>(item))
  451. memcpy(childDelimiter, "{}", sizeof(*childDelimiter) * 2);
  452. else
  453. memcpy(childDelimiter, "[]", sizeof(*childDelimiter) * 2);
  454. if (!opening) // Display close brackets
  455. {
  456. w.cursor.first -= INDENT_LEVEL;
  457. w.cursor.second += write(w.cursor.first, w.cursor.second, childDelimiter[1], maxSize.first, CurseSplitOutput::getFlag(item));
  458. }
  459. else if (collapsed.find((const JSonContainer *)item) != collapsed.end()) // inline collapsed
  460. {
  461. std::string ss;
  462. ss.append(&childDelimiter[0], 1).append(" ... ").append(&childDelimiter[1], 1);
  463. w.cursor.second += write(w.cursor.first, w.cursor.second, ss, 7, maxSize.first, CurseSplitOutput::getFlag(item));
  464. }
  465. else // Display open brackets
  466. {
  467. w.cursor.second += write(w.cursor.first, w.cursor.second, childDelimiter[0], maxSize.first, CurseSplitOutput::getFlag(item));
  468. if (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1)
  469. return false;
  470. w.parentsIterators.push(std::pair<int, JSonContainer *>(-1, item));
  471. w.cursor.first += INDENT_LEVEL;
  472. return true;
  473. }
  474. return (w.cursor.second < w.scrollTop || (w.cursor.second - w.scrollTop) <= maxSize.second -1);
  475. }
  476. bool CurseSplitOutput::writeKey(t_subWindow &w, const std::string &key, const size_t keylen, const t_Cursor &maxSize, OutputFlag flags, unsigned int extraLen)
  477. {
  478. if (w.cursor.second <= w.scrollTop)
  479. {
  480. w.cursor.second++;
  481. return true;
  482. }
  483. char oldType = flags.type();
  484. flags.type(OutputFlag::TYPE_OBJKEY);
  485. w.cursor.second += write(w.cursor.first, w.cursor.second, key, keylen, maxSize.first -extraLen -2, flags);
  486. flags.type(OutputFlag::TYPE_OBJ);
  487. write(": ", flags);
  488. flags.type(oldType);
  489. return (w.cursor.second < w.scrollTop || (w.cursor.second - w.scrollTop) <= maxSize.second);
  490. }
  491. bool CurseSplitOutput::writeKey(t_subWindow &w, const std::string &key, const size_t keylen, const std::string &after, const size_t afterlen, t_Cursor &cursor, const t_Cursor &maxWidth, OutputFlag flags)
  492. {
  493. if (cursor.second <= w.scrollTop)
  494. {
  495. cursor.second++;
  496. return true;
  497. }
  498. char oldType = flags.type();
  499. flags.type(OutputFlag::TYPE_OBJKEY);
  500. write(cursor.first, cursor.second, key, 0, 1, flags);
  501. flags.type(OutputFlag::TYPE_OBJ);
  502. write(": ", flags);
  503. flags.type(oldType);
  504. write(after, flags);
  505. cursor.second += getNbLines(cursor.first +keylen +2 +afterlen, maxWidth.first);
  506. return (cursor.second < w.scrollTop || (cursor.second - w.scrollTop) <= maxWidth.second);
  507. }
  508. bool CurseSplitOutput::writeKey(t_subWindow &w, const std::string &key, const size_t keylen, const std::string &after, t_Cursor &cursor, const t_Cursor &maxWidth, OutputFlag flags)
  509. {
  510. return writeKey(w, key, keylen, after, after.length(), cursor, maxWidth, flags);
  511. }
  512. bool CurseSplitOutput::isAdded(const JSonElement *e) const
  513. {
  514. return diffMatrice->getEquivalence(e) == nullptr;
  515. }
  516. bool CurseSplitOutput::isAdded(const std::pair<int, JSonContainer *> &item) const
  517. {
  518. const JSonElement *e;
  519. if ((unsigned int) (item.first +1) >= item.second->size())
  520. e = item.second;
  521. else
  522. e = list_at<JSonElement*>(*(item.second), item.first +1);
  523. return isAdded(e);
  524. }
  525. bool CurseSplitOutput::redraw(t_subWindow &w, const t_Cursor &maxSize, std::pair<int, JSonContainer *> &item)
  526. {
  527. JSonElement *currentItem;
  528. (item.first)++;
  529. if ((unsigned int) item.first == item.second->size())
  530. {
  531. w.parentsIterators.pop();
  532. return writeContainer(maxSize, item.second, false);
  533. }
  534. currentItem = list_at<JSonElement*>(*(item.second), item.first);
  535. return redraw(w, maxSize, currentItem);
  536. }
  537. bool CurseSplitOutput::redraw(t_subWindow &w, const t_Cursor &maxSize, JSonElement *item)
  538. {
  539. checkSelection(item);
  540. if (dynamic_cast<const JSonContainer*>(item))
  541. {
  542. if (!writeContainer(maxSize, (JSonContainer *) item))
  543. return false;
  544. }
  545. else if (dynamic_cast<JSonObjectEntry*>(item))
  546. {
  547. return writeObjectEntry(w, maxSize, (JSonObjectEntry*) item);
  548. }
  549. else
  550. {
  551. w.cursor.second += CurseOutput::write(w.cursor.first, w.cursor.second, item, maxSize.first, CurseSplitOutput::getFlag(item));
  552. if (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1)
  553. return false;
  554. }
  555. return true;
  556. }
  557. bool CurseSplitOutput::writeObjectEntry(t_subWindow &w, const t_Cursor &maxSize, JSonObjectEntry *ent)
  558. {
  559. bool isContainer = (dynamic_cast<JSonContainer *>(**ent) != nullptr);
  560. std::string key = ent->stringify();
  561. if (isContainer && collapsed.find((JSonContainer*)(**ent)) != collapsed.cend()) // inline collapsed container
  562. {
  563. if (dynamic_cast<JSonObject *>(**ent))
  564. {
  565. if (!writeKey(w, key, ent->lazystrlen(), "{ ... }", w.cursor, maxSize, getFlag(ent)) || (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1))
  566. return false;
  567. }
  568. else if (!writeKey(w, key, ent->lazystrlen(), "[ ... ]", w.cursor, maxSize, getFlag(ent)) || (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1))
  569. return false;
  570. }
  571. else if (!isContainer) // inline value
  572. {
  573. JSonElement *eContent = **ent;
  574. if (!writeKey(w, key, ent->lazystrlen(), eContent->stringify(), eContent->lazystrlen(), w.cursor, maxSize, getFlag(ent))
  575. || (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1))
  576. return false;
  577. }
  578. else if (((JSonContainer*)(**ent))->size() == 0) // inline empty
  579. {
  580. if (dynamic_cast<const JSonObject *>(**ent) )
  581. {
  582. if (!writeKey(w, key, ent->lazystrlen(), "{ }", w.cursor, maxSize, getFlag(ent)) || (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1))
  583. return false;
  584. }
  585. else if (!writeKey(w, key, ent->lazystrlen(), "[ ]", w.cursor, maxSize, getFlag(ent)) || (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1))
  586. return false;
  587. }
  588. else // Container
  589. {
  590. if (!writeKey(w, key, ent->lazystrlen(), maxSize, getFlag(ent)))
  591. return false;
  592. writeContainer(maxSize, (JSonContainer*)(**ent), true);
  593. }
  594. if (w.cursor.second > w.scrollTop && (w.cursor.second - w.scrollTop) > maxSize.second -1)
  595. return false;
  596. return true;
  597. }
  598. unsigned int CurseSplitOutput::write(const int &x, const int &y, const char item, unsigned int maxWidth, OutputFlag flags)
  599. {
  600. int offsetY = y - subWindows.at(workingWin).scrollTop;
  601. WINDOW *currentWin = subWindows.at(workingWin).innerWin;
  602. char color = OutputFlag::SPECIAL_NONE;
  603. if (offsetY <= 0)
  604. return 1;
  605. if (flags.selected())
  606. wattron(currentWin, A_REVERSE | A_BOLD);
  607. if (flags.searched())
  608. color = OutputFlag::SPECIAL_SEARCH;
  609. else if (colors.find(flags.type()) != colors.end())
  610. color = flags.type();
  611. if (color != OutputFlag::SPECIAL_NONE)
  612. wattron(currentWin, COLOR_PAIR(color));
  613. mvwprintw(currentWin, offsetY, x, "%c", item);
  614. wattroff(currentWin, A_REVERSE | A_BOLD);
  615. if (color != OutputFlag::SPECIAL_NONE)
  616. wattroff(currentWin, COLOR_PAIR(color));
  617. displayDiffOp(currentWin, offsetY, flags.diffOp());
  618. return getNbLines(x +1, maxWidth);
  619. }
  620. void CurseSplitOutput::displayDiffOp(WINDOW *w, const int &y, const eLevenshteinOperator &op) const
  621. {
  622. switch (op)
  623. {
  624. case eLevenshteinOperator::add:
  625. wattron(w, A_REVERSE | A_BOLD);
  626. wattron(w, COLOR_PAIR(OutputFlag::DIFF_ADD));
  627. mvwprintw(w, y, 0, "++");
  628. wattroff(w, COLOR_PAIR(OutputFlag::DIFF_ADD));
  629. wattroff(w, A_REVERSE | A_BOLD);
  630. break;
  631. case eLevenshteinOperator::mod:
  632. wattron(w, A_REVERSE | A_BOLD);
  633. wattron(w, COLOR_PAIR(OutputFlag::DIFF_MOD));
  634. mvwprintw(w, y, 0, "!!");
  635. wattroff(w, COLOR_PAIR(OutputFlag::DIFF_MOD));
  636. wattroff(w, A_REVERSE | A_BOLD);
  637. break;
  638. case eLevenshteinOperator::rem:
  639. wattron(w, A_REVERSE | A_BOLD);
  640. wattron(w, COLOR_PAIR(OutputFlag::DIFF_REM));
  641. mvwprintw(w, y, 0, "--");
  642. wattroff(w, COLOR_PAIR(OutputFlag::DIFF_REM));
  643. wattroff(w, A_REVERSE | A_BOLD);
  644. break;
  645. case eLevenshteinOperator::equ: // skip
  646. break;
  647. }
  648. }
  649. unsigned int CurseSplitOutput::write(const int &x, const int &y, const std::string &str, const size_t strlen, unsigned int maxWidth, const OutputFlag flags)
  650. {
  651. const t_subWindow &w = subWindows.at(workingWin);
  652. WINDOW *currentWin = w.innerWin;
  653. int offsetY = y - w.scrollTop;
  654. if (offsetY <= 0)
  655. return 1;
  656. displayDiffOp(currentWin, y, flags.diffOp());
  657. wmove(subWindows.at(workingWin).innerWin, offsetY, x);
  658. write(str, flags);
  659. return getNbLines(strlen +x, maxWidth);
  660. }
  661. void CurseSplitOutput::write(const std::string &str, const OutputFlag flags) const
  662. {
  663. char color = OutputFlag::SPECIAL_NONE;
  664. WINDOW *currentWin = subWindows.at(workingWin).innerWin;
  665. if (flags.selected())
  666. wattron(currentWin, A_REVERSE | A_BOLD);
  667. if (flags.searched())
  668. color = OutputFlag::SPECIAL_SEARCH;
  669. else if (colors.find(flags.type()) != colors.end())
  670. color = flags.type();
  671. if (color != OutputFlag::SPECIAL_NONE)
  672. wattron(currentWin, COLOR_PAIR(color));
  673. wprintw(currentWin, "%s", str.c_str());
  674. wattroff(currentWin, A_REVERSE | A_BOLD);
  675. if (color != OutputFlag::SPECIAL_NONE)
  676. wattroff(currentWin, COLOR_PAIR(color));
  677. }
  678. void CurseSplitOutput::destroyAllSubWin()
  679. {
  680. for (t_subWindow &w : subWindows)
  681. {
  682. if (w.outerWin)
  683. {
  684. wborder(w.outerWin, ' ', ' ', ' ',' ',' ',' ',' ',' ');
  685. wrefresh(w.outerWin);
  686. delwin(w.outerWin);
  687. w.outerWin = nullptr;
  688. }
  689. if (w.innerWin)
  690. {
  691. delwin(w.innerWin);
  692. w.innerWin = nullptr;
  693. }
  694. }
  695. }
  696. void CurseSplitOutput::writeTopLine(const std::string &buffer, short color) const
  697. {
  698. const t_Cursor screenSize = getScreenSize();
  699. const size_t bufsize = buffer.size();
  700. WINDOW *currentWin = subWindows.at(workingWin).innerWin;
  701. if (params.colorEnabled())
  702. wattron(currentWin, COLOR_PAIR(color));
  703. mvwprintw(currentWin, 0, 0, "%s%*c", buffer.c_str(), screenSize.first - bufsize -2, ' ');
  704. if (params.colorEnabled())
  705. wattroff(currentWin, COLOR_PAIR(color));
  706. }
  707. const t_Cursor CurseSplitOutput::getScreenSize() const
  708. {
  709. t_Cursor result = getScreenSizeUnsafe();
  710. return t_Cursor(result.first / nbInputs, result.second -2);
  711. }
  712. void CurseSplitOutput::shutdown()
  713. {
  714. destroyAllSubWin();
  715. endwin();
  716. delscreen(screen);
  717. if (screen_fd)
  718. {
  719. fclose(screen_fd);
  720. screen_fd = nullptr;
  721. }
  722. screen = nullptr;
  723. }
  724. const OutputFlag CurseSplitOutput::getFlag(const JSonElement *e) const
  725. {
  726. return getFlag(e, subWindows.at(workingWin).selection);
  727. }
  728. const OutputFlag CurseSplitOutput::getFlag(const JSonElement *item, const JSonElement *selection) const
  729. {
  730. OutputFlag res;
  731. const JSonElement *i = dynamic_cast<const JSonObjectEntry*>(item) ? **((const JSonObjectEntry*)item) : item;
  732. res.selected(item == selection
  733. || (item->getParent() && dynamic_cast<const JSonObjectEntry*>(item->getParent()) && selection == item->getParent()));
  734. res.searched(std::find(subWindows.at(selectedWin).searchResults.cbegin(),
  735. subWindows.at(selectedWin).searchResults.cend(),
  736. item) != subWindows.at(selectedWin).searchResults.cend());
  737. try {
  738. res.diffOp(diffMatrice->get(item));
  739. }
  740. catch (std::out_of_range &e) {
  741. res.diffOp(eLevenshteinOperator::add);
  742. }
  743. if (dynamic_cast<const JSonPrimitive<std::string> *>(i))
  744. res.type(OutputFlag::TYPE_STRING);
  745. else if (dynamic_cast<const JSonPrimitive<bool> *>(i))
  746. res.type(OutputFlag::TYPE_BOOL);
  747. else if (dynamic_cast<const JSonPrimitive<Null> *>(i))
  748. res.type(OutputFlag::TYPE_NULL);
  749. else if (dynamic_cast<const AJSonPrimitive *>(i))
  750. res.type(OutputFlag::TYPE_NUMBER);
  751. else if (dynamic_cast<const JSonObject*>(i))
  752. res.type(OutputFlag::TYPE_OBJ);
  753. else if (dynamic_cast<const JSonArray*>(i))
  754. res.type(OutputFlag::TYPE_ARR);
  755. return res;
  756. }