Try to not repeat function in Lua

Question

There is a way to not rewrite the same part (the cycle for i = 1...end) twice?

local function myfunc(frame,method)
 if frame:IsShown() then
 for i = 1, #frame.buttons do
 frame.buttons[i]:HookScript("OnMouseDown", method)
 end
 else
 self:SetScript(
 "OnShow",
 function(frame)
 for i = 1, #frame.buttons do
 frame.buttons[i]:HookScript("OnMouseDown", method)
 end
 end
 )
 end
end

Where

• IsShown(), HookScript() and SetScript() are all in-game API

Answer 1

Not a completely review, but this should work as intended:

local function myfunc(frame)
 function hookFunc(frame)
 for i = 1, #frame.buttons do
 frame.buttons[i]:HookScript("OnMouseDown", method)
 end
 end
 if frame:IsShown() then
 hookFunc(frame)
 else
 self:SetScript(
 "OnShow",
 hookFunc(frame)
 )
 end
end

• user3204810: I forgot the method argument in my original code.
• Tobias Grothe: My code should be adaptable accordingly.
• Tobias Grothe: For more explanation, see here.

Answer 2

Maybe a recursive function? It seems to work:

local function myfunc(frame,method)
 if frame:IsShown() then
 for i = 1, #frame.buttons do
 frame.buttons[i]:HookScript("OnMouseDown", method)
 end
 else
 frame:SetScript(
 "OnShow", --when "OnShow" is fired, IsShown() become true
 function()
 myfunc(frame,method)
 end
 )
 end
end