Script: nick_complete_wrapper.lua

Wrap nick completion with custom prefix and/or suffix.
Author: singalaut — Version: 0.1 — License: WTFPL
For WeeChat ≥ 0.3.8.
Tags: nick
Added: 2016-10-30

Download GitHub Repository

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
--[[
Author: singalaut <rumia.youkai.of.dusk@gmail.com>
License: WTFPL
URL: https://github.com/tomoe-mami/weechat-scripts/tree/master/nick_complete_wrapper

A script for adding custom prefix and/or suffix when doing nick completion.
Originally written to make nick completion in Bitlbee #twitter_* channels
prepended with '@'.

To use this script, set local variable `ncw_prefix` and/or `ncw_suffix` in the
buffer you want with `/buffer set` command. For example:

    /buffer set localvar_set_ncw_prefix @
    /buffer set localvar_set_ncw_suffix >

To delete those variables, replace `localvar_set` with `localvar_del`.
For example:

    /buffer set localvar_del_ncw_suffix 1

The '1' in the example above is just a dummy arg. Weechat requires 2 args
after `/buffer set `.

---------------------------------------------------------------------------------

For Lua < 5.3, luautf8 (https://luarocks.org/modules/xavier-wang/luautf8) module
is an optional (but recommended) dependency. In Lua 5.3, the script doesn't
have any dependency.
--]]

w, script_name = weechat, "nick_complete_wrapper"
config, hooks = {}, {}

function main()
   local reg = w.register(
      script_name, "singalaut <https://github.com/tomoe-mami>", "0.1", "WTFPL",
      "Wraps nick completion with prefix and/or suffix",
      "", "")
   if reg then
      check_utf8_support()
      config.nick_add_space = w.config_get("weechat.completion.nick_add_space")
      w.hook_command_run("9000|/input complete_*", "complete_cb", "")
   end
end

function check_utf8_support()
   if utf8 and type(utf8.len) == "function" then
      -- lua 5.3 with builtin utf8 support
      string_length = utf8.len
   else
      -- lua < 5.3 with [lua-]utf8 module
      local pkgs = { "lua-utf8", "utf8" }
      for _, searcher in ipairs(package.searchers or package.loaders) do
         for _, pkg_name in ipairs(pkgs) do
            local loader = searcher(pkg_name)
            if type(loader) == "function" then
               package.preload[pkg_name] = loader
               utf8 = require(pkg_name)
               if type(utf8.len) == "function" then
                  string_length = utf8.len
                  return
               end
            end
         end
      end
   end
   if not string_length then
      string_length = w.strlen_screen
   end
end

function get_completion(ptr_buffer)
   local t = {}
   local ptr_comp = w.hdata_pointer(w.hdata_get("buffer"), ptr_buffer, "completion")
   if ptr_comp and ptr_comp ~= "" then
      local h_comp = w.hdata_get("completion")
      t.start_pos = w.hdata_integer(h_comp, ptr_comp, "position_replace")
      t.word_found = w.hdata_string(h_comp, ptr_comp, "word_found")
      t.is_nick = w.hdata_integer(h_comp, ptr_comp, "word_found_is_nick") == 1
      t.is_command = w.hdata_string(h_comp, ptr_comp, "base_command") ~= ""
      if not t.is_command and t.word_found == "" then
         local last_nick = w.buffer_get_string(ptr_buffer, "localvar_ncw_last_nick")
         if last_nick ~= "" then
            t.word_found, t.is_nick = last_nick, true
            t.start_pos = tonumber(w.buffer_get_string(ptr_buffer, "localvar_ncw_last_pos")) or 0
            w.buffer_set(ptr_buffer, "localvar_del_ncw_last_nick", "")
            w.buffer_set(ptr_buffer, "localvar_del_ncw_last_pos", "")
         end
      end
      return t
   end
end

function get_prefix_suffix(ptr_buffer)
   return {
      prefix = w.buffer_get_string(ptr_buffer, "localvar_ncw_prefix"),
      suffix = w.buffer_get_string(ptr_buffer, "localvar_ncw_suffix")
   }
end

function cleanup_previous_completion(ptr_buffer)
   local ps = get_prefix_suffix(ptr_buffer)
   if ps.prefix == "" and ps.suffix == "" then
      return w.WEECHAT_RC_OK
   end
   local comp = get_completion(ptr_buffer)
   if comp and comp.is_nick and not comp.is_command then
      local current_pos = w.buffer_get_integer(ptr_buffer, "input_pos")
      local input = w.buffer_get_string(ptr_buffer, "input")
      local space = w.config_boolean(config.nick_add_space) and " " or ""
      local str_nick = ps.prefix..comp.word_found..ps.suffix..space
      local str_before = input:sub(1, comp.start_pos)
      if string_length(str_before..str_nick) == current_pos then
         w.buffer_set(ptr_buffer, "completion_freeze", "1")
         w.buffer_set(ptr_buffer, "input", str_before..comp.word_found..input:sub(comp.start_pos + #str_nick))
         w.buffer_set(ptr_buffer, "input_pos", string_length(str_before..comp.word_found..space))
         w.buffer_set(ptr_buffer, "completion_freeze", "0")
         w.buffer_set(ptr_buffer, "localvar_set_ncw_last_nick", comp.word_found)
         w.buffer_set(ptr_buffer, "localvar_set_ncw_last_pos", comp.start_pos)
      else
         w.buffer_set(ptr_buffer, "localvar_del_ncw_last_nick", "")
         w.buffer_set(ptr_buffer, "localvar_del_ncw_last_pos", "")
      end
   end
end

function complete_cb(_, ptr_buffer)
   cleanup_previous_completion(ptr_buffer)
   hooks[ptr_buffer] = w.hook_signal("input_text_changed", "input_changed_cb", ptr_buffer)
   return w.WEECHAT_RC_OK
end

function input_changed_cb(ptr_buffer)
   if not hooks[ptr_buffer] then
      return w.WEECHAT_RC_OK
   end
   w.unhook(hooks[ptr_buffer])
   hooks[ptr_buffer] = nil

   local ps = get_prefix_suffix(ptr_buffer)
   if ps.prefix == "" and ps.suffix == "" then
      return w.WEECHAT_RC_OK
   end

   local comp = get_completion(ptr_buffer)
   if not comp or comp.is_command or not comp.is_nick then
      return w.WEECHAT_RC_OK
   end

   local str_nick = ps.prefix..comp.word_found..ps.suffix
   if str_nick ~= comp.word_found then
      local input = w.buffer_get_string(ptr_buffer, "input")
      local current_pos = w.buffer_get_integer(ptr_buffer, "input_pos")
      local str_before = input:sub(1, comp.start_pos)
      local add_space = w.config_boolean(config.nick_add_space) and 1 or 0
      local str_after = input:sub(comp.start_pos + #comp.word_found + add_space)
      w.buffer_set(ptr_buffer, "completion_freeze", "1")
      w.buffer_set(ptr_buffer, "input", str_before..str_nick..str_after)
      w.buffer_set(ptr_buffer, "input_pos", string_length(str_before..str_nick) + add_space)
      w.buffer_set(ptr_buffer, "completion_freeze", "0")
   end

   return w.WEECHAT_RC_OK
end

main()