sapling/eden/scm/edenscm/hgext/debugshell.py
Jun Wu 3e0b781197 py3: only use binary stdin/stdout/stderr
Summary:
Drop stdoutbytes/stdinbytes. They make things unnecessarily complicated
(especially for chg / Rust dispatch entry point).

The new idea is IO are using bytes. Text are written in utf-8 (Python 3) or
local encoding (Python 2). To make stdout behave reasonably on systems not
using utf-8 locale (ex. Windows), we might add a Rust binding to Rust's stdout,
which does the right thing:
- When writing to stdout console, expect text to be utf-8 encoded and do proper decoding.
- Wehn writing to stdout file, write the raw bytes without translation.

Note Python's `sys.stdout.buffer` does not do translation when writing to stdout console
like Rust's stdout.

For now, my main motivation of this change is to fix chg on Python 3.

Reviewed By: xavierd

Differential Revision: D19702533

fbshipit-source-id: 74704c83e1b200ff66fb3a2d23d97ff21c7239c8
2020-02-03 18:26:57 -08:00

93 lines
2.2 KiB
Python

# Portions Copyright (c) Facebook, Inc. and its affiliates.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2.
# Copyright 2010 Mercurial Contributors
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.
# debugshell extension
"""a python shell with repo, changelog & manifest objects"""
from __future__ import absolute_import
import code
import os
import sys
import bindings
import edenscm
# pyre-fixme[21]: Could not find `edenscmnative`.
import edenscmnative
from edenscm import hgext, mercurial
from edenscm.mercurial import registrar
from edenscm.mercurial.i18n import _
from edenscm.mercurial.pycompat import decodeutf8
cmdtable = {}
command = registrar.command(cmdtable)
def _assignobjects(objects, repo):
objects.update(
{
"m": mercurial,
"e": edenscm,
"n": edenscmnative,
"b": bindings,
"x": hgext,
"mercurial": mercurial,
}
)
if repo:
objects.update({"repo": repo, "cl": repo.changelog, "mf": repo.manifestlog})
# Import other handy modules
for name in ["os", "subprocess", "re"]:
objects[name] = __import__(name)
@command(
"debugshell|dbsh|debugsh",
[("c", "command", "", _("program passed in as string"), _("CMD"))],
optionalrepo=True,
)
def debugshell(ui, repo, **opts):
command = opts.get("command")
_assignobjects(locals(), repo)
if command:
exec(command)
return 0
if not ui.interactive():
command = decodeutf8(ui.fin.read())
exec(command)
return 0
bannermsg = "loaded repo: %s\n" "using source: %s" % (
repo and repo.root or "(none)",
mercurial.__path__[0],
) + (
"\n\nAvailable variables:\n"
" e: edenscm\n"
" n: edenscmnative\n"
" m: edenscm.mercurial\n"
" x: edenscm.hgext\n"
" b: bindings\n"
" ui: the ui object"
)
if repo:
bannermsg += (
"\n repo: the repo object\n cl: repo.changelog\n mf: repo.manifestlog"
)
import IPython
IPython.embed(header=bannermsg)