mirror of
https://github.com/erkin/ponysay
synced 2024-11-23 03:43:11 +00:00
153 lines
3.2 KiB
C
153 lines
3.2 KiB
C
/* ponysaytruncater
|
||
* Output truncater used by ponysay to stop
|
||
* large ponies from being printed badly.
|
||
*
|
||
* Licensed under WTFPL
|
||
* See COPYING for details
|
||
*/
|
||
#include <stdio.h>
|
||
#define String char*
|
||
#define boolean char
|
||
#define true 1
|
||
#define false 0
|
||
|
||
/* Stdin file descriptor ID */
|
||
#define STDIN 0
|
||
|
||
/* The number of columns on the current line */
|
||
static int col = 0;
|
||
|
||
/* Escape sequence state */
|
||
static int esc = 0;
|
||
|
||
/* Last bytes as written */
|
||
static boolean ok = true;
|
||
|
||
void write(char b, int width);
|
||
int toInt(String string);
|
||
|
||
|
||
/* Mane method!
|
||
* The only argument, in addition to the executed file,
|
||
* should be the width of the terminal which you get by
|
||
* adding <code>`tput cols || echo 0`</code> as an argument.
|
||
*
|
||
* @param argc The number of startup arguments
|
||
* @param argv The startup arguments, the first is the file itself
|
||
*
|
||
* @author Mattias Andrée, maandree@kth.se
|
||
*/
|
||
void main(int argc, String* argv)
|
||
{
|
||
int width = 0;
|
||
if (argc > 1)
|
||
width = toInt(*(argv + 1));
|
||
|
||
char b = 0;
|
||
|
||
if (width > 15) /* sanity */
|
||
while (read(STDIN, &b, 1))
|
||
write(b, width);
|
||
else
|
||
while (read(STDIN, &b, 1))
|
||
printf("%c", b);
|
||
}
|
||
|
||
/* Writes a character to stdout, iff it fits within the terminal
|
||
*
|
||
* @param b The character (byte) to write
|
||
* @param width The width of the terminal
|
||
*/
|
||
void write(char b, int width)
|
||
{
|
||
int i;
|
||
char tabstop;
|
||
|
||
if (esc == 0)
|
||
{
|
||
if (b == '\n')
|
||
{
|
||
if (col >= width)
|
||
{
|
||
/* Reset background colour */
|
||
write('\e', width);
|
||
write('[', width);
|
||
write('4', width);
|
||
write('9', width);
|
||
write('m', width);
|
||
}
|
||
col = -1;
|
||
}
|
||
else if (b == '\t')
|
||
{
|
||
/* Tab to next pos ≡₈ 0 */
|
||
tabstop = 8 - (col & 7);
|
||
for (i = 0; i < tabstop; i++)
|
||
write(' ', width);
|
||
return; /* (!) */
|
||
}
|
||
else if (b == '\e')
|
||
esc = 1;
|
||
}
|
||
else if (esc == 1)
|
||
{
|
||
if (b == '[') esc = 2; /* CSI: CSI ends with a letter, m is for colour */
|
||
else if (b == ']') esc = 3; /* OSI: OSI P is for palette editing in Linux VT */
|
||
else esc = 10; /* Nothing to see here, move along */
|
||
}
|
||
else if (esc == 2)
|
||
{
|
||
if ((('a' <= b) && (b <= 'z')) || (('A' <= b) && (b <= 'Z')))
|
||
esc = 10;
|
||
}
|
||
else if ((esc == 3) && (b == 'P'))
|
||
{
|
||
esc = ~0;
|
||
}
|
||
else if (esc < 0)
|
||
{
|
||
esc--;
|
||
if (esc == ~7)
|
||
esc = 10;
|
||
}
|
||
else
|
||
esc = 10;
|
||
|
||
if (
|
||
/* Can be printed:
|
||
within bounds ∨
|
||
∨ escape sequence ∨
|
||
∨ last with printed ∧ not first byte in character */
|
||
(col < width) ||
|
||
(esc != 0) ||
|
||
(ok && ((b & 0xC0) == 0x80)))
|
||
{
|
||
printf("%c", b);
|
||
if ((esc == 0) && ((b & 0xC0) != 0x80))
|
||
/* Count up columns of not in escape sequnce and */
|
||
col++; /* the byte is not the first byte in the character */
|
||
ok = true;
|
||
}
|
||
else
|
||
ok = false;
|
||
|
||
if (esc == 10)
|
||
esc = 0;
|
||
}
|
||
|
||
/* Converts a string to an integer
|
||
*
|
||
* @param string The string to convert
|
||
* @return The integer represented by the string
|
||
*/
|
||
int toInt(String string)
|
||
{
|
||
int rc = 0;
|
||
String str = string;
|
||
char c = 0;
|
||
|
||
while ((c = *str++) != 0)
|
||
rc = (rc << 1) + (rc << 3) - (c & 15);
|
||
|
||
return -rc;
|
||
}
|