-- Copyright 2006-2019 Mitchell mitchell.att.foicica.com. See License.txt. -- Lexer enhanced to conform to the realities of Prologs on the ground by -- Michael T. Richter. Copyright is explicitly assigned back to Mitchell. -- Prolog LPeg lexer. --[[ Prologs are notoriously fractious with many barely-compatible dialects. To make Textadept more useful for these cases, directives and keywords are grouped by dialect. Selecting a dialect is a simple matter of setting the buffer/lexer property "prolog.dialect" in init.lua. Dialects currently in the lexer file are: - 'iso': the generic ISO standard without modules. - 'gprolog': GNU Prolog. - 'swipl': SWI-Prolog. The default dialect is 'iso' if none is defined. (You probably don't want this.) Note that there will be undoubtedly duplicated entries in various categories because of the flexibility of Prolog and the automated tools used to gather most information. This is not an issue, however, because directives override arity-0 predicates which override arity-1+ predicates which override bifs which override operators. ]] local lexer = require('lexer') local token, word_match = lexer.token, lexer.word_match local P, R, S, B, V, C = lpeg.P, lpeg.R, lpeg.S, lpeg.B, lpeg.V, lpeg.C local lex = lexer.new('prolog') local dialects = setmetatable({gprolog = 'gprolog', swipl = 'swipl'}, {__index = function(_, _) return 'iso' end}) local dialect = dialects[lexer.property['prolog.dialect']] -- Directives. local directives = {} directives.iso = [[ -- Gathered by inspection of GNU Prolog documentation. dynamic multifile discontiguous include ensure_loaded op char_conversion set_prolog_flag initialization ]] directives.gprolog = directives.iso .. [[ -- Gathered by inspection of GNU Prolog documentation. public ensure_linked built_in if else endif elif foreign ]] directives.swipl = directives.iso .. [[ -- Gathered by liberal use of grep on the SWI source and libraries. coinductive current_predicate_option expects_dialect http_handler listen module multifile use_foreign_library use_module dynamic http_handler initialization json_object multifile record use_module abolish arithmetic_function asserta at_halt begin_tests chr_constraint chr_option chr_type clear_cache constraints consult create_prolog_flag current_prolog_flag debug discontiguous dynamic elif else encoding end_tests endif expects_dialect export forall format format_predicate html_meta html_resource http_handler http_request_expansion if include init_color_term_flag init_options initialization json_object lazy_list_iterator license listen load_extensions load_files load_foreign_library meta_predicate mode module module_transparent multifile noprofile op pce_begin_class pce_end_class pce_global pce_group persistent pop_operators pred predicate_options print_message prolog_load_context prompt public push_hprolog_library push_ifprolog_library, push_operators push_sicstus_library push_xsb_library push_yap_library, quasi_quotation_syntax record redefine_system_predicate reexport register_iri_scheme residual_goals retract set_module set_prolog_flag set_script_dir set_test_options setenv setting style_check table thread_local thread_local message type use_class_template use_foreign_library use_module utter volatile build_schema chr_constraint chr_option chr_type cql_option determinate discontiguous dynamic endif format_predicate if initialization license meta_predicate mode module multifile op reexport thread_local use_module volatile ]] lex:add_rule('directive', token(lexer.WHITESPACE, lexer.starts_line(S(' \t'))^0) * token(lexer.OPERATOR, P':-') * token(lexer.WHITESPACE, S(' \t')^0) * token(lexer.PREPROCESSOR, P(word_match(directives[dialect])))) -- Whitespace. lex:add_rule('whitespace', token(lexer.WHITESPACE, lexer.space^1)) -- Keywords. local zero_arity_keywords = {} zero_arity_keywords.iso = [[ -- eyeballed from GNU Prolog documentation true fail pi float_overflow int_overflow int_underflow undefined asserta assertz retract retractall clause abolish current_predicate findall bagof setof at_end_of_stream flush_output nl halt false ]] zero_arity_keywords.gprolog = [[ -- Collected automatically via current_predicate/1 with some cleanup. at_end_of_stream wam_debug listing flush_output fail told false top_level shell trace debugging seen repeat abort nl statistics halt notrace randomize true nospyall nodebug debug stop break ]] zero_arity_keywords.swipl = [[ -- Collected automatically via current_predicate/1 with some cleanup. noprotocol compiling ttyflush true abort license known_licenses print_toplevel_variables initialize mutex_statistics break reset_profiler win_has_menu version prolog abolish_nonincremental_tables false halt undefined abolish_all_tables reload_library_index garbage_collect repeat nospyall tracing trace notrace trim_stacks garbage_collect_clauses garbage_collect_atoms mutex_unlock_all seen told nl debugging fail at_end_of_stream attach_packs flush_output true ]] local one_plus_arity_keywords = {} one_plus_arity_keywords.iso = [[ -- eyeballed from GNU Prolog documentation call catch throw var nonvar atom integer float number atomic compound callable ground unify_with_occurs_check compare functor arg copy_term term_variables subsumes_term acyclic_term predicate_property current_input current_output set_input set_output open close current_stream stream_property set_stream_position get_char get_code is peek_char peek_code put_char putcode get_byte peek_byte read_term read write_term write writeq write_canonical char_conversion current_char_conversion call once repeat atom_length atom_concat sub_atom char_code atom_chars atom_codes ]] one_plus_arity_keywords.gprolog = [[ -- Collected automatically via current_predicate/1 with some cleanup. abolish absolute_file_name acyclic_term add_linedit_completion add_stream_alias add_stream_mirror append architecture arg argument_counter argument_list argument_value asserta assertz at_end_of_stream atom atom_chars atom_codes atom_concat atom_length atom_property atomic bagof between bind_variables call call_det call_with_args callable catch change_directory char_code char_conversion character_count clause close close_input_atom_stream close_input_chars_stream close_input_codes_stream close_output_atom_stream close_output_chars_stream close_output_codes_stream compare compound consult copy_term cpu_time create_pipe current_alias current_atom current_bip_name current_char_conversion current_input current_mirror current_op current_output current_predicate current_prolog_flag current_stream date_time decompose_file_name delete delete_directory delete_file directory_files display display_to_atom display_to_chars display_to_codes environ exec expand_term fd_all_different fd_at_least_one fd_at_most_one fd_atleast fd_atmost fd_cardinality fd_dom fd_domain fd_domain_bool fd_element fd_element_var fd_exactly fd_has_extra_cstr fd_has_vector fd_labeling fd_labelingff fd_max fd_max_integer fd_maximize fd_min fd_minimize fd_not_prime fd_only_one fd_prime fd_reified_in fd_relation fd_relationc fd_set_vector_max fd_size fd_use_vector fd_var fd_vector_max file_exists file_permission file_property find_linedit_completion findall flatten float flush_output for forall fork_prolog format format_to_atom format_to_chars format_to_codes functor g_array_size g_assign g_assignb g_dec g_deco g_inc g_inco g_link g_read g_reset_bit g_set_bit g_test_reset_bit g_test_set_bit generic_var get get_byte get_char get_code get_key get_key_no_echo get_linedit_prompt get_print_stream get_seed get0 ground halt host_name hostname_address integer is_absolute_file_name is_list is_relative_file_name keysort last last_read_start_line_column leash length line_count line_position list list_or_partial_list listing load lower_upper make_directory maplist max_list member memberchk min_list msort name name_query_vars name_singleton_vars new_atom nl non_fd_var non_generic_var nonvar nospy nth nth0 nth1 number number_atom number_chars number_codes numbervars once op open open_input_atom_stream open_input_chars_stream open_input_codes_stream open_output_atom_stream open_output_chars_stream open_output_codes_stream os_version partial_list peek_byte peek_char peek_code permutation phrase popen portray_clause predicate_property prefix print print_to_atom print_to_chars print_to_codes prolog_file_name prolog_pid put put_byte put_char put_code random read read_atom read_from_atom read_from_chars read_from_codes read_integer read_number read_pl_state_file read_term read_term_from_atom read_term_from_chars read_term_from_codes read_token read_token_from_atom read_token_from_chars read_token_from_codes real_time remove_stream_mirror rename_file retract retractall reverse see seeing seek select send_signal set_bip_name set_input set_linedit_prompt set_output set_prolog_flag set_seed set_stream_buffering set_stream_eof_action set_stream_line_column set_stream_position set_stream_type setarg setof shell skip sleep socket socket_accept socket_bind socket_close socket_connect socket_listen sort spawn spy spypoint_condition sr_change_options sr_close sr_current_descriptor sr_error_from_exception sr_get_error_counters sr_get_file_name sr_get_include_list sr_get_include_stream_list sr_get_module sr_get_position sr_get_size_counters sr_get_stream sr_new_pass sr_open sr_read_term sr_set_error_counters sr_write_error sr_write_message statistics stream_line_column stream_position stream_property sub_atom sublist subsumes_term subtract succ suffix sum_list syntax_error_info system system_time tab tell telling temporary_file temporary_name term_hash term_ref term_variables throw unget_byte unget_char unget_code unify_with_occurs_check unlink user_time var wait working_directory write write_canonical write_canonical_to_atom write_canonical_to_chars write_canonical_to_codes write_pl_state_file write_term write_term_to_atom write_term_to_chars write_term_to_codes write_to_atom write_to_chars write_to_codes writeq writeq_to_atom writeq_to_chars writeq_to_codes ]] one_plus_arity_keywords.swipl = [[ -- Collected automatically via current_predicate/1 with some cleanup. prolog_exception_hook term_expansion expand_answer message_property resource help goal_expansion file_search_path prolog_clause_name thread_message_hook prolog_file_type goal_expansion prolog_predicate_name exception writeln term_expansion expand_query url_path message_hook library_directory resource portray prolog_load_file prolog_list_goal ansi_format source_file_property asserta call_dcg source_location wait_for_input locale_destroy set_locale read_pending_codes thread_join open_dde_conversation win_folder protocol copy_stream_data current_locale read_pending_chars win_add_dll_directory protocola thread_property win_shell goal_expansion phrase gc_file_search_cache dcg_translate_rule protocolling win_registry_get_value term_expansion dcg_translate_rule assert copy_stream_data once bagof prompt1 tnot assertz phrase sort ignore thread_statistics assert locale_create win_remove_dll_directory term_expansion read_term asserta clause assertz predicate_option_type is_thread get_single_char set_prolog_IO expand_goal ground message_queue_create locale_property close_dde_conversation goal_expansion clause zipper_open_new_file_in_zip term_to_atom with_output_to module expand_term redefine_system_predicate thread_detach dde_execute term_string read_clause compile_predicates predicate_option_mode noprofile read_term_from_atom cancel_halt non_terminal atom_to_term line_position frozen dde_request findnsols prolog_skip_level prolog_current_choice get get_attrs license var_property nb_delete unwrap_predicate zipper_open_current put_attrs dde_poke set_stream read_term zip_file_info_ memberchk seek expand_goal get0 call var integer attach_packs byte_count zipper_goto findnsols character_count expand_term get_flag atom line_count set_flag atomic tab create_prolog_flag copy_term import_module verbose_expansion b_setval duplicate_term prolog_load_context attach_packs prolog_listen b_getval prolog_frame_attribute prompt copy_term_nat nb_linkval tab prolog_choice_attribute set_prolog_flag nb_getval prolog_skip_frame del_attrs skip sort license open_null_stream nb_current prolog_listen msort is_list is_stream get keysort win_shell prolog_unlisten notrace get0 add_import_module wildcard_match profiler delete_directory trie_gen_compiled expand_file_name file_name_extension delete_file writeq win_module_file call write get_dict win_exec directory_files trie_insert make_directory engine_next_reified del_dict sleep getenv call_continuation trie_gen_compiled prolog_to_os_filename is_absolute_file_name trie_insert engine_fetch engine_create strip_module call delete_import_module write_canonical compile_aux_clauses setenv callable is_engine write_term call set_module call halt catch findall trie_gen trie_destroy rename_file shift unify_with_occurs_check engine_yield forall unsetenv trie_term file_directory_name version current_engine file_base_name engine_self import trie_gen trie_lookup write_term trie_update freeze engine_post export put_dict same_file trie_new call trie_delete start_tabling is_trie residual_goals thread_peek_message thread_get_message dict_pairs set_end_of_stream call_cleanup current_predicate arg dict_create thread_setconcurrency read_link is_dict at_halt tmp_file not put_dict setup_call_cleanup abolish_nonincremental_tables time_file start_subsumptive_tabling char_conversion compound sub_atom access_file call call_cleanup abolish nonvar current_functor abolish_module_tables subsumes_term engine_post call retractall compare engine_next prolog_cut_to size_file current_char_conversion predicate_property nonground engine_destroy message_queue_property format abolish qcompile thread_send_message stream_pair message_queue_create same_term number select_dict catch_with_backtrace thread_get_message thread_send_message win_insert_menu_item message_queue_set exists_directory copy_term nb_set_dict prolog_nodebug functor current_table cyclic_term untable read exists_file thread_peek_message b_set_dict engine_create prolog_debug acyclic_term writeln get_dict compound_name_arity abolish_table_subgoals start_tabling trie_insert nb_link_dict message_queue_destroy thread_get_message is_dict nth_clause absolute_file_name term_singletons make_library_index set_output retract context_module current_trie term_attvars load_files get_char ensure_loaded current_input prolog_current_frame make_library_index term_variables compound_name_arguments reexport autoload_path get_code set_input flag thread_create use_module findall thread_join call_with_inference_limit var_number dwim_match consult peek_code close nospy print_message term_variables trie_property read_history get_byte default_module get_byte print on_signal get_char call_residue_vars dwim_match atom_prefix unifiable use_module numbervars load_files get_code open format_time copy_predicate_clauses reexport leash current_output sub_string close format_time atom_codes stamp_date_time require name open_shared_object open atom_chars current_predicate format tmp_file_stream term_hash rational source_file reset atom_concat atom_length current_prolog_flag rational dwim_predicate date_time_stamp stream_property string_upper setlocale format writeln current_module normalize_space writeq current_flag shell upcase_atom qcompile char_code atomic_concat read string_lower write term_string numbervars working_directory number_codes set_prolog_gc_thread downcase_atom format_predicate number_string open_shared_object style_check char_type print stream_position_data code_type write_canonical number_chars length current_arithmetic_function atomic_list_concat del_attr read_string zip_unlock open_resource string_length zip_lock see erase open_resource setof atomic_list_concat current_format_predicate current_resource with_mutex atomics_to_string term_hash absolute_file_name deterministic current_atom thread_create collation_key get_attr variant_hash string_concat atom_number put put_attr variant_sha1 thread_signal mutex_unlock tty_size current_key mutex_create fill_buffer expand_file_search_path blob shell register_iri_scheme skip fast_read divmod mutex_trylock thread_self put mutex_property fast_write mutex_lock current_blob sub_atom_icasechk mutex_destroy fast_term_serialized split_string set_stream_position recorda telling setarg thread_exit zip_open_stream instance mutex_create statistics append get_time zip_close_ tell atomics_to_string clause_property attvar zip_clone seeing nth_integer_root_and_remainder recorda put_byte string_chars spy recordz print_message_lines current_op put_char nl source_file string_codes op setup_call_catcher_cleanup nb_linkarg recorded put_code peek_byte apply module_property atom_string nb_setarg succ recordz message_to_string close_shared_object peek_char between recorded visible plus call_shared_object_function peek_code peek_byte set_prolog_stack float throw at_end_of_stream get_string_code call_with_depth_limit random_property flush_output peek_string open_xterm peek_char open_string string_code set_random prolog_stack_property put_char unload_file nb_setval put_byte current_signal put_code write_length string read_string text_to_string ]] lex:add_rule('keyword', token(lexer.KEYWORD, word_match(zero_arity_keywords[dialect]) + (word_match(one_plus_arity_keywords[dialect]) * #(P'(')))) -- BIFs. local bifs = {} bifs.iso = [[ -- eyeballed from GNU Prolog documentation xor abs sign min max sqrt tan atan atan2 cos acos sin asin exp log float ceiling floor round truncate float_fractional_part float_integer_part rem div mod ]] bifs.gprolog = bifs.iso .. [[ -- eyeballed from GNU Prolog documentation inc dec lsb msb popcount gcd tanh atanh cosh acosh sinh asinh log10 rnd ]] bifs.swipl = [[ -- Collected automatically via current_arithmetic_function/1 with some -- cleanup. abs acos acosh asinh atan atan atanh atan2 ceil ceiling copysign cos cosh cputime div getbit e epsilon erf erfc eval exp float float_fractional_part float_integer_part floor gcd inf integer lgamma log log10 lsb max min mod msb nan pi popcount powm random random_float rational rationalize rdiv rem round sign sin sinh sqrt tan tanh truncate xor ]] lex:add_rule('bif', token(lexer.FUNCTION, word_match(bifs[dialect]) * #(P'('))) -- Numbers. local decimal_group = S('+-')^-1 * (lexer.digit + '_')^1 local binary_number = '0b' * (S('01') + '_')^1 local character_code = '0\'' * S('\\')^-1 * (lexer.print - lexer.space) local decimal_number = decimal_group * ('.' * decimal_group)^-1 * ('e' * decimal_group)^-1 local hexadecimal_number = '0x' * (lexer.xdigit + '_')^1 local octal_number = '0o' * (S('01234567') + '_')^1 lex:add_rule('number', token(lexer.NUMBER, character_code + binary_number + hexadecimal_number + octal_number + decimal_number)) -- Comments. local line_comment = '%' * lexer.nonnewline^0 local block_comment = '/*' * (lexer.any - '*/')^0 * P('*/')^-1 lex:add_rule('comment', token(lexer.COMMENT, line_comment + block_comment)) -- Operators. local operators = {} operators.iso = [[ -- Collected automatically via current_op/3 with some cleanup and comparison -- to docs. rem div mod is ]] operators.gprolog = operators.iso -- GNU Prolog's textual operators are the same operators.swipl = [[ -- Collected automatically via current_op/3 with some cleanup. is as volatile mod discontiguous div rdiv meta_predicate public xor module_transparent multifile table dynamic thread_initialization thread_local initialization rem ]] lex:add_rule('operator', token(lexer.OPERATOR, word_match(operators[dialect]) + S('-!+\\|=:;&<>()[]{}/*^@?.'))) -- Variables. lex:add_rule('variable', token(lexer.VARIABLE, (lexer.upper + '_') * (lexer.word^1 + lexer.digit^1 + P('_')^1)^0)) -- Identifiers. lex:add_rule('identifier', token(lexer.IDENTIFIER, lexer.word)) -- Strings. lex:add_rule('string', token(lexer.STRING, lexer.delimited_range("'", true) + lexer.delimited_range('"', true))) return lex