lua: extension: refactor error as early return

Those commits are mostly doing minor refactor where early return is used instead of branch nesting.