1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922 |
- /* vim: tabstop=4 shiftwidth=4 noexpandtab
- * This file is part of ToaruOS and is released under the terms
- * of the NCSA / University of Illinois License - see LICENSE.md
- * Copyright (C) 2012-2018 K. Lange
- *
- * Alternatively, this source file is also released under the
- * following terms:
- *
- * Permission to use, copy, modify, and/or distribute this software for any
- * purpose with or without fee is hereby granted, provided that the above
- * copyright notice and this permission notice appear in all copies.
- *
- * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
- * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
- * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
- * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
- * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
- * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
- *
- * bim - Text editor
- *
- * Bim is inspired by vim, and its name is short for "Bad IMitation".
- *
- * Bim supports syntax highlighting, extensive editing, line selection
- * and copy-paste, and can be built for ToaruOS and Linux (and should be
- * easily portable to other Unix-like environments).
- *
- * Future goals:
- * - History stack
- * - Character selection
- */
- #define _XOPEN_SOURCE 500
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <stdint.h>
- #include <stdarg.h>
- #include <unistd.h>
- #include <termios.h>
- #include <signal.h>
- #include <locale.h>
- #include <wchar.h>
- #include <ctype.h>
- #include <dirent.h>
- #include <sys/types.h>
- #include <sys/ioctl.h>
- #include <sys/stat.h>
- #ifdef __toaru__
- #include <sys/fswait.h>
- #include <toaru/decodeutf8.h>
- #else
- #include <poll.h>
- #include "../base/usr/include/toaru/decodeutf8.h"
- #endif
- #define BLOCK_SIZE 4096
- #define ENTER_KEY '\n'
- #define BACKSPACE_KEY 0x08
- #define DELETE_KEY 0x7F
- /**
- * Theming data
- */
- const char * COLOR_FG = "5;230";
- const char * COLOR_BG = "5;235";
- const char * COLOR_ALT_FG = "5;244";
- const char * COLOR_ALT_BG = "5;236";
- const char * COLOR_NUMBER_BG = "5;232";
- const char * COLOR_NUMBER_FG = "5;101";
- const char * COLOR_STATUS_FG = "5;230";
- const char * COLOR_STATUS_BG = "5;238";
- const char * COLOR_TABBAR_BG = "5;230";
- const char * COLOR_TAB_BG = "5;248";
- const char * COLOR_ERROR_FG = "5;15";
- const char * COLOR_ERROR_BG = "5;196";
- const char * COLOR_SEARCH_FG = "5;234";
- const char * COLOR_SEARCH_BG = "5;226";
- const char * COLOR_KEYWORD = "5;117";
- const char * COLOR_STRING = "5;113";
- const char * COLOR_COMMENT = "5;102;3";
- const char * COLOR_TYPE = "5;185";
- const char * COLOR_PRAGMA = "5;173";
- const char * COLOR_NUMERAL = "5;173";
- #define FLAG_NONE 0
- #define FLAG_KEYWORD 1
- #define FLAG_STRING 2
- #define FLAG_COMMENT 3
- #define FLAG_TYPE 4
- #define FLAG_PRAGMA 5
- #define FLAG_NUMERAL 6
- #define FLAG_SELECT 7
- #define FLAG_NORM_MAX 15
- #define FLAG_COMMENT_ML 16
- #define FLAG_STRING_ML1 17
- #define FLAG_STRING_ML2 18
- void load_colorscheme_wombat(void) {
- /* Based on the wombat256 theme for vim */
- COLOR_FG = "5;230";
- COLOR_BG = "5;235";
- COLOR_ALT_FG = "5;244";
- COLOR_ALT_BG = "5;236";
- COLOR_NUMBER_BG = "5;232";
- COLOR_NUMBER_FG = "5;101";
- COLOR_STATUS_FG = "5;230";
- COLOR_STATUS_BG = "5;238";
- COLOR_TABBAR_BG = "5;230";
- COLOR_TAB_BG = "5;248";
- COLOR_KEYWORD = "5;117";
- COLOR_STRING = "5;113";
- COLOR_COMMENT = "5;102;3";
- COLOR_TYPE = "5;185";
- COLOR_PRAGMA = "5;173";
- COLOR_NUMERAL = COLOR_PRAGMA;
- COLOR_ERROR_FG = "5;15";
- COLOR_ERROR_BG = "5;196";
- COLOR_SEARCH_FG = "5;234";
- COLOR_SEARCH_BG = "5;226";
- }
- void load_colorscheme_citylights(void) {
- /* "City Lights" based on citylights.xyz */
- COLOR_FG = "2;113;140;161";
- COLOR_BG = "2;29;37;44";
- COLOR_ALT_FG = "2;45;55;65";
- COLOR_ALT_BG = "2;33;42;50";
- COLOR_NUMBER_FG = "2;71;89;103";
- COLOR_NUMBER_BG = "2;37;47;56";
- COLOR_STATUS_FG = "2;116;144;166";
- COLOR_STATUS_BG = "2;53;67;78";
- COLOR_TABBAR_BG = "2;37;47;56";
- COLOR_TAB_BG = "2;29;37;44";
- COLOR_KEYWORD = "2;94;196;255";
- COLOR_STRING = "2;83;154;252";
- COLOR_COMMENT = "2;107;133;153;3";
- COLOR_TYPE = "2;139;212;156";
- COLOR_PRAGMA = "2;0;139;148";
- COLOR_NUMERAL = "2;207;118;132";
- COLOR_ERROR_FG = "5;15";
- COLOR_ERROR_BG = "5;196";
- COLOR_SEARCH_FG = "5;234";
- COLOR_SEARCH_BG = "5;226";
- }
- void load_colorscheme_solarized_dark(void) {
- /* Solarized Dark, popular theme */
- COLOR_FG = "2;147;161;161";
- COLOR_BG = "2;0;43;54";
- COLOR_ALT_FG = "2;147;161;161";
- COLOR_ALT_BG = "2;7;54;66";
- COLOR_NUMBER_FG = "2;131;148;149";
- COLOR_NUMBER_BG = "2;7;54;66";
- COLOR_STATUS_FG = "2;131;148;150";
- COLOR_STATUS_BG = "2;7;54;66";
- COLOR_TABBAR_BG = "2;7;54;66";
- COLOR_TAB_BG = "2;131;148;150";
- COLOR_KEYWORD = "2;133;153;0";
- COLOR_STRING = "2;42;161;152";
- COLOR_COMMENT = "2;101;123;131";
- COLOR_TYPE = "2;181;137;0";
- COLOR_PRAGMA = "2;203;75;22";
- COLOR_NUMERAL = "2;220;50;47";
- COLOR_ERROR_FG = "5;15";
- COLOR_ERROR_BG = "5;196";
- COLOR_SEARCH_FG = "5;234";
- COLOR_SEARCH_BG = "5;226";
- }
- void load_colorscheme_ansi(void) {
- /* 16-color theme */
- COLOR_FG = "@15";
- COLOR_BG = "@0";
- COLOR_ALT_FG = "@8";
- COLOR_ALT_BG = "@0";
- COLOR_NUMBER_FG = "@3";
- COLOR_NUMBER_BG = "@0";
- COLOR_STATUS_FG = "@15";
- COLOR_STATUS_BG = "@8";
- COLOR_TABBAR_BG = "@8";
- COLOR_TAB_BG = "@0";
- COLOR_KEYWORD = "@12";
- COLOR_STRING = "@2";
- COLOR_COMMENT = "@7";
- COLOR_TYPE = "@6";
- COLOR_PRAGMA = "@1";
- COLOR_NUMERAL = "@5";
- COLOR_ERROR_FG = "@15";
- COLOR_ERROR_BG = "@1";
- COLOR_SEARCH_FG = "@0";
- COLOR_SEARCH_BG = "@11";
- }
- struct theme_def {
- const char * name;
- void (*load)(void);
- } themes[] = {
- {"wombat", load_colorscheme_wombat},
- {"citylights", load_colorscheme_citylights},
- {"solarized-dark", load_colorscheme_solarized_dark},
- {"ansi", load_colorscheme_ansi},
- {NULL, NULL}
- };
- /**
- * Convert syntax hilighting flag to color code
- */
- const char * flag_to_color(int flag) {
- switch (flag) {
- case FLAG_KEYWORD:
- return COLOR_KEYWORD;
- case FLAG_STRING:
- case FLAG_STRING_ML1:
- case FLAG_STRING_ML2:
- return COLOR_STRING;
- case FLAG_COMMENT:
- case FLAG_COMMENT_ML:
- return COLOR_COMMENT;
- case FLAG_TYPE:
- return COLOR_TYPE;
- case FLAG_NUMERAL:
- return COLOR_NUMERAL;
- case FLAG_PRAGMA:
- return COLOR_PRAGMA;
- case FLAG_SELECT:
- return COLOR_BG;
- default:
- return COLOR_FG;
- }
- }
- /**
- * Line buffer definitions
- *
- * Lines are essentially resizable vectors of char_t structs,
- * which represent single codepoints in the file.
- */
- typedef struct {
- uint32_t display_width:5;
- uint32_t flags:6;
- uint32_t codepoint:21;
- } __attribute__((packed)) char_t;
- /**
- * Lines have available and actual lengths, describing
- * how much space was allocated vs. how much is being
- * used at the moment.
- */
- typedef struct {
- int available;
- int actual;
- int istate;
- char_t text[0];
- } line_t;
- /**
- * Global configuration state
- */
- struct {
- /* Terminal size */
- int term_width, term_height;
- int bottom_size;
- /* Command-line parameters */
- int hilight_on_open;
- int initial_file_is_read_only;
- line_t ** yanks;
- size_t yank_count;
- int tty_in;
- } global_config = {
- 0, /* term_width */
- 0, /* term_height */
- 2, /* bottom_size */
- 1, /* hilight_on_open */
- 0, /* initial_file_is_read_only */
- NULL, /* yanks */
- 0, /* yank_count */
- STDIN_FILENO
- };
- void redraw_line(int j, int x);
- /**
- * Special implementation of getch with a timeout
- */
- int _bim_unget = -1;
- void bim_unget(int c) {
- _bim_unget = c;
- }
- int bim_getch(void) {
- if (_bim_unget != -1) {
- int out = _bim_unget;
- _bim_unget = -1;
- return out;
- }
- #ifdef __toaru__
- int fds[] = {global_config.tty_in};
- int index = fswait2(1,fds,200);
- if (index == 0) {
- unsigned char buf[1];
- read(global_config.tty_in, buf, 1);
- return buf[0];
- } else {
- return -1;
- }
- #else
- struct pollfd fds[1];
- fds[0].fd = global_config.tty_in;
- fds[0].events = POLLIN;
- int ret = poll(fds,1,200);
- if (ret > 0 && fds[0].revents & POLLIN) {
- unsigned char buf[1];
- read(global_config.tty_in, buf, 1);
- return buf[0];
- } else {
- return -1;
- }
- #endif
- }
- /**
- * Buffer data
- *
- * A buffer describes a file, and stores
- * its name as well as the editor state
- * (cursor offsets, etc.) and the actual
- * line buffers.
- */
- typedef struct _env {
- short loading:1;
- short tabs:1;
- short modified:1;
- short readonly:1;
- short mode;
- short tabstop;
- char * file_name;
- int offset;
- int coffset;
- int line_no;
- int line_count;
- int line_avail;
- int col_no;
- char * search;
- struct syntax_definition * syntax;
- line_t ** lines;
- } buffer_t;
- /**
- * Pointer to current active buffer
- */
- buffer_t * env;
- /**
- * Editor modes (like in vim)
- */
- #define MODE_NORMAL 0
- #define MODE_INSERT 1
- #define MODE_LINE_SELECTION 2
- /**
- * Available buffers
- */
- int buffers_len;
- int buffers_avail;
- buffer_t ** buffers;
- /**
- * Create a new buffer
- */
- buffer_t * buffer_new(void) {
- if (buffers_len == buffers_avail) {
- /* If we are out of buffer space, expand the buffers vector */
- buffers_avail *= 2;
- buffers = realloc(buffers, sizeof(buffer_t *) * buffers_avail);
- }
- /* Allocate a new buffer */
- buffers[buffers_len] = malloc(sizeof(buffer_t));
- memset(buffers[buffers_len], 0x00, sizeof(buffer_t));
- buffers_len++;
- return buffers[buffers_len-1];
- }
- /**
- * Close a buffer
- */
- buffer_t * buffer_close(buffer_t * buf) {
- int i;
- /* Locate the buffer in the buffer pointer vector */
- for (i = 0; i < buffers_len; i++) {
- if (buf == buffers[i])
- break;
- }
- /* Invalid buffer? */
- if (i == buffers_len) {
- return env; /* wtf */
- }
- /* Remove the buffer from the vector, moving others up */
- if (i != buffers_len - 1) {
- memmove(&buffers[i], &buffers[i+1], sizeof(*buffers) * (buffers_len - i));
- }
- /* There is one less buffer */
- buffers_len--;
- if (!buffers_len) {
- /* There are no more buffers. */
- return NULL;
- }
- /* If this was the last buffer, return the previous last buffer */
- if (i == buffers_len) {
- return buffers[buffers_len-1];
- }
- /* Otherwise return the new last buffer */
- return buffers[i];
- }
- /**
- * Syntax definition for C
- */
- int syn_c_iskeywordchar(int c) {
- if (isalnum(c)) return 1;
- if (c == '_') return 1;
- return 0;
- }
- static char * syn_c_keywords[] = {
- "while","if","for","continue","return","break","switch","case","sizeof",
- "struct","union","typedef","do","default","else","goto",
- "alignas","alignof","offsetof",
- NULL
- };
- static char * syn_c_types[] = {
- "static","int","char","short","float","double","void","unsigned","volatile","const",
- "register","long","inline","restrict","enum","auto","extern","bool","complex",
- "uint8_t","uint16_t","uint32_t","uint64_t",
- "int8_t","int16_t","int32_t","int64_t",
- NULL
- };
- static int syn_c_extended(line_t * line, int i, int c, int last, int * out_left) {
- if (i == 0 && c == '#') {
- *out_left = line->actual+1;
- return FLAG_PRAGMA;
- }
- if ((!last || !syn_c_iskeywordchar(last)) && (i < line->actual - 3) &&
- line->text[i].codepoint == 'N' &&
- line->text[i+1].codepoint == 'U' &&
- line->text[i+2].codepoint == 'L' &&
- line->text[i+3].codepoint == 'L' &&
- (i == line->actual - 4 || !syn_c_iskeywordchar(line->text[i+4].codepoint))) {
- *out_left = 3;
- return FLAG_NUMERAL;
- }
- if ((!last || !syn_c_iskeywordchar(last)) && isdigit(c)) {
- if (c == '0' && i < line->actual - 1 && line->text[i+1].codepoint == 'x') {
- int j = 2;
- for (; i + j < line->actual && isxdigit(line->text[i+j].codepoint); ++j);
- if (i + j < line->actual && syn_c_iskeywordchar(line->text[i+j].codepoint)) {
- return FLAG_NONE;
- }
- *out_left = j - 1;
- return FLAG_NUMERAL;
- } else {
- int j = 1;
- while (i + j < line->actual && isdigit(line->text[i+j].codepoint)) {
- j++;
- }
- if (i + j < line->actual && syn_c_iskeywordchar(line->text[i+j].codepoint)) {
- return FLAG_NONE;
- }
- *out_left = j - 1;
- return FLAG_NUMERAL;
- }
- }
- if (c == '/') {
- if (i < line->actual - 1 && line->text[i+1].codepoint == '/') {
- *out_left = (line->actual + 1) - i;
- return FLAG_COMMENT;
- }
- if (i < line->actual - 1 && line->text[i+1].codepoint == '*') {
- int last = 0;
- for (int j = i + 2; j < line->actual; ++j) {
- int c = line->text[j].codepoint;
- if (c == '/' && last == '*') {
- *out_left = j - i;
- return FLAG_COMMENT;
- }
- last = c;
- }
- /* TODO multiline - update next */
- *out_left = (line->actual + 1) - i;
- return FLAG_COMMENT_ML;
- }
- }
- if (c == '\'') {
- if (i < line->actual - 3 && line->text[i+1].codepoint == '\\' &&
- line->text[i+3].codepoint == '\'') {
- *out_left = 3;
- return FLAG_NUMERAL;
- }
- if (i < line->actual - 2 && line->text[i+2].codepoint == '\'') {
- *out_left = 2;
- return FLAG_NUMERAL;
- }
- }
- if (c == '"') {
- int last = 0;
- for (int j = i+1; j < line->actual; ++j) {
- int c = line->text[j].codepoint;
- if (last != '\\' && c == '"') {
- *out_left = j - i;
- return FLAG_STRING;
- }
- if (last == '\\' && c == '\\') {
- last = 0;
- }
- last = c;
- }
- *out_left = (line->actual + 1) - i; /* unterminated string */
- return FLAG_STRING;
- }
- return 0;
- }
- char * syn_c_ext[] = {".c",".h",NULL};
- static int syn_c_finish(line_t * line, int * left, int state) {
- if (state == FLAG_COMMENT_ML) {
- int last = 0;
- for (int i = 0; i < line->actual; ++i) {
- if (line->text[i].codepoint == '/' && last == '*') {
- *left = i+2;
- return FLAG_COMMENT;
- }
- last = line->text[i].codepoint;
- }
- return FLAG_COMMENT_ML;
- }
- return 0;
- }
- /**
- * Syntax definition for Python
- */
- static char * syn_py_keywords[] = {
- "class","def","return","del","if","else","elif",
- "for","while","continue","break","assert",
- "as","and","or","except","finally","from",
- "global","import","in","is","lambda","with",
- "nonlocal","not","pass","raise","try","yield",
- NULL
- };
- static char * syn_py_types[] = {
- "True","False","None",
- "object","set","dict","int","str","bytes",
- NULL
- };
- static int syn_py_extended(line_t * line, int i, int c, int last, int * out_left) {
- if (i == 0 && c == 'i') {
- /* Check for import */
- char * import = "import ";
- for (int j = 0; j < line->actual + 1; ++j) {
- if (import[j] == '\0') {
- *out_left = j - 2;
- return FLAG_PRAGMA;
- }
- if (line->text[j].codepoint != import[j]) break;
- }
- }
- if (c == '#') {
- *out_left = (line->actual + 1) - i;
- return FLAG_COMMENT;
- }
- if (c == '@') {
- for (int j = i+1; j < line->actual + 1; ++j) {
- if (!syn_c_iskeywordchar(line->text[j].codepoint)) {
- *out_left = j - i - 1;
- return FLAG_PRAGMA;
- }
- *out_left = (line->actual + 1) - i;
- return FLAG_PRAGMA;
- }
- }
- if ((!last || !syn_c_iskeywordchar(last)) && isdigit(c)) {
- if (c == '0' && i < line->actual - 1 && line->text[i+1].codepoint == 'x') {
- int j = 2;
- for (; i + j < line->actual && isxdigit(line->text[i+j].codepoint); ++j);
- if (i + j < line->actual && syn_c_iskeywordchar(line->text[i+j].codepoint)) {
- return FLAG_NONE;
- }
- *out_left = j - 1;
- return FLAG_NUMERAL;
- } else {
- int j = 1;
- while (i + j < line->actual && isdigit(line->text[i+j].codepoint)) {
- j++;
- }
- if (i + j < line->actual && syn_c_iskeywordchar(line->text[i+j].codepoint)) {
- return FLAG_NONE;
- }
- *out_left = j - 1;
- return FLAG_NUMERAL;
- }
- }
- if (line->text[i].codepoint == '\'') {
- if (i + 2 < line->actual && line->text[i+1].codepoint == '\'' && line->text[i+2].codepoint == '\'') {
- /* Begin multiline */
- for (int j = i + 3; j < line->actual - 2; ++j) {
- if (line->text[j].codepoint == '\'' &&
- line->text[j+1].codepoint == '\'' &&
- line->text[j+2].codepoint == '\'') {
- *out_left = (j+2) - i;
- return FLAG_STRING;
- }
- }
- return FLAG_STRING_ML1;
- }
- int last = 0;
- for (int j = i+1; j < line->actual; ++j) {
- int c = line->text[j].codepoint;
- if (last != '\\' && c == '\'') {
- *out_left = j - i;
- return FLAG_STRING;
- }
- if (last == '\\' && c == '\\') {
- last = 0;
- }
- last = c;
- }
- *out_left = (line->actual + 1) - i; /* unterminated string */
- return FLAG_STRING;
- }
- if (line->text[i].codepoint == '"') {
- if (i + 2 < line->actual && line->text[i+1].codepoint == '"' && line->text[i+2].codepoint == '"') {
- /* Begin multiline */
- for (int j = i + 3; j < line->actual - 2; ++j) {
- if (line->text[j].codepoint == '"' &&
- line->text[j+1].codepoint == '"' &&
- line->text[j+2].codepoint == '"') {
- *out_left = (j+2) - i;
- return FLAG_STRING;
- }
- }
- return FLAG_STRING_ML2;
- }
- int last = 0;
- for (int j = i+1; j < line->actual; ++j) {
- int c = line->text[j].codepoint;
- if (last != '\\' && c == '"') {
- *out_left = j - i;
- return FLAG_STRING;
- }
- if (last == '\\' && c == '\\') {
- last = 0;
- }
- last = c;
- }
- *out_left = (line->actual + 1) - i; /* unterminated string */
- return FLAG_STRING;
- }
- return 0;
- }
- static int syn_py_finish(line_t * line, int * left, int state) {
- /* TODO support multiline quotes */
- if (state == FLAG_STRING_ML1) {
- for (int j = 0; j < line->actual - 2; ++j) {
- if (line->text[j].codepoint == '\'' &&
- line->text[j+1].codepoint == '\'' &&
- line->text[j+2].codepoint == '\'') {
- *left = (j+3);
- return FLAG_STRING;
- }
- }
- return FLAG_STRING_ML1;
- }
- if (state == FLAG_STRING_ML2) {
- for (int j = 0; j < line->actual - 2; ++j) {
- if (line->text[j].codepoint == '"' &&
- line->text[j+1].codepoint == '"' &&
- line->text[j+2].codepoint == '"') {
- *left = (j+3);
- return FLAG_STRING;
- }
- }
- return FLAG_STRING_ML2;
- }
- return 0;
- }
- char * syn_py_ext[] = {".py",NULL};
- /**
- * Syntax definition for ToaruOS shell
- */
- static char * syn_sh_keywords[] = {
- "cd","exit","export","help","history","if",
- "empty?","equals?","return","export-cmd",
- "source","exec","not","while","then","else",
- NULL,
- };
- static char * syn_sh_types[] = {NULL};
- static int syn_sh_extended(line_t * line, int i, int c, int last, int * out_left) {
- (void)last;
- if (c == '#') {
- *out_left = (line->actual + 1) - i;
- return FLAG_COMMENT;
- }
- if (line->text[i].codepoint == '\'') {
- int last = 0;
- for (int j = i+1; j < line->actual + 1; ++j) {
- int c = line->text[j].codepoint;
- if (last != '\\' && c == '\'') {
- *out_left = j - i;
- return FLAG_STRING;
- }
- if (last == '\\' && c == '\\') {
- last = 0;
- }
- last = c;
- }
- *out_left = (line->actual + 1) - i; /* unterminated string */
- return FLAG_STRING;
- }
- if (line->text[i].codepoint == '"') {
- int last = 0;
- for (int j = i+1; j < line->actual + 1; ++j) {
- int c = line->text[j].codepoint;
- if (last != '\\' && c == '"') {
- *out_left = j - i;
- return FLAG_STRING;
- }
- if (last == '\\' && c == '\\') {
- last = 0;
- }
- last = c;
- }
- *out_left = (line->actual + 1) - i; /* unterminated string */
- return FLAG_STRING;
- }
- return 0;
- }
- static int syn_sh_iskeywordchar(int c) {
- if (isalnum(c)) return 1;
- if (c == '-') return 1;
- if (c == '_') return 1;
- if (c == '?') return 1;
- return 0;
- }
- static char * syn_sh_ext[] = {".sh",".eshrc",".esh",NULL};
- static int syn_sh_finish(line_t * line, int * left, int state) {
- /* No multilines supported */
- (void)line;
- (void)left;
- (void)state;
- return 0;
- }
- /**
- * Syntax hilighting definition database
- */
- struct syntax_definition {
- char * name;
- char ** ext;
- char ** keywords;
- char ** types;
- int (*extended)(line_t *, int, int, int, int *);
- int (*iskwchar)(int);
- int (*finishml)(line_t *, int *, int);
- } syntaxes[] = {
- {"c",syn_c_ext,syn_c_keywords,syn_c_types,syn_c_extended,syn_c_iskeywordchar,syn_c_finish},
- {"python",syn_py_ext,syn_py_keywords,syn_py_types,syn_py_extended,syn_c_iskeywordchar,syn_py_finish},
- {"esh",syn_sh_ext,syn_sh_keywords,syn_sh_types,syn_sh_extended,syn_sh_iskeywordchar,syn_sh_finish},
- {NULL}
- };
- /**
- * Checks whether the character pointed to by `c` is the start of a match for
- * keyword or type name `str`.
- */
- int check_line(line_t * line, int c, char * str, int last) {
- if (env->syntax->iskwchar(last)) return 0;
- for (int i = c; i < line->actual; ++i, ++str) {
- if (*str == '\0' && !env->syntax->iskwchar(line->text[i].codepoint)) return 1;
- if (line->text[i].codepoint == *str) continue;
- return 0;
- }
- if (*str == '\0') return 1;
- return 0;
- }
- /**
- * Calculate syntax hilighting for the given line.
- */
- void recalculate_syntax(line_t * line, int offset) {
- if (!env->syntax) {
- for (int i = 0; i < line->actual; ++i) {
- line->text[i].flags = 0;
- }
- return;
- }
- /* Start from the line's stored in initial state */
- int state = line->istate;
- int left = 0;
- int last = 0;
- if (state) {
- /*
- * If we are already highlighting coming in, then we need to check
- * for a finishing sequence for the curent state.
- */
- state = env->syntax->finishml(line,&left,state);
- if (state > FLAG_NORM_MAX) {
- /* The finish check said that this multiline state continues. */
- for (int i = 0; i < line->actual; i++) {
- /* Set the entire line to draw with this state */
- line->text[i].flags = state;
- }
- /* Recalculate later lines if needed */
- goto _multiline;
- }
- }
- for (int i = 0; i < line->actual; last = line->text[i++].codepoint) {
- if (!left) state = 0;
- if (state) {
- /* Currently hilighting, have `left` characters remaining with this state */
- left--;
- line->text[i].flags = state;
- if (!left) {
- /* Done hilighting this state, go back to parsing on next character */
- state = 0;
- }
- /* If we are hilighting something, don't parse */
- continue;
- }
- int c = line->text[i].codepoint;
- line->text[i].flags = FLAG_NONE;
- /* Language-specific syntax hilighting */
- if (env->syntax->extended) {
- int s = env->syntax->extended(line,i,c,last,&left);
- if (s) {
- state = s;
- if (state > FLAG_NORM_MAX) {
- /* A multiline state was returned. Fill the rest of the line */
- for (; i < line->actual; i++) {
- line->text[i].flags = state;
- }
- /* And recalculate later lines if needed */
- goto _multiline;
- }
- goto _continue;
- }
- }
- /* Keywords */
- for (char ** kw = env->syntax->keywords; *kw; kw++) {
- int c = check_line(line, i, *kw, last);
- if (c == 1) {
- left = strlen(*kw)-1;
- state = FLAG_KEYWORD;
- goto _continue;
- }
- }
- /* Type names */
- for (char ** kw = env->syntax->types; *kw; kw++) {
- int c = check_line(line, i, *kw, last);
- if (c == 1) {
- left = strlen(*kw)-1;
- state = FLAG_TYPE;
- goto _continue;
- }
- }
- _continue:
- line->text[i].flags = state;
- }
- state = 0;
- _multiline:
- /*
- * If the next line's initial state does not match the state we ended on,
- * then it needs to be recalculated (and redraw). This may lead to multiple
- * recursive calls until a match is found.
- */
- if (offset + 1 < env->line_count && env->lines[offset+1]->istate != state) {
- /* Set the next line's initial state to our ending state */
- env->lines[offset+1]->istate = state;
- /* Recursively recalculate */
- recalculate_syntax(env->lines[offset+1],offset+1);
- /*
- * Determine if this is an on-screen line so we can redraw it;
- * this ends up drawing from bottom to top when multiple lines
- * need to be redrawn by a recursive call.
- */
- if (offset+1 >= env->offset && offset+1 < env->offset + global_config.term_height - global_config.bottom_size - 1) {
- redraw_line(offset + 1 - env->offset,offset+1);
- }
- }
- }
- /**
- * Recalculate tab widths.
- */
- void recalculate_tabs(line_t * line) {
- if (env->loading) return;
- int j = 0;
- for (int i = 0; i < line->actual; ++i) {
- if (line->text[i].codepoint == '\t') {
- line->text[i].display_width = env->tabstop - (j % env->tabstop);
- }
- j += line->text[i].display_width;
- }
- }
- /**
- * TODO:
- *
- * The line editing functions should probably take a buffer_t *
- * so that they can act on buffers other than the active one.
- */
- /**
- * Insert a character into an existing line.
- */
- line_t * line_insert(line_t * line, char_t c, int offset, int lineno) {
- /* If there is not enough space... */
- if (line->actual == line->available) {
- /* Expand the line buffer */
- line->available *= 2;
- line = realloc(line, sizeof(line_t) + sizeof(char_t) * line->available);
- }
- /* If this was not the last character, then shift remaining characters forward. */
- if (offset < line->actual) {
- memmove(&line->text[offset+1], &line->text[offset], sizeof(char_t) * (line->actual - offset));
- }
- /* Insert the new character */
- line->text[offset] = c;
- /* There is one new character in the line */
- line->actual += 1;
- recalculate_tabs(line);
- recalculate_syntax(line, lineno);
- return line;
- }
- /**
- * Delete a character from a line
- */
- void line_delete(line_t * line, int offset, int lineno) {
- /* Can't delete character before start of line. */
- if (offset == 0) return;
- /* If this isn't the last character, we need to move all subsequent characters backwards */
- if (offset < line->actual) {
- memmove(&line->text[offset-1], &line->text[offset], sizeof(char_t) * (line->actual - offset));
- }
- /* The line is one character shorter */
- line->actual -= 1;
- recalculate_tabs(line);
- recalculate_syntax(line, lineno);
- }
- /**
- * Remove a line from the active buffer
- */
- line_t ** remove_line(line_t ** lines, int offset) {
- /* If there is only one line, clear it instead of removing it. */
- if (env->line_count == 1) {
- lines[offset]->actual = 0;
- return lines;
- }
- /* Otherwise, free the data used by the line */
- free(lines[offset]);
- /* Move other lines up */
- if (offset < env->line_count) {
- memmove(&lines[offset], &lines[offset+1], sizeof(line_t *) * (env->line_count - (offset - 1)));
- lines[env->line_count-1] = NULL;
- }
- /* There is one less line */
- env->line_count -= 1;
- return lines;
- }
- /**
- * Add a new line to the active buffer.
- */
- line_t ** add_line(line_t ** lines, int offset) {
- /* Invalid offset? */
- if (offset > env->line_count) return lines;
- /* Not enough space */
- if (env->line_count == env->line_avail) {
- /* Allocate more space */
- env->line_avail *= 2;
- lines = realloc(lines, sizeof(line_t *) * env->line_avail);
- }
- /* If this isn't the last line, move other lines down */
- if (offset < env->line_count) {
- memmove(&lines[offset+1], &lines[offset], sizeof(line_t *) * (env->line_count - offset));
- }
- /* Allocate the new line */
- lines[offset] = malloc(sizeof(line_t) + sizeof(char_t) * 32);
- lines[offset]->available = 32;
- lines[offset]->actual = 0;
- lines[offset]->istate = 0;
- /* There is one new line */
- env->line_count += 1;
- env->lines = lines;
- if (offset > 0) {
- recalculate_syntax(lines[offset-1],offset-1);
- }
- return lines;
- }
- /**
- * Replace a line with data from another line (used by paste to paste yanked lines)
- */
- void replace_line(line_t ** lines, int offset, line_t * replacement) {
- if (lines[offset]->available < replacement->actual) {
- lines[offset] = realloc(lines[offset], sizeof(line_t) + sizeof(char_t) * replacement->available);
- lines[offset]->available = replacement->available;
- }
- lines[offset]->actual = replacement->actual;
- memcpy(&lines[offset]->text, &replacement->text, sizeof(char_t) * replacement->actual);
- recalculate_syntax(lines[offset],offset);
- }
- /**
- * Merge two consecutive lines.
- * lineb is the offset of the second line.
- */
- line_t ** merge_lines(line_t ** lines, int lineb) {
- /* linea is the line immediately before lineb */
- int linea = lineb - 1;
- /* If there isn't enough space in linea hold both... */
- while (lines[linea]->available < lines[linea]->actual + lines[lineb]->actual) {
- /* ... allocate more space until it fits */
- lines[linea]->available *= 2;
- /* XXX why not just do this once after calculating appropriate size */
- lines[linea] = realloc(lines[linea], sizeof(line_t) + sizeof(char_t) * lines[linea]->available);
- }
- /* Copy the second line into the first line */
- memcpy(&lines[linea]->text[lines[linea]->actual], &lines[lineb]->text, sizeof(char_t) * lines[lineb]->actual);
- /* The first line is now longer */
- lines[linea]->actual = lines[linea]->actual + lines[lineb]->actual;
- recalculate_tabs(lines[linea]);
- recalculate_syntax(lines[linea], linea);
- /* Remove the second line */
- return remove_line(lines, lineb);
- }
- /**
- * Split a line into two lines at the given column
- */
- line_t ** split_line(line_t ** lines, int line, int split) {
- /* If we're trying to split from the start, just add a new blank line before */
- if (split == 0) {
- return add_line(lines, line - 1);
- }
- /* Allocate more space as needed */
- if (env->line_count == env->line_avail) {
- env->line_avail *= 2;
- lines = realloc(lines, sizeof(line_t *) * env->line_avail);
- }
- /* Shift later lines down */
- if (line < env->line_count) {
- memmove(&lines[line+1], &lines[line], sizeof(line_t *) * (env->line_count - line));
- }
- /* I have no idea what this is doing */
- int remaining = lines[line-1]->actual - split;
- int v = remaining;
- v--;
- v |= v >> 1;
- v |= v >> 2;
- v |= v >> 4;
- v |= v >> 8;
- v |= v >> 16;
- v++;
- /* Allocate space for the new line */
- lines[line] = malloc(sizeof(line_t) + sizeof(char_t) * v);
- lines[line]->available = v;
- lines[line]->actual = remaining;
- lines[line]->istate = 0;
- /* Move the data from the old line into the new line */
- memmove(lines[line]->text, &lines[line-1]->text[split], sizeof(char_t) * remaining);
- lines[line-1]->actual = split;
- recalculate_tabs(lines[line-1]);
- recalculate_tabs(lines[line]);
- recalculate_syntax(lines[line-1], line-1);
- recalculate_syntax(lines[line], line);
- /* There is one new line */
- env->line_count += 1;
- /* We may have reallocated lines */
- return lines;
- }
- /**
- * Initialize a buffer with default values
- */
- void setup_buffer(buffer_t * env) {
- /* If this buffer was already initialized, clear out its line data */
- if (env->lines) {
- for (int i = 0; i < env->line_count; ++i) {
- free(env->lines[i]);
- }
- free(env->lines);
- }
- /* Default state parameters */
- env->line_no = 1; /* Default cursor position */
- env->col_no = 1;
- env->line_count = 1; /* Buffers always have at least one line */
- env->modified = 0;
- env->readonly = 0;
- env->offset = 0;
- env->line_avail = 8; /* Default line buffer capacity */
- env->tabs = 1; /* Tabs by default */
- env->tabstop = 4; /* Tab stop width */
- /* Allocate line buffer */
- env->lines = malloc(sizeof(line_t *) * env->line_avail);
- /* Initialize the first line */
- env->lines[0] = malloc(sizeof(line_t) + sizeof(char_t) * 32);
- env->lines[0]->available = 32;
- env->lines[0]->actual = 0;
- env->lines[0]->istate = 0;
- }
- /**
- * Toggle buffered / unbuffered modes
- */
- struct termios old;
- void set_unbuffered(void) {
- tcgetattr(STDOUT_FILENO, &old);
- struct termios new = old;
- new.c_lflag &= (~ICANON & ~ECHO);
- tcsetattr(STDOUT_FILENO, TCSAFLUSH, &new);
- }
- void set_buffered(void) {
- tcsetattr(STDOUT_FILENO, TCSAFLUSH, &old);
- }
- /**
- * Convert codepoint to utf-8 string
- */
- int to_eight(uint32_t codepoint, char * out) {
- memset(out, 0x00, 7);
- if (codepoint < 0x0080) {
- out[0] = (char)codepoint;
- } else if (codepoint < 0x0800) {
- out[0] = 0xC0 | (codepoint >> 6);
- out[1] = 0x80 | (codepoint & 0x3F);
- } else if (codepoint < 0x10000) {
- out[0] = 0xE0 | (codepoint >> 12);
- out[1] = 0x80 | ((codepoint >> 6) & 0x3F);
- out[2] = 0x80 | (codepoint & 0x3F);
- } else if (codepoint < 0x200000) {
- out[0] = 0xF0 | (codepoint >> 18);
- out[1] = 0x80 | ((codepoint >> 12) & 0x3F);
- out[2] = 0x80 | ((codepoint >> 6) & 0x3F);
- out[3] = 0x80 | ((codepoint) & 0x3F);
- } else if (codepoint < 0x4000000) {
- out[0] = 0xF8 | (codepoint >> 24);
- out[1] = 0x80 | (codepoint >> 18);
- out[2] = 0x80 | ((codepoint >> 12) & 0x3F);
- out[3] = 0x80 | ((codepoint >> 6) & 0x3F);
- out[4] = 0x80 | ((codepoint) & 0x3F);
- } else {
- out[0] = 0xF8 | (codepoint >> 30);
- out[1] = 0x80 | ((codepoint >> 24) & 0x3F);
- out[2] = 0x80 | ((codepoint >> 18) & 0x3F);
- out[3] = 0x80 | ((codepoint >> 12) & 0x3F);
- out[4] = 0x80 | ((codepoint >> 6) & 0x3F);
- out[5] = 0x80 | ((codepoint) & 0x3F);
- }
- return strlen(out);
- }
- /**
- * Get the presentation width of a codepoint
- */
- int codepoint_width(wchar_t codepoint) {
- if (codepoint == '\t') {
- return 1; /* Recalculate later */
- }
- if (codepoint < 32) {
- /* We render these as ^@ */
- return 2;
- }
- if (codepoint == 0x7F) {
- /* Renders as ^? */
- return 2;
- }
- if (codepoint > 0x7f && codepoint < 0xa0) {
- /* Upper control bytes <xx> */
- return 4;
- }
- if (codepoint == 0xa0) {
- /* Non-breaking space _ */
- return 1;
- }
- /* Skip wcwidth for anything under 256 */
- if (codepoint > 256) {
- /* Higher codepoints may be wider (eg. Japanese) */
- return wcwidth(codepoint);
- }
- return 1;
- }
- /**
- * Move the terminal cursor
- */
- void place_cursor(int x, int y) {
- printf("\033[%d;%dH", y, x);
- fflush(stdout);
- }
- /**
- * Move the terminal cursor, but only horizontally
- */
- void place_cursor_h(int h) {
- printf("\033[%dG", h);
- fflush(stdout);
- }
- /**
- * Set text colors
- *
- * Normally, text colors are just strings, but if they
- * start with @ they will be parsed as integers
- * representing one of the 16 standard colors, suitable
- * for terminals without support for the 256- or 24-bit
- * color modes.
- */
- void set_colors(const char * fg, const char * bg) {
- printf("\033[22;23;");
- if (*bg == '@') {
- int _bg = atoi(bg+1);
- if (_bg < 8) {
- printf("4%d;", _bg);
- } else {
- printf("10%d;", _bg-8);
- }
- } else {
- printf("48;%s;", bg);
- }
- if (*fg == '@') {
- int _fg = atoi(fg+1);
- if (_fg < 8) {
- printf("3%dm", _fg);
- } else {
- printf("9%dm", _fg-8);
- }
- } else {
- printf("38;%sm", fg);
- }
- fflush(stdout);
- }
- /**
- * Set just the foreground color
- *
- * (See set_colors above)
- */
- void set_fg_color(const char * fg) {
- printf("\033[22;23;");
- if (*fg == '@') {
- int _fg = atoi(fg+1);
- if (_fg < 8) {
- printf("3%dm", _fg);
- } else {
- printf("9%dm", _fg-8);
- }
- } else {
- printf("38;%sm", fg);
- }
- fflush(stdout);
- }
- /**
- * Clear the rest of this line
- */
- void clear_to_end(void) {
- printf("\033[K");
- fflush(stdout);
- }
- /**
- * Enable bold text display
- */
- void set_bold(void) {
- printf("\033[1m");
- fflush(stdout);
- }
- /**
- * Enable underlined text display
- */
- void set_underline(void) {
- printf("\033[4m");
- fflush(stdout);
- }
- /**
- * Reset text display attributes
- */
- void reset(void) {
- printf("\033[0m");
- fflush(stdout);
- }
- /**
- * Clear the entire screen
- */
- void clear_screen(void) {
- printf("\033[H\033[2J");
- fflush(stdout);
- }
- /**
- * Hide the cursor
- */
- void hide_cursor(void) {
- printf("\033[?25l");
- fflush(stdout);
- }
- /*
- * Show the cursor
- */
- void show_cursor(void) {
- printf("\033[?25h");
- fflush(stdout);
- }
- /**
- * Request mouse events
- */
- void mouse_enable(void) {
- printf("\033[?1000h");
- fflush(stdout);
- }
- /**
- * Stop mouse events
- */
- void mouse_disable(void) {
- printf("\033[?1000l");
- fflush(stdout);
- }
- /**
- * Shift the screen up one line
- */
- void shift_up(void) {
- printf("\033[1S");
- }
- /**
- * Shift the screen down one line.
- */
- void shift_down(void) {
- printf("\033[1T");
- }
- /**
- * Redaw the tabbar, with a tab for each buffer.
- *
- * The active buffer is highlighted.
- */
- void redraw_tabbar(void) {
- /* Hide cursor while rendering UI */
- hide_cursor();
- /* Move to upper left */
- place_cursor(1,1);
- /* For each buffer... */
- for (int i = 0; i < buffers_len; i++) {
- buffer_t * _env = buffers[i];
- if (_env == env) {
- /* If this is the active buffer, hilight it */
- reset();
- set_colors(COLOR_FG, COLOR_BG);
- set_bold();
- } else {
- /* Otherwise use default tab color */
- reset();
- set_colors(COLOR_FG, COLOR_TAB_BG);
- set_underline();
- }
- /* If this buffer is modified, indicate that with a prepended + */
- if (_env->modified) {
- printf(" +");
- }
- /* Print the filename */
- if (_env->file_name) {
- printf(" %s ", _env->file_name);
- } else {
- printf(" [No Name] ");
- }
- }
- /* Reset bold/underline */
- reset();
- /* Fill the rest of the tab bar */
- set_colors(COLOR_FG, COLOR_TABBAR_BG);
- clear_to_end();
- }
- /**
- * Braindead log10 implementation for the line numbers
- */
- int log_base_10(unsigned int v) {
- int r = (v >= 1000000000) ? 9 : (v >= 100000000) ? 8 : (v >= 10000000) ? 7 :
- (v >= 1000000) ? 6 : (v >= 100000) ? 5 : (v >= 10000) ? 4 :
- (v >= 1000) ? 3 : (v >= 100) ? 2 : (v >= 10) ? 1 : 0;
- return r;
- }
- /**
- * Render a line of text
- *
- * This handles rendering the actual text content. A full line of text
- * also includes a line number and some padding.
- *
- * width: width of the text display region (term width - line number width)
- * offset: how many cells into the line to start rendering at
- */
- void render_line(line_t * line, int width, int offset) {
- int i = 0; /* Offset in char_t line data entries */
- int j = 0; /* Offset in terminal cells */
- const char * last_color = NULL;
- /* Set default text colors */
- set_colors(COLOR_FG, COLOR_BG);
- /*
- * When we are rendering in the middle of a wide character,
- * we render -'s to fill the remaining amount of the
- * charater's width
- */
- int remainder = 0;
- /* For each character in the line ... */
- while (i < line->actual) {
- /* If there is remaining text... */
- if (remainder) {
- /* If we should be drawing by now... */
- if (j >= offset) {
- /* Fill remainder with -'s */
- set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("-");
- set_colors(COLOR_FG, COLOR_BG);
- }
- /* One less remaining width cell to fill */
- remainder--;
- /* Terminal offset moves forward */
- j++;
- /*
- * If this was the last remaining character, move to
- * the next codepoint in the line
- */
- if (remainder == 0) {
- i++;
- }
- continue;
- }
- /* Get the next character to draw */
- char_t c = line->text[i];
- /* If we should be drawing by now... */
- if (j >= offset) {
- /* If this character is going to fall off the edge of the screen... */
- if (j - offset + c.display_width >= width) {
- /* We draw this with special colors so it isn't ambiguous */
- set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- /* If it's wide, draw ---> as needed */
- while (j - offset < width - 1) {
- printf("-");
- j++;
- }
- /* End the line with a > to show it overflows */
- printf(">");
- break;
- }
- /* Syntax hilighting */
- const char * color = flag_to_color(c.flags);
- if (c.flags == FLAG_SELECT) {
- set_colors(COLOR_BG, COLOR_FG);
- } else {
- if (!last_color || strcmp(color, last_color)) {
- set_fg_color(color);
- last_color = color;
- }
- }
- void _set_colors(const char * fg, const char * bg) {
- if (c.flags != FLAG_SELECT) {
- set_colors(fg,bg);
- }
- }
- /* Render special characters */
- if (c.codepoint == '\t') {
- _set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("»");
- for (int i = 1; i < c.display_width; ++i) {
- printf("·");
- }
- _set_colors(last_color ? last_color : COLOR_FG, COLOR_BG);
- } else if (c.codepoint < 32) {
- /* Codepoints under 32 to get converted to ^@ escapes */
- _set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("^%c", '@' + c.codepoint);
- _set_colors(last_color ? last_color : COLOR_FG, COLOR_BG);
- } else if (c.codepoint == 0x7f) {
- _set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("^?");
- _set_colors(last_color ? last_color : COLOR_FG, COLOR_BG);
- } else if (c.codepoint > 0x7f && c.codepoint < 0xa0) {
- _set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("<%2x>", c.codepoint);
- _set_colors(last_color ? last_color : COLOR_FG, COLOR_BG);
- } else if (c.codepoint == 0xa0) {
- _set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("_");
- _set_colors(last_color ? last_color : COLOR_FG, COLOR_BG);
- } else {
- /* Normal characters get output */
- char tmp[7]; /* Max six bytes, use 7 to ensure last is always nil */
- to_eight(c.codepoint, tmp);
- printf("%s", tmp);
- }
- /* Advance the terminal cell offset by the render width of this character */
- j += c.display_width;
- /* Advance to the next character */
- i++;
- } else if (c.display_width > 1) {
- /*
- * If this is a wide character but we aren't ready to render yet,
- * we may need to draw some filler text for the remainder of its
- * width to ensure we don't jump around when horizontally scrolling
- * past wide characters.
- */
- remainder = c.display_width - 1;
- j++;
- } else {
- /* Regular character, not ready to draw, advance without doing anything */
- j++;
- i++;
- }
- }
- }
- /**
- * Get the width of the line number region
- */
- int num_width(void) {
- int w = log_base_10(env->line_count) + 1;
- if (w < 2) return 2;
- return w;
- }
- /**
- * Redraw line.
- *
- * This draws the line number as well as the actual text.
- * j = screen-relative line offset.
- */
- void redraw_line(int j, int x) {
- if (env->loading) return;
- /* Hide cursor when drawing */
- hide_cursor();
- /* Move cursor to upper left most cell of this line */
- place_cursor(1,2 + j);
- /* Draw a gutter on the left.
- * TODO: The gutter can be used to show single-character
- * line annotations, such as collapse state, or
- * whether a search result was found on this line.
- */
- set_colors(COLOR_NUMBER_FG, COLOR_ALT_FG);
- printf(" ");
- /* Draw the line number */
- set_colors(COLOR_NUMBER_FG, COLOR_NUMBER_BG);
- int num_size = num_width();
- for (int y = 0; y < num_size - log_base_10(x + 1); ++y) {
- printf(" ");
- }
- printf("%d%c", x + 1, (x+1 == env->line_no && env->coffset > 0) ? '<' : ' ');
- /*
- * Draw the line text
- * If this is the active line, the current character cell offset should be used.
- * (Non-active lines are not shifted and always render from the start of the line)
- */
- render_line(env->lines[x], global_config.term_width - 3 - num_size, (x + 1 == env->line_no) ? env->coffset : 0);
- /* Clear the rest of the line */
- clear_to_end();
- }
- /**
- * Redraw the entire text area
- */
- void redraw_text(void) {
- /* Hide cursor while rendering */
- hide_cursor();
- /* Figure out the available size of the text region */
- int l = global_config.term_height - global_config.bottom_size - 1;
- int j = 0;
- /* Draw each line */
- for (int x = env->offset; j < l && x < env->line_count; x++) {
- redraw_line(j,x);
- j++;
- }
- /* Draw the rest of the text region as ~ lines */
- for (; j < l; ++j) {
- place_cursor(1,2 + j);
- set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("~");
- clear_to_end();
- }
- }
- /**
- * Draw the status bar
- *
- * The status bar shows the name of the file, whether it has modifications,
- * and (in the future) what syntax highlighting mode is enabled.
- *
- * The right side of the tatus bar shows the line number and column.
- */
- void redraw_statusbar(void) {
- /* Hide cursor while rendering */
- hide_cursor();
- /* Move cursor to the status bar line (second from bottom */
- place_cursor(1, global_config.term_height - 1);
- /* Set background colors for status line */
- set_colors(COLOR_STATUS_FG, COLOR_STATUS_BG);
- /* Print the file name */
- if (env->file_name) {
- printf("%s", env->file_name);
- } else {
- printf("[No Name]");
- }
- printf(" ");
- if (env->syntax) {
- printf("[%s]", env->syntax->name);
- }
- /* Print file status indicators */
- if (env->modified) {
- printf("[+]");
- }
- if (env->readonly) {
- printf("[ro]");
- }
- printf(" ");
- if (global_config.yanks) {
- printf("[y:%ld]", global_config.yank_count);
- }
- /* Clear the rest of the status bar */
- clear_to_end();
- /* Pre-render the right hand side of the status bar */
- char right_hand[1024];
- sprintf(right_hand, "Line %d/%d Col: %d ", env->line_no, env->line_count, env->col_no);
- /* Move the cursor appropriately to draw it */
- place_cursor_h(global_config.term_width - strlen(right_hand)); /* TODO: What if we're localized and this has wide chars? */
- printf("%s",right_hand);
- fflush(stdout);
- }
- /**
- * Draw the command line
- *
- * The command line either has input from the user (:quit, :!make, etc.)
- * or shows the INSERT (or VISUAL in the future) mode name.
- */
- void redraw_commandline(void) {
- /* Hide cursor while rendering */
- hide_cursor();
- /* Move cursor to the last line */
- place_cursor(1, global_config.term_height);
- /* Set background color */
- set_colors(COLOR_FG, COLOR_BG);
- /* If we are in an edit mode, note that. */
- if (env->mode == MODE_INSERT) {
- set_bold();
- printf("-- INSERT --");
- clear_to_end();
- reset();
- } else if (env->mode == MODE_LINE_SELECTION) {
- set_bold();
- printf("-- LINE SELECTION --");
- clear_to_end();
- reset();
- } else {
- clear_to_end();
- }
- }
- /**
- * Draw all screen elements
- */
- void redraw_all(void) {
- redraw_tabbar();
- redraw_text();
- redraw_statusbar();
- redraw_commandline();
- }
- /**
- * Update the terminal title bar
- */
- void update_title(void) {
- char cwd[1024] = {'/',0};
- getcwd(cwd, 1024);
- /*
- * XXX: I think this only works in a few terminals.
- * VTE seems to have a different escape sequence for tab names.
- */
- printf("\033]1;%s%s (%s) - BIM\007", env->file_name, env->modified ? " +" : "", cwd);
- }
- /**
- * Mark this buffer as modified and
- * redraw the status and tabbar if needed.
- */
- void set_modified(void) {
- /* If it was already marked modified, no need to do anything */
- if (env->modified) return;
- /* Mark as modified */
- env->modified = 1;
- /* Redraw some things */
- update_title();
- redraw_tabbar();
- redraw_statusbar();
- }
- /**
- * Draw a message on the status line
- */
- void render_status_message(char * message, ...) {
- /* varargs setup */
- va_list args;
- va_start(args, message);
- char buf[1024];
- /* Process format string */
- vsprintf(buf, message, args);
- va_end(args);
- /* Hide cursor while rendering */
- hide_cursor();
- /* Move cursor to the status bar line (second from bottom */
- place_cursor(1, global_config.term_height - 1);
- /* Set background colors for status line */
- set_colors(COLOR_STATUS_FG, COLOR_STATUS_BG);
- printf("%s", buf);
- /* Clear the rest of the status bar */
- clear_to_end();
- }
- /**
- * Draw an errormessage to the command line.
- */
- void render_error(char * message, ...) {
- /* varargs setup */
- va_list args;
- va_start(args, message);
- char buf[1024];
- /* Process format string */
- vsprintf(buf, message, args);
- va_end(args);
- /* Hide cursor and redraw command line */
- redraw_commandline();
- /* Set appropriate error message colors */
- set_colors(COLOR_ERROR_FG, COLOR_ERROR_BG);
- /* Draw the message */
- printf("%s", buf);
- fflush(stdout);
- }
- /**
- * Move the cursor to the appropriate location based
- * on where it is in the text region.
- *
- * This does some additional math to set the text
- * region horizontal offset.
- */
- void place_cursor_actual(void) {
- /* Account for the left hand gutter */
- int num_size = num_width() + 3;
- int x = num_size + 1 - env->coffset;
- /* Determine where the cursor is physically */
- for (int i = 0; i < env->col_no - 1; ++i) {
- char_t * c = &env->lines[env->line_no-1]->text[i];
- x += c->display_width;
- }
- /* y is a bit easier to calculate */
- int y = env->line_no - env->offset + 1;
- int needs_redraw = 0;
- while (y < 2) {
- y++;
- env->offset--;
- needs_redraw = 1;
- }
- while (y > global_config.term_height - global_config.bottom_size) {
- y--;
- env->offset++;
- needs_redraw = 1;
- }
- if (needs_redraw) {
- redraw_text();
- redraw_tabbar();
- redraw_statusbar();
- redraw_commandline();
- }
- /* If the cursor has gone off screen to the right... */
- if (x > global_config.term_width - 1) {
- /* Adjust the offset appropriately to scroll horizontally */
- int diff = x - (global_config.term_width - 1);
- env->coffset += diff;
- x -= diff;
- redraw_text();
- }
- /* Same for scrolling horizontally to the left */
- if (x < num_size + 1) {
- int diff = (num_size + 1) - x;
- env->coffset -= diff;
- x += diff;
- redraw_text();
- }
- /* Move the actual terminal cursor */
- place_cursor(x,y);
- /* Show the cursor */
- show_cursor();
- }
- /**
- * Handle terminal size changes
- */
- void SIGWINCH_handler(int sig) {
- (void)sig;
- struct winsize w;
- ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
- global_config.term_width = w.ws_col;
- global_config.term_height = w.ws_row;
- redraw_all();
- signal(SIGWINCH, SIGWINCH_handler);
- }
- /**
- * Run global initialization tasks
- */
- void initialize(void) {
- setlocale(LC_ALL, "");
- buffers_avail = 4;
- buffers = malloc(sizeof(buffer_t *) * buffers_avail);
- struct winsize w;
- ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
- global_config.term_width = w.ws_col;
- global_config.term_height = w.ws_row;
- set_unbuffered();
- mouse_enable();
- signal(SIGWINCH, SIGWINCH_handler);
- }
- /**
- * Move the cursor to a specific line.
- */
- void goto_line(int line) {
- /* Respect file bounds */
- if (line < 1) line = 1;
- if (line > env->line_count) line = env->line_count;
- /* Move the cursor / text region offsets */
- env->coffset = 0;
- env->offset = line - 1;
- env->line_no = line;
- env->col_no = 1;
- redraw_all();
- }
- /**
- * UTF-8 parser state
- */
- static uint32_t codepoint_r;
- static uint32_t state = 0;
- /**
- * Processs (part of) a file and add it to a buffer.
- */
- void add_buffer(uint8_t * buf, int size) {
- for (int i = 0; i < size; ++i) {
- if (!decode(&state, &codepoint_r, buf[i])) {
- uint32_t c = codepoint_r;
- if (c == '\n') {
- env->lines = add_line(env->lines, env->line_no);
- env->col_no = 1;
- env->line_no += 1;
- } else {
- char_t _c;
- _c.codepoint = (uint32_t)c;
- _c.flags = 0;
- _c.display_width = codepoint_width((wchar_t)c);
- line_t * line = env->lines[env->line_no - 1];
- line_t * nline = line_insert(line, _c, env->col_no - 1, env->line_no-1);
- if (line != nline) {
- env->lines[env->line_no - 1] = nline;
- }
- env->col_no += 1;
- }
- } else if (state == UTF8_REJECT) {
- state = 0;
- }
- }
- }
- struct syntax_definition * match_syntax(char * file) {
- for (struct syntax_definition * s = syntaxes; s->name; ++s) {
- for (char ** ext = s->ext; *ext; ++ext) {
- int i = strlen(file);
- int j = strlen(*ext);
- do {
- if (file[i] != (*ext)[j]) break;
- if (j == 0) return s;
- if (i == 0) break;
- i--;
- j--;
- } while (1);
- }
- }
- return NULL;
- }
- /**
- * Create a new buffer from a file.
- */
- void open_file(char * file) {
- env = buffer_new();
- env->loading = 1;
- setup_buffer(env);
- FILE * f;
- if (!strcmp(file,"-")) {
- /**
- * Read file from stdin. stderr provides terminal input.
- */
- f = stdin;
- global_config.tty_in = STDERR_FILENO;
- env->modified = 1;
- } else {
- f = fopen(file, "r");
- env->file_name = strdup(file);
- }
- if (!f) {
- if (global_config.hilight_on_open) {
- env->syntax = match_syntax(file);
- }
- env->loading = 0;
- return;
- }
- uint8_t buf[BLOCK_SIZE];
- state = 0;
- while (!feof(f)) {
- size_t r = fread(buf, 1, BLOCK_SIZE, f);
- add_buffer(buf, r);
- }
- if (env->line_no && env->lines[env->line_no-1] && env->lines[env->line_no-1]->actual == 0) {
- /* Remove blank line from end */
- remove_line(env->lines, env->line_no-1);
- }
- if (global_config.hilight_on_open) {
- env->syntax = match_syntax(file);
- for (int i = 0; i < env->line_count; ++i) {
- recalculate_syntax(env->lines[i],i);
- }
- }
- env->loading = 0;
- for (int i = 0; i < env->line_count; ++i) {
- recalculate_tabs(env->lines[i]);
- }
- update_title();
- goto_line(0);
- fclose(f);
- }
- /**
- * Clean up the terminal and exit the editor.
- */
- void quit(void) {
- mouse_disable();
- set_buffered();
- reset();
- clear_screen();
- show_cursor();
- printf("Thanks for flying bim!\n");
- exit(0);
- }
- /**
- * Try to quit, but don't continue if there are
- * modified buffers open.
- */
- void try_quit(void) {
- for (int i = 0; i < buffers_len; i++ ) {
- buffer_t * _env = buffers[i];
- if (_env->modified) {
- if (_env->file_name) {
- render_error("Modifications made to file `%s` in tab %d. Aborting.", _env->file_name, i+1);
- } else {
- render_error("Unsaved new file in tab %d. Aborting.", i+1);
- }
- return;
- }
- }
- quit();
- }
- /**
- * Switch to the previous buffer
- */
- void previous_tab(void) {
- buffer_t * last = NULL;
- for (int i = 0; i < buffers_len; i++) {
- buffer_t * _env = buffers[i];
- if (_env == env) {
- if (last) {
- /* Wrap around */
- env = last;
- redraw_all();
- return;
- } else {
- env = buffers[buffers_len-1];
- redraw_all();
- return;
- }
- }
- last = _env;
- }
- }
- /**
- * Switch to the next buffer
- */
- void next_tab(void) {
- for (int i = 0; i < buffers_len; i++) {
- buffer_t * _env = buffers[i];
- if (_env == env) {
- if (i != buffers_len - 1) {
- env = buffers[i+1];
- redraw_all();
- return;
- } else {
- /* Wrap around */
- env = buffers[0];
- redraw_all();
- return;
- }
- }
- }
- }
- /**
- * Write active buffer to file
- */
- void write_file(char * file) {
- if (!file) {
- render_error("Need a file to write to.");
- return;
- }
- FILE * f = fopen(file, "w+");
- if (!f) {
- render_error("Failed to open file for writing.");
- return;
- }
- /* Go through each line and convert it back to UTF-8 */
- int i, j;
- for (i = 0; i < env->line_count; ++i) {
- line_t * line = env->lines[i];
- for (j = 0; j < line->actual; j++) {
- char_t c = line->text[j];
- if (c.codepoint == 0) {
- char buf[1] = {0};
- fwrite(buf, 1, 1, f);
- } else {
- char tmp[8] = {0};
- int i = to_eight(c.codepoint, tmp);
- fwrite(tmp, i, 1, f);
- }
- }
- fputc('\n', f);
- }
- fclose(f);
- /* Mark it no longer modified */
- env->modified = 0;
- /* If there was no file name set, set one */
- if (!env->file_name) {
- env->file_name = malloc(strlen(file) + 1);
- memcpy(env->file_name, file, strlen(file) + 1);
- }
- redraw_all();
- }
- /**
- * Close the active buffer
- */
- void close_buffer(void) {
- buffer_t * previous_env = env;
- buffer_t * new_env = buffer_close(env);
- if (new_env == previous_env) {
- /* ?? Failed to close buffer */
- render_error("lolwat");
- }
- /* No more buffers, exit */
- if (!new_env) {
- quit();
- }
- /* Clean up the old buffer */
- free(previous_env);
- /* Set the new active buffer */
- env = new_env;
- /* Redraw the screen */
- redraw_all();
- }
- /**
- * Move the cursor down one line in the text region
- */
- void cursor_down(void) {
- /* If this isn't already the last line... */
- if (env->line_no < env->line_count) {
- /* Move the cursor down */
- env->line_no += 1;
- /*
- * If the horizontal cursor position exceeds the width the new line,
- * then move the cursor left to the extent of the new line.
- *
- * If we're in insert mode, we can go one cell beyond the end of the line
- */
- if (env->col_no > env->lines[env->line_no-1]->actual + (env->mode == MODE_INSERT)) {
- env->col_no = env->lines[env->line_no-1]->actual + (env->mode == MODE_INSERT);
- if (env->col_no == 0) env->col_no = 1;
- }
- /*
- * If the screen was scrolled horizontally, unscroll it;
- * if it will be scrolled on this line as well, that will
- * be handled by place_cursor_actual
- */
- int redraw = 0;
- if (env->coffset != 0) {
- env->coffset = 0;
- redraw = 1;
- }
- /* If we've scrolled past the bottom of the screen, scroll the screen */
- if (env->line_no > env->offset + global_config.term_height - global_config.bottom_size - 1) {
- env->offset += 1;
- /* Tell terminal to scroll */
- shift_up();
- /* A new line appears on screen at the bottom, draw it */
- int l = global_config.term_height - global_config.bottom_size - 1;
- if (env->offset + l < env->line_count + 1) {
- redraw_line(l-1, env->offset + l-1);
- } else {
- place_cursor(1, 2 + l - 1);
- set_colors(COLOR_ALT_FG, COLOR_ALT_BG);
- printf("~");
- clear_to_end();
- }
- /* Redraw elements that were moved by scrolling */
- redraw_tabbar();
- redraw_statusbar();
- redraw_commandline();
- place_cursor_actual();
- return;
- } else if (redraw) {
- /* Otherwise, if we need to redraw because of coffset change, do that */
- redraw_text();
- }
- /* Update the status bar */
- redraw_statusbar();
- /* Place the terminal cursor again */
- place_cursor_actual();
- }
- }
- /**
- * Move the cursor up one line in the text region
- */
- void cursor_up(void) {
- /* If this isn't the first line already */
- if (env->line_no > 1) {
- /* Move the cursor down */
- env->line_no -= 1;
- /*
- * If the horizontal cursor position exceeds the width the new line,
- * then move the cursor left to the extent of the new line.
- *
- * If we're in insert mode, we can go one cell beyond the end of the line
- */
- if (env->col_no > env->lines[env->line_no-1]->actual + (env->mode == MODE_INSERT)) {
- env->col_no = env->lines[env->line_no-1]->actual + (env->mode == MODE_INSERT);
- if (env->col_no == 0) env->col_no = 1;
- }
- /*
- * If the screen was scrolled horizontally, unscroll it;
- * if it will be scrolled on this line as well, that will
- * be handled by place_cursor_actual
- */
- int redraw = 0;
- if (env->coffset != 0) {
- env->coffset = 0;
- redraw = 1;
- }
- if (env->line_no <= env->offset) {
- env->offset -= 1;
- /* Tell terminal to scroll */
- shift_down();
- /*
- * The line at the top of the screen should always be real
- * so we can just call redraw_line here
- */
- redraw_line(0,env->offset);
- /* Redraw elements that were moved by scrolling */
- redraw_tabbar();
- redraw_statusbar();
- redraw_commandline();
- place_cursor_actual();
- return;
- } else if (redraw) {
- /* Otherwise, if we need to redraw because of coffset change, do that */
- redraw_text();
- }
- /* Update the status bar */
- redraw_statusbar();
- /* Place the terminal cursor again */
- place_cursor_actual();
- }
- }
- /**
- * Move the cursor one column left.
- */
- void cursor_left(void) {
- if (env->col_no > 1) {
- env->col_no -= 1;
- /* Update the status bar */
- redraw_statusbar();
- /* Place the terminal cursor again */
- place_cursor_actual();
- }
- }
- /**
- * Move the cursor one column right.
- */
- void cursor_right(void) {
- /* If this isn't already the rightmost column we can reach on this line in this mode... */
- if (env->col_no < env->lines[env->line_no-1]->actual + !!(env->mode == MODE_INSERT)) {
- env->col_no += 1;
- /* Update the status bar */
- redraw_statusbar();
- /* Place the terminal cursor again */
- place_cursor_actual();
- }
- }
- /**
- * Move the cursor to the fron the of the line
- */
- void cursor_home(void) {
- env->col_no = 1;
- /* Update the status bar */
- redraw_statusbar();
- /* Place the terminal cursor again */
- place_cursor_actual();
- }
- /**
- * Move the cursor to the end of the line.
- *
- * In INSERT mode, moves one cell right of the end of the line.
- * In NORMAL mode, moves the cursor to the last occupied cell.
- */
- void cursor_end(void) {
- env->col_no = env->lines[env->line_no-1]->actual+!!(env->mode == MODE_INSERT);
- /* Update the status bar */
- redraw_statusbar();
- /* Place the terminal cursor again */
- place_cursor_actual();
- }
- /**
- * Leave INSERT mode
- *
- * If the cursor is too far right, adjust it.
- * Redraw the command line.
- */
- void leave_insert(void) {
- if (env->col_no > env->lines[env->line_no-1]->actual) {
- env->col_no = env->lines[env->line_no-1]->actual;
- if (env->col_no == 0) env->col_no = 1;
- }
- env->mode = MODE_NORMAL;
- redraw_commandline();
- }
- /**
- * Process a user command.
- */
- void process_command(char * cmd) {
- /* Special case ! to run shell commands without parsing tokens */
- if (*cmd == '!') {
- /* Reset and draw some line feeds */
- reset();
- printf("\n\n");
- /* Set buffered for shell application */
- set_buffered();
- /* Call the shell and wait for completion */
- system(&cmd[1]);
- /* Return to the editor, wait for user to press enter. */
- set_unbuffered();
- printf("\n\nPress ENTER to continue.");
- fflush(stdout);
- while (bim_getch() != ENTER_KEY);
- /* Redraw the screen */
- redraw_all();
- /* Done processing command */
- return;
- }
- /* Tokenize argument string on spaces */
- char *p, *argv[512], *last;
- int argc = 0;
- for ((p = strtok_r(cmd, " ", &last)); p;
- (p = strtok_r(NULL, " ", &last)), argc++) {
- if (argc < 511) argv[argc] = p;
- }
- argv[argc] = NULL;
- if (argc < 1) {
- /* no op */
- return;
- }
- if (!strcmp(argv[0], "e")) {
- /* e: edit file */
- if (argc > 1) {
- /* This actually opens a new tab */
- open_file(argv[1]);
- } else {
- /* TODO: Reopen file? */
- render_error("Expected a file to open...");
- }
- } else if (!strcmp(argv[0], "tabnew")) {
- env = buffer_new();
- setup_buffer(env);
- redraw_all();
- } else if (!strcmp(argv[0], "w")) {
- /* w: write file */
- if (argc > 1) {
- write_file(argv[1]);
- } else {
- write_file(env->file_name);
- }
- } else if (!strcmp(argv[0], "wq")) {
- /* wq: write file and close buffer; if there's no file to write to, may do weird things */
- write_file(env->file_name);
- close_buffer();
- } else if (!strcmp(argv[0], "q")) {
- /* close buffer if unmodified */
- if (env->modified) {
- render_error("No write since last change. Use :q! to force exit.");
- } else {
- close_buffer();
- }
- } else if (!strcmp(argv[0], "q!")) {
- /* close buffer without warning if unmodified */
- close_buffer();
- } else if (!strcmp(argv[0], "qa") || !strcmp(argv[0], "qall")) {
- /* Close all */
- try_quit();
- } else if (!strcmp(argv[0], "qa!")) {
- /* Forcefully exit editor */
- quit();
- } else if (!strcmp(argv[0], "tabp")) {
- /* Next tab */
- previous_tab();
- } else if (!strcmp(argv[0], "tabn")) {
- /* Previous tab */
- next_tab();
- } else if (!strcmp(argv[0], "noh")) {
- if (env->search) {
- free(env->search);
- redraw_text();
- }
- } else if (!strcmp(argv[0], "help")) {
- /*
- * The repeated calls to redraw_commandline here make use
- * of scrolling to draw this multiline help message on
- * the same background as the command line.
- */
- redraw_commandline(); printf("\n");
- redraw_commandline(); printf(" \033[1mbim - The standard ToaruOS Text Editor\033[22m\n");
- redraw_commandline(); printf("\n");
- redraw_commandline(); printf(" Available commands:\n");
- redraw_commandline(); printf(" Quit with \033[3m:q\033[23m, \033[3m:qa\033[23m, \033[3m:q!\033[23m, \033[3m:qa!\033[23m\n");
- redraw_commandline(); printf(" Write out with \033[3m:w \033[4mfile\033[24;23m\n");
- redraw_commandline(); printf(" Set syntax with \033[3m:syntax \033[4mlanguage\033[24;23m\n");
- redraw_commandline(); printf(" Open a new tab with \033[3m:e \033[4mpath/to/file\033[24;23m\n");
- redraw_commandline(); printf(" \033[3m:tabn\033[23m and \033[3m:tabp\033[23m can be used to switch tabs\n");
- redraw_commandline(); printf(" Set the color scheme with \033[3m:theme \033[4mtheme\033[24;23m\n");
- redraw_commandline(); printf(" Set the behavior of the tab key with \033[3m:tabs\033[23m or \033[3m:spaces\033[23m\n");
- redraw_commandline(); printf(" Set tabstop with \033[3m:tabstop \033[4mwidth\033[24;23m\n");
- redraw_commandline(); printf("\n");
- redraw_commandline(); printf(" Copyright 2013-2018 K. Lange <\033[3mklange@toaruos.org\033[23m>\n");
- redraw_commandline(); printf("\n");
- /* Redrawing the tabbar makes it look like we just shifted the whole view up */
- redraw_tabbar();
- redraw_commandline();
- fflush(stdout);
- /* Wait for a character so we can redraw the screen before continuing */
- int c;
- while ((c = bim_getch())== -1);
- /* Make sure that key press actually gets used */
- bim_unget(c);
- /*
- * Redraw everything to hide the help message and get the
- * upper few lines of text on screen again
- */
- redraw_all();
- } else if (!strcmp(argv[0], "theme")) {
- if (argc < 2) {
- return;
- }
- for (struct theme_def * d = themes; d->name; ++d) {
- if (!strcmp(argv[1], d->name)) {
- d->load();
- redraw_all();
- return;
- }
- }
- } else if (!strcmp(argv[0], "syntax")) {
- if (argc < 2) {
- render_status_message("syntax=%s", env->syntax ? env->syntax->name : "none");
- return;
- }
- if (!strcmp(argv[1],"none")) {
- for (int i = 0; i < env->line_count; ++i) {
- env->lines[i]->istate = 0;
- for (int j = 0; j < env->lines[i]->actual; ++j) {
- env->lines[i]->text[j].flags = 0;
- }
- }
- redraw_all();
- return;
- }
- for (struct syntax_definition * s = syntaxes; s->name; ++s) {
- if (!strcmp(argv[1],s->name)) {
- env->syntax = s;
- for (int i = 0; i < env->line_count; ++i) {
- env->lines[i]->istate = 0;
- }
- for (int i = 0; i < env->line_count; ++i) {
- recalculate_syntax(env->lines[i],i);
- }
- redraw_all();
- return;
- }
- }
- render_error("unrecognized syntax type");
- } else if (!strcmp(argv[0], "recalc")) {
- for (int i = 0; i < env->line_count; ++i) {
- env->lines[i]->istate = 0;
- }
- for (int i = 0; i < env->line_count; ++i) {
- recalculate_syntax(env->lines[i],i);
- }
- redraw_all();
- } else if (!strcmp(argv[0], "tabs")) {
- env->tabs = 1;
- } else if (!strcmp(argv[0], "spaces")) {
- env->tabs = 0;
- } else if (!strcmp(argv[0], "tabstop")) {
- if (argc < 2) {
- render_status_message("tabstop=%d", env->tabstop);
- } else {
- int t = atoi(argv[1]);
- if (t > 0 && t < 32) {
- env->tabstop = t;
- for (int i = 0; i < env->line_count; ++i) {
- recalculate_tabs(env->lines[i]);
- }
- redraw_all();
- } else {
- render_error("Invalid tabstop: %s", argv[1]);
- }
- }
- } else if (!strcmp(argv[0], "clearyank")) {
- if (global_config.yanks) {
- for (unsigned int i = 0; i < global_config.yank_count; ++i) {
- free(global_config.yanks[i]);
- }
- free(global_config.yanks);
- global_config.yanks = NULL;
- global_config.yank_count = 0;
- redraw_statusbar();
- }
- } else if (isdigit(*argv[0])) {
- /* Go to line number */
- goto_line(atoi(argv[0]));
- } else {
- /* Unrecognized command */
- render_error("Not an editor command: %s", argv[0]);
- }
- }
- /**
- * Tab completion for command mode.
- */
- void command_tab_complete(char * buffer) {
- /* Figure out which argument this is and where it starts */
- int arg = 0;
- char * buf = strdup(buffer);
- char * b = buf;
- char * args[32];
- int candidate_count= 0;
- int candidate_space = 4;
- char ** candidates = malloc(sizeof(char*)*candidate_space);
- /* Accept whitespace before first argument */
- while (*b == ' ') b++;
- char * start = b;
- args[0] = start;
- while (*b && *b != ' ') b++;
- while (*b) {
- while (*b == ' ') {
- *b = '\0';
- b++;
- }
- start = b;
- arg++;
- if (arg < 32) {
- args[arg] = start;
- }
- while (*b && *b != ' ') b++;
- }
- /**
- * Check a possible candidate and add it to the
- * candidates list, expanding as necessary,
- * if it matches for the current argument.
- */
- void add_candidate(const char * candidate) {
- char * _arg = args[arg];
- int r = strncmp(_arg, candidate, strlen(_arg));
- if (!r) {
- if (candidate_count == candidate_space) {
- candidate_space *= 2;
- candidates = realloc(candidates,sizeof(char *) * candidate_space);
- }
- candidates[candidate_count] = strdup(candidate);
- candidate_count++;
- }
- }
- if (arg == 0) {
- /* Complete command names */
- add_candidate("help");
- add_candidate("recalc");
- add_candidate("syntax");
- add_candidate("tabn");
- add_candidate("tabp");
- add_candidate("tabnew");
- add_candidate("theme");
- add_candidate("tabs");
- add_candidate("tabstop");
- add_candidate("spaces");
- add_candidate("noh");
- add_candidate("clearyank");
- goto _accept_candidate;
- }
- if (arg == 1 && !strcmp(args[0], "syntax")) {
- /* Complete syntax options */
- add_candidate("none");
- for (struct syntax_definition * s = syntaxes; s->name; ++s) {
- add_candidate(s->name);
- }
- goto _accept_candidate;
- }
- if (arg == 1 && !strcmp(args[0], "theme")) {
- /* Complete color theme names */
- for (struct theme_def * s = themes; s->name; ++s) {
- add_candidate(s->name);
- }
- goto _accept_candidate;
- }
- if (arg == 1 && !strcmp(args[0], "e")) {
- /* Complete file paths */
- /* First, find the deepest directory match */
- char * tmp = strdup(args[arg]);
- char * last_slash = strrchr(tmp, '/');
- DIR * dirp;
- if (last_slash) {
- *last_slash = '\0';
- if (last_slash == tmp) {
- /* Started with slash, and it was the only slash */
- dirp = opendir("/");
- } else {
- dirp = opendir(tmp);
- }
- } else {
- /* No directory match, completing from current directory */
- dirp = opendir(".");
- tmp[0] = '\0';
- }
- if (!dirp) {
- /* Directory match doesn't exist, no candidates to populate */
- free(tmp);
- goto done;
- }
- struct dirent * ent = readdir(dirp);
- while (ent != NULL) {
- if (ent->d_name[0] != '.') {
- struct stat statbuf;
- /* Figure out if this file is a directory */
- if (last_slash) {
- char * x = malloc(strlen(tmp) + 1 + strlen(ent->d_name) + 1);
- sprintf(x,"%s/%s",tmp,ent->d_name);
- lstat(x, &statbuf);
- free(x);
- } else {
- lstat(ent->d_name, &statbuf);
- }
- /* Build the complete argument name to tab complete */
- char s[1024] = {0};
- if (last_slash == tmp) {
- strcat(s,"/");
- } else if (*tmp) {
- strcat(s,tmp);
- strcat(s,"/");
- }
- strcat(s,ent->d_name);
- /*
- * If it is a directory, add a / to the end so the next completion
- * attempt will complete the directory's contents.
- */
- if (S_ISDIR(statbuf.st_mode)) {
- strcat(s,"/");
- }
- add_candidate(s);
- }
- ent = readdir(dirp);
- }
- closedir(dirp);
- free(tmp);
- goto _accept_candidate;
- }
- _accept_candidate:
- if (candidate_count == 0) {
- redraw_statusbar();
- goto done;
- }
- if (candidate_count == 1) {
- /* Only one completion possibility */
- redraw_statusbar();
- /* Fill out the rest of the command */
- char * cstart = (buffer) + (start - buf);
- for (unsigned int i = 0; i < strlen(candidates[0]); ++i) {
- *cstart = candidates[0][i];
- cstart++;
- }
- *cstart = '\0';
- } else {
- /* Print candidates in status bar */
- char tmp[global_config.term_width+1];
- memset(tmp, 0, global_config.term_width+1);
- int offset = 0;
- for (int i = 0; i < candidate_count; ++i) {
- if (offset + 1 + (signed)strlen(candidates[i]) > global_config.term_width - 5) {
- strcat(tmp, "...");
- break;
- }
- if (offset > 0) {
- strcat(tmp, " ");
- offset++;
- }
- strcat(tmp, candidates[i]);
- offset += strlen(candidates[i]);
- }
- render_status_message("%s", tmp);
- /* Complete to longest common substring */
- char * cstart = (buffer) + (start - buf);
- for (int i = 0; i < 1023 /* max length of command */; i++) {
- for (int j = 1; j < candidate_count; ++j) {
- if (candidates[0][i] != candidates[j][i]) goto _reject;
- }
- *cstart = candidates[0][i];
- cstart++;
- }
- /* End of longest common substring */
- _reject:
- *cstart = '\0';
- }
- /* Free candidates */
- for (int i = 0; i < candidate_count; ++i) {
- free(candidates[i]);
- }
- /* Redraw command line */
- done:
- redraw_commandline();
- printf(":%s", buffer);
- free(candidates);
- free(buf);
- }
- /**
- * Command mode
- *
- * Accept a user command and then process it and
- * return to normal mode.
- *
- * TODO: We only have basic line editing here; it might be
- * nice to add more advanced line editing, like cursor
- * movement, tab completion, etc. This is easier than
- * with the shell since we have a lot more control over
- * where the command input bar is rendered.
- */
- void command_mode(void) {
- char c;
- char buffer[1024] = {0};
- int buffer_len = 0;
- redraw_commandline();
- printf(":");
- show_cursor();
- while ((c = bim_getch())) {
- if (c == -1) {
- /* Time out */
- continue;
- }
- if (c == '\033') {
- /* Escape, cancel command */
- break;
- } else if (c == ENTER_KEY) {
- /* Enter, run command */
- process_command(buffer);
- break;
- } else if (c == '\t') {
- /* Handle tab completion */
- command_tab_complete(buffer);
- buffer_len = strlen(buffer);
- } else if (c == BACKSPACE_KEY || c == DELETE_KEY) {
- /* Backspace, delete last character in command buffer */
- if (buffer_len > 0) {
- buffer_len -= 1;
- buffer[buffer_len] = '\0';
- redraw_commandline();
- printf(":%s", buffer);
- } else {
- /* If backspaced through entire command, cancel command mode */
- redraw_commandline();
- break;
- }
- } else {
- /* Regular character */
- buffer[buffer_len] = c;
- buffer_len++;
- printf("%c", c);
- }
- show_cursor();
- }
- }
- /**
- * Search forward from the given cursor position
- * to find a basic search match.
- *
- * This could be more complicated...
- */
- void find_match(int from_line, int from_col, int * out_line, int * out_col, char * str) {
- int col = from_col;
- for (int i = from_line; i <= env->line_count; ++i) {
- line_t * line = env->lines[i - 1];
- int j = col - 1;
- while (j < line->actual + 1) {
- int k = j;
- char * match = str;
- while (k < line->actual + 1) {
- if (*match == '\0') {
- *out_line = i;
- *out_col = j + 1;
- return;
- }
- /* TODO search for UTF-8 sequences? */
- if (*match != line->text[k].codepoint) break;
- match++;
- k++;
- }
- j++;
- }
- col = 0;
- }
- }
- /**
- * Draw the matched search result.
- */
- void draw_search_match(int line, char * buffer, int redraw_buffer) {
- place_cursor_actual();
- redraw_text();
- if (line != -1) {
- /*
- * TODO this should probably mark the relevant
- * regions so that redraw_text can hilight it
- */
- set_colors(COLOR_SEARCH_FG, COLOR_SEARCH_BG);
- place_cursor_actual();
- printf("%s", buffer);
- }
- redraw_statusbar();
- redraw_commandline();
- if (redraw_buffer) {
- printf("/%s", buffer);
- }
- }
- /**
- * Search mode
- *
- * Search text for substring match.
- */
- void search_mode(void) {
- char c;
- char buffer[1024] = {0};
- int buffer_len = 0;
- /* Remember where the cursor is so we can cancel */
- int prev_line = env->line_no;
- int prev_col = env->col_no;
- int prev_coffset = env->coffset;
- int prev_offset = env->offset;
- redraw_commandline();
- printf("/");
- show_cursor();
- while ((c = bim_getch())) {
- if (c == -1) {
- /* Time out */
- continue;
- }
- if (c == '\033') {
- /* Cancel search */
- env->line_no = prev_line;
- env->col_no = prev_col;
- redraw_all();
- break;
- } else if (c == ENTER_KEY) {
- /* Exit search */
- if (env->search) {
- free(env->search);
- }
- env->search = strdup(buffer);
- break;
- } else if (c == BACKSPACE_KEY || c == DELETE_KEY) {
- /* Backspace, delete last character in search buffer */
- if (buffer_len > 0) {
- buffer_len -= 1;
- buffer[buffer_len] = '\0';
- /* Search from beginning to find first match */
- int line = -1, col = -1;
- find_match(prev_line, prev_col, &line, &col, buffer);
- if (line != -1) {
- env->coffset = 0;
- env->offset = line - 1;
- env->col_no = col;
- env->line_no = line;
- }
- draw_search_match(line, buffer, 1);
- } else {
- /* If backspaced through entire search term, cancel search */
- redraw_commandline();
- env->coffset = prev_coffset;
- env->offset = prev_offset;
- env->col_no = prev_col;
- env->line_no = prev_line;
- redraw_all();
- break;
- }
- } else {
- /* Regular character */
- buffer[buffer_len] = c;
- buffer_len++;
- buffer[buffer_len] = '\0';
- printf("%c", c);
- /* Find the next search match */
- int line = -1, col = -1;
- find_match(prev_line, prev_col, &line, &col, buffer);
- if (line != -1) {
- env->coffset = 0;
- env->offset = line - 1;
- env->col_no = col;
- env->line_no = line;
- } else {
- env->coffset = prev_coffset;
- env->offset = prev_offset;
- env->col_no = prev_col;
- env->line_no = prev_line;
- }
- draw_search_match(line, buffer, 1);
- }
- show_cursor();
- }
- }
- /**
- * Find the next search result, or loop back around if at the end.
- */
- void search_next(void) {
- if (!env->search) return;
- int line = -1, col = -1;
- find_match(env->line_no, env->col_no+1, &line, &col, env->search);
- if (line == -1) {
- find_match(1,1, &line, &col, env->search);
- if (line == -1) return;
- }
- env->coffset = 0;
- env->offset = line - 1;
- env->col_no = col;
- env->line_no = line;
- draw_search_match(line, env->search, 0);
- }
- /**
- * Handle mouse event
- */
- void handle_mouse(void) {
- int buttons = bim_getch() - 32;
- int x = bim_getch() - 32;
- int y = bim_getch() - 32;
- if (buttons == 64) {
- /* Scroll up */
- for (int i = 0; i < 5; ++i) {
- cursor_up();
- }
- return;
- } else if (buttons == 65) {
- /* Scroll down */
- for (int i = 0; i < 5; ++i) {
- cursor_down();
- }
- return;
- } else if (buttons == 3) {
- /* Move cursor to position */
- if (x < 0) return;
- if (y < 0) return;
- if (y == 1) {
- /* Pick from tabs */
- int _x = 0;
- for (int i = 0; i < buffers_len; i++) {
- buffer_t * _env = buffers[i];
- if (_env->modified) {
- _x += 2;
- }
- if (_env->file_name) {
- _x += 2 + strlen(_env->file_name);
- } else {
- _x += strlen(" [No Name] ");
- }
- if (_x > x) {
- env = buffers[i];
- redraw_all();
- return;
- }
- }
- return;
- }
- /* Figure out y coordinate */
- int line_no = y + env->offset - 1;
- int col_no = -1;
- if (line_no > env->line_count) {
- line_no = env->line_count;
- }
- /* Account for the left hand gutter */
- int num_size = num_width() + 3;
- int _x = num_size - (line_no == env->line_no ? env->coffset : 0);
- /* Determine where the cursor is physically */
- for (int i = 0; i < env->lines[line_no-1]->actual; ++i) {
- char_t * c = &env->lines[line_no-1]->text[i];
- _x += c->display_width;
- if (_x > x) {
- col_no = i;
- break;
- }
- }
- if (col_no == -1 || col_no > env->lines[line_no-1]->actual) {
- col_no = env->lines[line_no-1]->actual;
- }
- env->line_no = line_no;
- env->col_no = col_no;
- place_cursor_actual();
- }
- return;
- }
- /**
- * Append a character at the current cursor point.
- */
- void insert_char(unsigned int c) {
- char_t _c;
- _c.codepoint = c;
- _c.flags = 0;
- _c.display_width = codepoint_width(c);
- line_t * line = env->lines[env->line_no - 1];
- line_t * nline = line_insert(line, _c, env->col_no - 1, env->line_no - 1);
- if (line != nline) {
- env->lines[env->line_no - 1] = nline;
- }
- redraw_line(env->line_no - env->offset - 1, env->line_no-1);
- env->col_no += 1;
- set_modified();
- }
- /**
- * Move the cursor the start of the previous word.
- */
- void word_left(void) {
- int line_no = env->line_no;
- int col_no = env->col_no;
- do {
- col_no--;
- if (col_no == 0) {
- line_no--;
- if (line_no == 0) {
- goto_line(1);
- return;
- }
- col_no = env->lines[line_no-1]->actual + 1;
- }
- } while (isspace(env->lines[line_no-1]->text[col_no-1].codepoint));
- do {
- col_no--;
- if (col_no == 0) {
- line_no--;
- if (line_no == 0) {
- goto_line(1);
- return;
- }
- col_no = env->lines[line_no-1]->actual + 1;
- }
- if (col_no == 1) {
- env->col_no = 1;
- env->line_no = line_no;
- redraw_statusbar();
- place_cursor_actual();
- return;
- }
- } while (!isspace(env->lines[line_no-1]->text[col_no-1].codepoint));
- env->col_no = col_no;
- env->line_no = line_no;
- cursor_right();
- }
- /**
- * Word right
- */
- void word_right(void) {
- int line_no = env->line_no;
- int col_no = env->col_no;
- do {
- col_no++;
- if (col_no >= env->lines[line_no-1]->actual + 1) {
- line_no++;
- if (line_no >= env->line_count) {
- env->col_no = env->lines[env->line_count-1]->actual;
- env->line_no = env->line_count;
- redraw_statusbar();
- place_cursor_actual();
- return;
- }
- col_no = 0;
- break;
- }
- } while (!isspace(env->lines[line_no-1]->text[col_no-1].codepoint));
- do {
- col_no++;
- if (col_no >= env->lines[line_no-1]->actual + 1) {
- line_no++;
- if (line_no >= env->line_count) {
- env->col_no = env->lines[env->line_count-1]->actual;
- env->line_no = env->line_count;
- redraw_statusbar();
- place_cursor_actual();
- return;
- }
- col_no = 1;
- break;
- }
- } while (isspace(env->lines[line_no-1]->text[col_no-1].codepoint));
- env->col_no = col_no;
- env->line_no = line_no;
- redraw_statusbar();
- place_cursor_actual();
- return;
- }
- int handle_escape(int * this_buf, int * timeout, int c) {
- if (*timeout >= 1 && this_buf[*timeout-1] == '\033' && c == '\033') {
- this_buf[*timeout] = c;
- (*timeout)++;
- return 1;
- }
- if (*timeout >= 1 && this_buf[*timeout-1] == '\033' && c == '[') {
- *timeout = 1;
- this_buf[*timeout] = c;
- (*timeout)++;
- return 0;
- }
- if (*timeout >= 2 && this_buf[0] == '\033' && this_buf[1] == '[' &&
- (isdigit(c) || c == ';')) {
- this_buf[*timeout] = c;
- (*timeout)++;
- return 0;
- }
- if (*timeout >= 2 && this_buf[0] == '\033' && this_buf[1] == '[') {
- switch (c) {
- case 'M':
- handle_mouse();
- break;
- case 'A': // up
- cursor_up();
- break;
- case 'B': // down
- cursor_down();
- break;
- case 'C': // right
- if (this_buf[*timeout-1] == '5') {
- word_right();
- } else {
- cursor_right();
- }
- break;
- case 'D': // left
- if (this_buf[*timeout-1] == '5') {
- word_left();
- } else {
- cursor_left();
- }
- break;
- case 'H': // home
- cursor_home();
- break;
- case 'F': // end
- cursor_end();
- break;
- case '~':
- switch (this_buf[*timeout-1]) {
- case '6':
- goto_line(env->line_no + global_config.term_height - 6);
- break;
- case '5':
- goto_line(env->line_no - (global_config.term_height - 6));
- break;
- case '3':
- if (env->mode == MODE_INSERT) {
- if (env->col_no < env->lines[env->line_no - 1]->actual + 1) {
- line_delete(env->lines[env->line_no - 1], env->col_no, env->line_no - 1);
- redraw_line(env->line_no - env->offset - 1, env->line_no-1);
- set_modified();
- redraw_statusbar();
- place_cursor_actual();
- } else if (env->line_no < env->line_count) {
- merge_lines(env->lines, env->line_no);
- redraw_text();
- set_modified();
- redraw_statusbar();
- place_cursor_actual();
- }
- }
- break;
- }
- break;
- default:
- render_error("Unrecognized escape sequence identifier: %c", c);
- break;
- }
- *timeout = 0;
- return 0;
- }
- *timeout = 0;
- return 0;
- }
- /**
- * Yank lines between line start and line end (which may be in either order)
- */
- void yank_lines(int start, int end) {
- if (global_config.yanks) {
- for (unsigned int i = 0; i < global_config.yank_count; ++i) {
- free(global_config.yanks[i]);
- }
- free(global_config.yanks);
- }
- int lines_to_yank;
- int start_point;
- if (start <= end) {
- lines_to_yank = end - start + 1;
- start_point = start - 1;
- } else {
- lines_to_yank = start - end + 1;
- start_point = end - 1;
- }
- global_config.yanks = malloc(sizeof(line_t *) * lines_to_yank);
- global_config.yank_count = lines_to_yank;
- for (int i = 0; i < lines_to_yank; ++i) {
- global_config.yanks[i] = malloc(sizeof(line_t) + sizeof(char_t) * (env->lines[start_point+i]->available));
- global_config.yanks[i]->available = env->lines[start_point+i]->available;
- global_config.yanks[i]->actual = env->lines[start_point+i]->actual;
- global_config.yanks[i]->istate = 0;
- memcpy(&global_config.yanks[i]->text, &env->lines[start_point+i]->text, sizeof(char_t) * (env->lines[start_point+i]->actual));
- for (int j = 0; j < global_config.yanks[i]->actual; ++j) {
- global_config.yanks[i]->text[j].flags = 0;
- }
- }
- }
- /**
- * LINE SELECTION mode
- *
- * Equivalent to visual line in vim; selects lines of texts.
- */
- void line_selection_mode(void) {
- int start_line = env->line_no;
- int prev_line = start_line;
- env->mode = MODE_LINE_SELECTION;
- redraw_commandline();
- int c;
- int timeout = 0;
- int this_buf[20];
- for (int j = 0; j < env->lines[env->line_no-1]->actual; ++j) {
- env->lines[env->line_no-1]->text[j].flags = FLAG_SELECT;
- }
- redraw_line(env->line_no - env->offset - 1, env->line_no-1);
- void _redraw_line(int line) {
- if (line == start_line) return;
- if ((env->line_no < start_line && line < env->line_no) ||
- (env->line_no > start_line && line > env->line_no) ||
- (env->line_no == start_line && line != start_line)) {
- recalculate_syntax(env->lines[line-1],line-1);
- } else {
- for (int j = 0; j < env->lines[line-1]->actual; ++j) {
- env->lines[line-1]->text[j].flags = FLAG_SELECT;
- }
- }
- if (line - env->offset + 1 > 1 &&
- line - env->offset - 1< global_config.term_height - global_config.bottom_size - 1) {
- redraw_line(line - env->offset - 1, line-1);
- }
- }
- while ((c = bim_getch())) {
- if (c == -1) {
- if (timeout && this_buf[timeout-1] == '\033') {
- goto _leave_select_line;
- }
- timeout = 0;
- continue;
- } else {
- if (timeout == 0) {
- switch (c) {
- case '\033':
- if (timeout == 0) {
- this_buf[timeout] = c;
- timeout++;
- }
- break;
- case DELETE_KEY:
- case BACKSPACE_KEY:
- cursor_left();
- break;
- case ':':
- /* Switch to command mode */
- command_mode();
- break;
- case '/':
- /* Switch to search mode */
- search_mode();
- break;
- case 'V':
- goto _leave_select_line;
- case 'n':
- search_next();
- break;
- case 'j':
- cursor_down();
- break;
- case 'k':
- cursor_up();
- break;
- case 'h':
- cursor_left();
- break;
- case 'l':
- cursor_right();
- break;
- case 'y':
- yank_lines(start_line, env->line_no);
- goto _leave_select_line;
- case 'D':
- case 'd':
- yank_lines(start_line, env->line_no);
- if (start_line <= env->line_no) {
- int lines_to_delete = env->line_no - start_line + 1;
- for (int i = 0; i < lines_to_delete; ++i) {
- remove_line(env->lines, start_line-1);
- }
- env->line_no = start_line;
- } else {
- int lines_to_delete = start_line - env->line_no + 1;
- for (int i = 0; i < lines_to_delete; ++i) {
- remove_line(env->lines, env->line_no-1);
- }
- }
- set_modified();
- goto _leave_select_line;
- case ' ':
- goto_line(env->line_no + global_config.term_height - 6);
- break;
- case '$':
- env->col_no = env->lines[env->line_no-1]->actual+1;
- break;
- case '0':
- env->col_no = 1;
- break;
- }
- } else {
- if (handle_escape(this_buf,&timeout,c)) {
- bim_unget(c);
- goto _leave_select_line;
- }
- }
- /* Mark current line */
- _redraw_line(env->line_no);
- /* Properly mark everything in the span we just moved through */
- if (prev_line < env->line_no) {
- for (int i = prev_line; i < env->line_no; ++i) {
- _redraw_line(i);
- }
- prev_line = env->line_no;
- } else if (prev_line > env->line_no) {
- for (int i = env->line_no + 1; i <= prev_line; ++i) {
- _redraw_line(i);
- }
- prev_line = env->line_no;
- }
- place_cursor_actual();
- }
- }
- _leave_select_line:
- env->mode = MODE_NORMAL;
- for (int i = 0; i < env->line_count; ++i) {
- recalculate_syntax(env->lines[i],i);
- }
- redraw_all();
- }
- /**
- * INSERT mode
- *
- * Accept input into the text buffer.
- */
- void insert_mode(void) {
- int cin;
- uint32_t c;
- /* Set mode line */
- env->mode = MODE_INSERT;
- redraw_commandline();
- /* Place the cursor in the text area */
- place_cursor_actual();
- int timeout = 0;
- int this_buf[20];
- uint32_t istate = 0;
- while ((cin = bim_getch())) {
- if (cin == -1) {
- if (timeout && this_buf[timeout-1] == '\033') {
- leave_insert();
- return;
- }
- timeout = 0;
- continue;
- }
- if (!decode(&istate, &c, cin)) {
- if (timeout == 0) {
- switch (c) {
- case '\033':
- if (timeout == 0) {
- this_buf[timeout] = c;
- timeout++;
- }
- break;
- case DELETE_KEY:
- case BACKSPACE_KEY:
- if (env->col_no > 1) {
- line_delete(env->lines[env->line_no - 1], env->col_no - 1, env->line_no - 1);
- env->col_no -= 1;
- redraw_line(env->line_no - env->offset - 1, env->line_no-1);
- set_modified();
- redraw_statusbar();
- place_cursor_actual();
- } else if (env->line_no > 1) {
- int tmp = env->lines[env->line_no - 2]->actual;
- merge_lines(env->lines, env->line_no - 1);
- env->line_no -= 1;
- env->col_no = tmp+1;
- redraw_text();
- set_modified();
- redraw_statusbar();
- place_cursor_actual();
- }
- break;
- case ENTER_KEY:
- if (env->col_no == env->lines[env->line_no - 1]->actual + 1) {
- env->lines = add_line(env->lines, env->line_no);
- } else {
- /* oh oh god we're all gonna die */
- env->lines = split_line(env->lines, env->line_no, env->col_no - 1);
- }
- env->col_no = 1;
- env->line_no += 1;
- if (env->line_no > env->offset + global_config.term_height - global_config.bottom_size - 1) {
- env->offset += 1;
- }
- redraw_text();
- set_modified();
- redraw_statusbar();
- place_cursor_actual();
- break;
- case '\t':
- if (env->tabs) {
- insert_char('\t');
- } else {
- for (int i = 0; i < env->tabstop; ++i) {
- insert_char(' ');
- }
- }
- redraw_statusbar();
- place_cursor_actual();
- break;
- default:
- insert_char(c);
- redraw_statusbar();
- place_cursor_actual();
- break;
- }
- } else {
- if (handle_escape(this_buf,&timeout,c)) {
- bim_unget(c);
- leave_insert();
- return;
- }
- }
- } else if (istate == UTF8_REJECT) {
- istate = 0;
- }
- }
- }
- static void show_usage(char * argv[]) {
- printf(
- "bim - Text editor\n"
- "\n"
- "usage: %s [-s] [path]\n"
- "\n"
- " -s \033[3mdisable automatic syntax highlighting\033[0m\n"
- " -? \033[3mshow this help text\033[0m\n"
- "\n", argv[0]);
- }
- int main(int argc, char * argv[]) {
- int opt;
- while ((opt = getopt(argc, argv, "?sR")) != -1) {
- switch (opt) {
- case 's':
- global_config.hilight_on_open = 0;
- break;
- case 'R':
- global_config.initial_file_is_read_only = 1;
- break;
- case '?':
- show_usage(argv);
- return 0;
- }
- }
- initialize();
- if (argc > optind) {
- open_file(argv[optind]);
- if (global_config.initial_file_is_read_only) {
- env->readonly = 1;
- }
- } else {
- env = buffer_new();
- update_title();
- setup_buffer(env);
- }
- redraw_all();
- while (1) {
- place_cursor_actual();
- int c;
- int timeout = 0;
- int this_buf[20];
- while ((c = bim_getch())) {
- if (timeout == 0) {
- switch (c) {
- case '\033':
- if (timeout == 0) {
- this_buf[timeout] = c;
- timeout++;
- }
- break;
- case DELETE_KEY:
- case BACKSPACE_KEY:
- cursor_left();
- break;
- case ':':
- /* Switch to command mode */
- command_mode();
- break;
- case '/':
- /* Switch to search mode */
- search_mode();
- break;
- case 'V':
- line_selection_mode();
- break;
- case 'n':
- search_next();
- break;
- case 'j':
- cursor_down();
- break;
- case 'k':
- cursor_up();
- break;
- case 'h':
- cursor_left();
- break;
- case 'l':
- cursor_right();
- break;
- case 'd':
- remove_line(env->lines, env->line_no-1);
- env->col_no = 1;
- if (env->line_no > env->line_count) {
- env->line_no--;
- }
- redraw_text();
- set_modified();
- place_cursor_actual();
- break;
- case ' ':
- goto_line(env->line_no + global_config.term_height - 6);
- break;
- case 'O':
- {
- if (env->readonly) goto _readonly;
- env->lines = add_line(env->lines, env->line_no-1);
- env->col_no = 1;
- redraw_text();
- set_modified();
- place_cursor_actual();
- goto _insert;
- }
- case 'o':
- {
- if (env->readonly) goto _readonly;
- env->lines = add_line(env->lines, env->line_no);
- env->col_no = 1;
- env->line_no += 1;
- if (env->line_no > env->offset + global_config.term_height - global_config.bottom_size - 1) {
- env->offset += 1;
- }
- redraw_text();
- set_modified();
- place_cursor_actual();
- goto _insert;
- }
- case 'a':
- if (env->col_no < env->lines[env->line_no-1]->actual + 1) {
- env->col_no += 1;
- }
- goto _insert;
- case 'P':
- case 'p':
- if (global_config.yanks) {
- for (unsigned int i = 0; i < global_config.yank_count; ++i) {
- env->lines = add_line(env->lines, env->line_no - (c == 'P' ? 1 : 0));
- }
- for (unsigned int i = 0; i < global_config.yank_count; ++i) {
- replace_line(env->lines, env->line_no - (c == 'P' ? 1 : 0) + i, global_config.yanks[i]);
- }
- for (int i = 0; i < env->line_count; ++i) {
- env->lines[i]->istate = 0;
- }
- for (int i = 0; i < env->line_count; ++i) {
- recalculate_syntax(env->lines[i],i);
- }
- redraw_all();
- }
- break;
- case '$':
- env->col_no = env->lines[env->line_no-1]->actual+1;
- break;
- case '0':
- env->col_no = 1;
- break;
- case 'i':
- _insert:
- if (env->readonly) goto _readonly;
- insert_mode();
- redraw_statusbar();
- redraw_commandline();
- timeout = 0;
- break;
- _readonly:
- render_error("Buffer is read-only");
- break;
- case 12:
- redraw_all();
- break;
- }
- } else {
- handle_escape(this_buf,&timeout,c);
- }
- place_cursor_actual();
- }
- }
- return 0;
- }
|