"""Execute code from an editor.

Check module: do a full syntax check of the current module.
Also run the tabnanny to catch any inconsistent tabs.

Run module: also execute the module's code in the __main__ namespace.
The window must have been saved previously. The module is added to
sys.modules, and is also added to the __main__ namespace.

TODO: Specify command line arguments in a dialog box.
"""
import os
import tabnanny
import tokenize

import tkinter.messagebox as tkMessageBox

from idlelib.config import idleConf
from idlelib import macosx
from idlelib import pyshell
from idlelib.query import CustomRun
from idlelib import outwin

indent_message = """Error: Inconsistent indentation detected!

1) Your indentation is outright incorrect (easy to fix), OR

2) Your indentation mixes tabs and spaces.

To fix case 2, change all tabs to spaces by using Edit->Select All followed \
by Format->Untabify Region and specify the number of columns used by each tab.
"""


class ScriptBinding:

 def __init__(self, editwin):
 self.editwin = editwin
 # Provide instance variables referenced by debugger
 # XXX This should be done differently
 self.flist = self.editwin.flist
 self.root = self.editwin.root
 # cli_args is list of strings that extends sys.argv
 self.cli_args = []

 if macosx.isCocoaTk():
 self.editwin.text_frame.bind('<<run-module-event-2>>', self._run_module_event)

 def check_module_event(self, event):
 if isinstance(self.editwin, outwin.OutputWindow):
 self.editwin.text.bell()
 return 'break'
 filename = self.getfilename()
 if not filename:
 return 'break'
 if not self.checksyntax(filename):
 return 'break'
 if not self.tabnanny(filename):
 return 'break'
 return "break"

 def tabnanny(self, filename):
 # XXX: tabnanny should work on binary files as well
 with tokenize.open(filename) as f:
 try:
 tabnanny.process_tokens(tokenize.generate_tokens(f.readline))
 except tokenize.TokenError as msg:
 msgtxt, (lineno, start) = msg.args
 self.editwin.gotoline(lineno)
 self.errorbox("Tabnanny Tokenizing Error",
 "Token Error: %s" % msgtxt)
 return False
 except tabnanny.NannyNag as nag:
 # The error messages from tabnanny are too confusing...
 self.editwin.gotoline(nag.get_lineno())
 self.errorbox("Tab/space error", indent_message)
 return False
 return True

 def checksyntax(self, filename):
 self.shell = shell = self.flist.open_shell()
 saved_stream = shell.get_warning_stream()
 shell.set_warning_stream(shell.stderr)
 with open(filename, 'rb') as f:
 source = f.read()
 if b'\r' in source:
 source = source.replace(b'\r\n', b'\n')
 source = source.replace(b'\r', b'\n')
 if source and source[-1] != ord(b'\n'):
 source = source + b'\n'
 editwin = self.editwin
 text = editwin.text
 text.tag_remove("ERROR", "1.0", "end")
 try:
 # If successful, return the compiled code
 return compile(source, filename, "exec")
 except (SyntaxError, OverflowError, ValueError) as value:
 msg = getattr(value, 'msg', '') or value or "<no detail available>"
 lineno = getattr(value, 'lineno', '') or 1
 offset = getattr(value, 'offset', '') or 0
 if offset == 0:
 lineno += 1 #mark end of offending line
 pos = "0.0 + %d lines + %d chars" % (lineno-1, offset-1)
 editwin.colorize_syntax_error(text, pos)
 self.errorbox("SyntaxError", "%-20s" % msg)
 return False
 finally:
 shell.set_warning_stream(saved_stream)

 def run_module_event(self, event):
 if macosx.isCocoaTk():
 # Tk-Cocoa in MacOSX is broken until at least
 # Tk 8.5.9, and without this rather
 # crude workaround IDLE would hang when a user
 # tries to run a module using the keyboard shortcut
 # (the menu item works fine).
 self.editwin.text_frame.after(200,
 lambda: self.editwin.text_frame.event_generate(
 '<<run-module-event-2>>'))
 return 'break'
 else:
 return self._run_module_event(event)

 def run_custom_event(self, event):
 return self._run_module_event(event, customize=True)

 def _run_module_event(self, event, *, customize=False):
 """Run the module after setting up the environment.

 First check the syntax. Next get customization. If OK, make
 sure the shell is active and then transfer the arguments, set
 the run environment's working directory to the directory of the
 module being executed and also add that directory to its
 sys.path if not already included.
 """
 if isinstance(self.editwin, outwin.OutputWindow):
 self.editwin.text.bell()
 return 'break'
 filename = self.getfilename()
 if not filename:
 return 'break'
 code = self.checksyntax(filename)
 if not code:
 return 'break'
 if not self.tabnanny(filename):
 return 'break'
 if customize:
 title = f"Customize {self.editwin.short_title()} Run"
 run_args = CustomRun(self.shell.text, title,
 cli_args=self.cli_args).result
 if not run_args: # User cancelled.
 return 'break'
 self.cli_args, restart = run_args if customize else ([], True)
 interp = self.shell.interp
 if pyshell.use_subprocess and restart:
 interp.restart_subprocess(
 with_cwd=False, filename=filename)
 dirname = os.path.dirname(filename)
 argv = [filename]
 if self.cli_args:
 argv += self.cli_args
 interp.runcommand(f"""if 1:
 __file__ = {filename!r}
 import sys as _sys
 from os.path import basename as _basename
 argv = {argv!r}
 if (not _sys.argv or
 _basename(_sys.argv[0]) != _basename(__file__) or
 len(argv) > 1):
 _sys.argv = argv
 import os as _os
 _os.chdir({dirname!r})
 del _sys, argv, _basename, _os
\n""")
 interp.prepend_syspath(filename)
 # XXX KBK 03Jul04 When run w/o subprocess, runtime warnings still
 # go to __stderr__. With subprocess, they go to the shell.
 # Need to change streams in pyshell.ModifiedInterpreter.
 interp.runcode(code)
 return 'break'

 def getfilename(self):
 """Get source filename. If not saved, offer to save (or create) file

 The debugger requires a source file. Make sure there is one, and that
 the current version of the source buffer has been saved. If the user
 declines to save or cancels the Save As dialog, return None.

 If the user has configured IDLE for Autosave, the file will be
 silently saved if it already exists and is dirty.

 """
 filename = self.editwin.io.filename
 if not self.editwin.get_saved():
 autosave = idleConf.GetOption('main', 'General',
 'autosave', type='bool')
 if autosave and filename:
 self.editwin.io.save(None)
 else:
 confirm = self.ask_save_dialog()
 self.editwin.text.focus_set()
 if confirm:
 self.editwin.io.save(None)
 filename = self.editwin.io.filename
 else:
 filename = None
 return filename

 def ask_save_dialog(self):
 msg = "Source Must Be Saved\n" + 5*' ' + "OK to Save?"
 confirm = tkMessageBox.askokcancel(title="Save Before Run or Check",
 message=msg,
 default=tkMessageBox.OK,
 parent=self.editwin.text)
 return confirm

 def errorbox(self, title, message):
 # XXX This should really be a function of EditorWindow...
 tkMessageBox.showerror(title, message, parent=self.editwin.text)
 self.editwin.text.focus_set()


if __name__ == "__main__":
 from unittest import main
 main('idlelib.idle_test.test_runscript', verbosity=2,)