# windows.py - Windows utility function implementations for Mercurial

#

#  Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

import errno

import msvcrt

import os

import re

import stat

import sys

from .i18n import _

from . import (

    encoding,

    error,

    policy,

    pycompat,

    win32,

)

try:

    import _winreg as winreg

    winreg.CloseKey

except ImportError:

    import winreg

osutil = policy.importmod(r'osutil')

getfsmountpoint = win32.getvolumename

getfstype = win32.getfstype

getuser = win32.getuser

hidewindow = win32.hidewindow

makedir = win32.makedir

nlinks = win32.nlinks

oslink = win32.oslink

samedevice = win32.samedevice

samefile = win32.samefile

setsignalhandler = win32.setsignalhandler

spawndetached = win32.spawndetached

split = os.path.split

testpid = win32.testpid

unlink = win32.unlink

umask = 0o022

class mixedfilemodewrapper(object):

    """Wraps a file handle when it is opened in read/write mode.

    fopen() and fdopen() on Windows have a specific-to-Windows requirement

    that files opened with mode r+, w+, or a+ make a call to a file positioning

    function when switching between reads and writes. Without this extra call,

    Python will raise a not very intuitive "IOError: [Errno 0] Error."

    This class wraps posixfile instances when the file is opened in read/write

    mode and automatically adds checks or inserts appropriate file positioning

    calls when necessary.

    """

    OPNONE = 0

    OPREAD = 1

    OPWRITE = 2

    def __init__(self, fp):

        object.__setattr__(self, r'_fp', fp)

        object.__setattr__(self, r'_lastop', 0)

    def __enter__(self):

        return self._fp.__enter__()

    def __exit__(self, exc_type, exc_val, exc_tb):

        self._fp.__exit__(exc_type, exc_val, exc_tb)

    def __getattr__(self, name):

        return getattr(self._fp, name)

    def __setattr__(self, name, value):

        return self._fp.__setattr__(name, value)

    def _noopseek(self):

        self._fp.seek(0, os.SEEK_CUR)

    def seek(self, *args, **kwargs):

        object.__setattr__(self, r'_lastop', self.OPNONE)

        return self._fp.seek(*args, **kwargs)

    def write(self, d):

        if self._lastop == self.OPREAD:

            self._noopseek()

        object.__setattr__(self, r'_lastop', self.OPWRITE)

        return self._fp.write(d)

    def writelines(self, *args, **kwargs):

        if self._lastop == self.OPREAD:

            self._noopeseek()

        object.__setattr__(self, r'_lastop', self.OPWRITE)

        return self._fp.writelines(*args, **kwargs)

    def read(self, *args, **kwargs):

        if self._lastop == self.OPWRITE:

            self._noopseek()

        object.__setattr__(self, r'_lastop', self.OPREAD)

        return self._fp.read(*args, **kwargs)

    def readline(self, *args, **kwargs):

        if self._lastop == self.OPWRITE:

            self._noopseek()

        object.__setattr__(self, r'_lastop', self.OPREAD)

        return self._fp.readline(*args, **kwargs)

    def readlines(self, *args, **kwargs):

        if self._lastop == self.OPWRITE:

            self._noopseek()

        object.__setattr__(self, r'_lastop', self.OPREAD)

        return self._fp.readlines(*args, **kwargs)

def posixfile(name, mode='r', buffering=-1):

    '''Open a file with even more POSIX-like semantics'''

    try:

        fp = osutil.posixfile(name, mode, buffering) # may raise WindowsError

        # The position when opening in append mode is implementation defined, so

        # make it consistent with other platforms, which position at EOF.

        if 'a' in mode:

            fp.seek(0, os.SEEK_END)

        if '+' in mode:

            return mixedfilemodewrapper(fp)

        return fp

    except WindowsError as err:

        # convert to a friendlier exception

        raise IOError(err.errno, '%s: %s' % (

            name, encoding.strtolocal(err.strerror)))

# may be wrapped by win32mbcs extension

listdir = osutil.listdir

