One Level Up Top Level

runtime/lib_ansi.c - ktap

Global variables defined

Functions defined

Source code

  1. /*
  2. * lib_ansi.c - ANSI escape sequences library
  3. *
  4. * http://en.wikipedia.org/wiki/ANSI_escape_code
  5. *
  6. * This file is part of ktap by Jovi Zhangwei.
  7. *
  8. * Copyright (C) 2012-2013 Jovi Zhangwei <jovi.zhangwei@gmail.com>.
  9. *
  10. * ktap is free software; you can redistribute it and/or modify it
  11. * under the terms and conditions of the GNU General Public License,
  12. * version 2, as published by the Free Software Foundation.
  13. *
  14. * ktap is distributed in the hope it will be useful, but WITHOUT
  15. * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  16. * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  17. * more details.
  18. *
  19. * You should have received a copy of the GNU General Public License along with
  20. * this program; if not, write to the Free Software Foundation, Inc.,
  21. * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
  22. */

  24. #include "../include/ktap_types.h"
  25. #include "ktap.h"
  26. #include "kp_vm.h"

  28. /**
  29. * function ansi.clear_screen - Move cursor to top left and clear screen.
  30. *
  31. * Description: Sends ansi code for moving cursor to top left and then the
  32. * ansi code for clearing the screen from the cursor position to the end.
  33. */

  35. ‌static int kplib_ansi_clear_screen(ktap_state_t *ks)
  36. {
  37.     kp_printf(ks, "\033[1;1H\033[J");
  38.     return 0;
  39. }

  41. /**
  42. * function ansi.set_color - Set the ansi Select Graphic Rendition mode.
  43. * @fg: Foreground color to set.
  44. *
  45. * Description: Sends ansi code for Select Graphic Rendition mode for the
  46. * given forground color. Black (30), Blue (34), Green (32), Cyan (36),
  47. * Red (31), Purple (35), Brown (33), Light Gray (37).
  48. */

  50. ‌static int kplib_ansi_set_color(ktap_state_t *ks)
  51. {
  52.     int fg = kp_arg_checknumber(ks, 1);

  54.     kp_printf(ks, "\033[%dm", fg);
  55.     return 0;
  56. }

  58. /**
  59. * function ansi.set_color2 - Set the ansi Select Graphic Rendition mode.
  60. * @fg: Foreground color to set.
  61. * @bg: Background color to set.
  62. *
  63. * Description: Sends ansi code for Select Graphic Rendition mode for the
  64. * given forground color, Black (30), Blue (34), Green (32), Cyan (36),
  65. * Red (31), Purple (35), Brown (33), Light Gray (37) and the given
  66. * background color, Black (40), Red (41), Green (42), Yellow (43),
  67. * Blue (44), Magenta (45), Cyan (46), White (47).
  68. */
  69. ‌static int kplib_ansi_set_color2(ktap_state_t *ks)
  70. {
  71.     int fg = kp_arg_checknumber(ks, 1);
  72.     int bg = kp_arg_checknumber(ks, 2);

  74.     kp_printf(ks, "\033[%d;%dm", fg, bg);
  75.     return 0;
  76. }

  78. /**
  79. * function ansi.set_color3 - Set the ansi Select Graphic Rendition mode.
  80. * @fg: Foreground color to set.
  81. * @bg: Background color to set.
  82. * @attr: Color attribute to set.
  83. *
  84. * Description: Sends ansi code for Select Graphic Rendition mode for the
  85. * given forground color, Black (30), Blue (34), Green (32), Cyan (36),
  86. * Red (31), Purple (35), Brown (33), Light Gray (37), the given
  87. * background color, Black (40), Red (41), Green (42), Yellow (43),
  88. * Blue (44), Magenta (45), Cyan (46), White (47) and the color attribute
  89. * All attributes off (0), Intensity Bold (1), Underline Single (4),
  90. * Blink Slow (5), Blink Rapid (6), Image Negative (7).
  91. */
  92. ‌static int kplib_ansi_set_color3(ktap_state_t *ks)
  93. {
  94.     int fg = kp_arg_checknumber(ks, 1);
  95.     int bg = kp_arg_checknumber(ks, 2);
  96.     int attr = kp_arg_checknumber(ks, 3);

  98.     if (attr)
  99.         kp_printf(ks, "\033[%d;%d;%dm", fg, bg, attr);
  100.     else
  101.         kp_printf(ks, "\033[%d;%dm", fg, bg);

  103.     return 0;
  104. }

  106. /**
  107. * function ansi.reset_color - Resets Select Graphic Rendition mode.
  108. *
  109. * Description: Sends ansi code to reset foreground, background and color
  110. * attribute to default values.
  111. */
  112. ‌static int kplib_ansi_reset_color(ktap_state_t *ks)
  113. {
  114.     kp_printf(ks, "\033[0;0m");
  115.     return 0;
  116. }

  118. /**
  119. * function ansi.new_line - Move cursor to new line.
  120. *
  121. * Description: Sends ansi code new line.
  122. */
  123. ‌static int kplib_ansi_new_line (ktap_state_t *ks)
  124. {
  125.     kp_printf(ks, "\12");
  126.     return 0;
  127. }

  129. ‌static const ktap_libfunc_t ansi_lib_funcs[] = {
  130.     {"clear_screen", kplib_ansi_clear_screen},
  131.     {"set_color", kplib_ansi_set_color},
  132.     {"set_color2", kplib_ansi_set_color2},
  133.     {"set_color3", kplib_ansi_set_color3},
  134.     {"reset_color", kplib_ansi_reset_color},
  135.     {"new_line", kplib_ansi_new_line},
  136.     {NULL}
  137. };

  139. ‌int kp_lib_init_ansi(ktap_state_t *ks)
  140. {
  141.     return kp_vm_register_lib(ks, "ansi", ansi_lib_funcs);
  142. }

One Level Up Top Level