class winstdout(object):

    '''stdout on windows misbehaves if sent through a pipe'''

    def __init__(self, fp):

        self.fp = fp

    def __getattr__(self, key):

        return getattr(self.fp, key)

    def close(self):

        try:

            self.fp.close()

        except IOError:

            pass

    def write(self, s):

        try:

            # This is workaround for "Not enough space" error on

            # writing large size of data to console.

            limit = 16000

            l = len(s)

            start = 0

            self.softspace = 0

            while start < l:

                end = start + limit

                self.fp.write(s[start:end])

                start = end

        except IOError as inst:

            if inst.errno != 0:

                raise

            self.close()

            raise IOError(errno.EPIPE, 'Broken pipe')

    def flush(self):

        try:

            return self.fp.flush()

        except IOError as inst:

            if inst.errno != errno.EINVAL:

                raise

            raise IOError(errno.EPIPE, 'Broken pipe')

def _is_win_9x():

    '''return true if run on windows 95, 98 or me.'''

    try:

        return sys.getwindowsversion()[3] == 1

    except AttributeError:

        return 'command' in encoding.environ.get('comspec', '')

def openhardlinks():

    return not _is_win_9x()

def parsepatchoutput(output_line):

    """parses the output produced by patch and returns the filename"""

    pf = output_line[14:]

    if pf[0] == '`':

        pf = pf[1:-1] # Remove the quotes

    return pf

def sshargs(sshcmd, host, user, port):

    '''Build argument list for ssh or Plink'''

    pflag = 'plink' in sshcmd.lower() and '-P' or '-p'

    args = user and ("%s@%s" % (user, host)) or host

    if args.startswith('-') or args.startswith('/'):

        raise error.Abort(

            _('illegal ssh hostname or username starting with - or /: %s') %

            args)

    args = shellquote(args)

    if port:

        args = '%s %s %s' % (pflag, shellquote(port), args)

    return args

def setflags(f, l, x):

    pass

def copymode(src, dst, mode=None):

    pass

def checkexec(path):

    return False

def checklink(path):

    return False

def setbinary(fd):

    # When run without console, pipes may expose invalid

    # fileno(), usually set to -1.

    fno = getattr(fd, 'fileno', None)

    if fno is not None and fno() >= 0:

        msvcrt.setmode(fno(), os.O_BINARY)

def pconvert(path):

    return path.replace(pycompat.ossep, '/')

def localpath(path):

    return path.replace('/', '\\')

def normpath(path):

    return pconvert(os.path.normpath(path))

def normcase(path):

    return encoding.upper(path) # NTFS compares via upper()

# see posix.py for definitions

normcasespec = encoding.normcasespecs.upper

normcasefallback = encoding.upperfallback

def samestat(s1, s2):

    return False

# A sequence of backslashes is special iff it precedes a double quote:

# - if there's an even number of backslashes, the double quote is not

#   quoted (i.e. it ends the quoted region)

# - if there's an odd number of backslashes, the double quote is quoted

# - in both cases, every pair of backslashes is unquoted into a single

#   backslash

# (See http://msdn2.microsoft.com/en-us/library/a1y7w461.aspx )

# So, to quote a string, we must surround it in double quotes, double

# the number of backslashes that precede double quotes and add another

# backslash before every double quote (being careful with the double

# quote we've appended to the end)

_quotere = None

_needsshellquote = None

def shellquote(s):

    r"""

    >>> shellquote(br'C:\Users\xyz')

    '"C:\\Users\\xyz"'

    >>> shellquote(br'C:\Users\xyz/mixed')

    '"C:\\Users\\xyz/mixed"'

    >>> # Would be safe not to quote too, since it is all double backslashes

    >>> shellquote(br'C:\\Users\\xyz')

    '"C:\\\\Users\\\\xyz"'

    >>> # But this must be quoted

    >>> shellquote(br'C:\\Users\\xyz/abc')

    '"C:\\\\Users\\\\xyz/abc"'

    """

    global _quotere

    if _quotere is None:

        _quotere = re.compile(r'(\\*)("|\\$)')

    global _needsshellquote

    if _needsshellquote is None:

        # ":" is also treated as "safe character", because it is used as a part