"""distutils.pypirc

Provides the PyPIRCCommand class, the base class for the command classes
that uses .pypirc in the distutils.command package.
"""
import os
from configparser import RawConfigParser

from distutils.cmd import Command

DEFAULT_PYPIRC = """\
[distutils]
index-servers =
 pypi

[pypi]
username:%s
password:%s
"""

class PyPIRCCommand(Command):
 """Base command that knows how to handle the .pypirc file
 """
 DEFAULT_REPOSITORY = 'https://upload.pypi.org/legacy/'
 DEFAULT_REALM = 'pypi'
 repository = None
 realm = None

 user_options = [
 ('repository=', 'r',
 "url of repository [default: %s]" % \
 DEFAULT_REPOSITORY),
 ('show-response', None,
 'display full response text from server')]

 boolean_options = ['show-response']

 def _get_rc_file(self):
 """Returns rc file path."""
 return os.path.join(os.path.expanduser('~'), '.pypirc')

 def _store_pypirc(self, username, password):
 """Creates a default .pypirc file."""
 rc = self._get_rc_file()
 with os.fdopen(os.open(rc, os.O_CREAT | os.O_WRONLY, 0o600), 'w') as f:
 f.write(DEFAULT_PYPIRC % (username, password))

 def _read_pypirc(self):
 """Reads the .pypirc file."""
 rc = self._get_rc_file()
 if os.path.exists(rc):
 self.announce('Using PyPI login from %s' % rc)
 repository = self.repository or self.DEFAULT_REPOSITORY

 config = RawConfigParser()
 config.read(rc)
 sections = config.sections()
 if 'distutils' in sections:
 # let's get the list of servers
 index_servers = config.get('distutils', 'index-servers')
 _servers = [server.strip() for server in
 index_servers.split('\n')
 if server.strip() != '']
 if _servers == []:
 # nothing set, let's try to get the default pypi
 if 'pypi' in sections:
 _servers = ['pypi']
 else:
 # the file is not properly defined, returning
 # an empty dict
 return {}
 for server in _servers:
 current = {'server': server}
 current['username'] = config.get(server, 'username')

 # optional params
 for key, default in (('repository',
 self.DEFAULT_REPOSITORY),
 ('realm', self.DEFAULT_REALM),
 ('password', None)):
 if config.has_option(server, key):
 current[key] = config.get(server, key)
 else:
 current[key] = default

 # work around people having "repository" for the "pypi"
 # section of their config set to the HTTP (rather than
 # HTTPS) URL
 if (server == 'pypi' and
 repository in (self.DEFAULT_REPOSITORY, 'pypi')):
 current['repository'] = self.DEFAULT_REPOSITORY
 return current

 if (current['server'] == repository or
 current['repository'] == repository):
 return current
 elif 'server-login' in sections:
 # old format
 server = 'server-login'
 if config.has_option(server, 'repository'):
 repository = config.get(server, 'repository')
 else:
 repository = self.DEFAULT_REPOSITORY
 return {'username': config.get(server, 'username'),
 'password': config.get(server, 'password'),
 'repository': repository,
 'server': server,
 'realm': self.DEFAULT_REALM}

 return {}

 def _read_pypi_response(self, response):
 """Read and decode a PyPI HTTP response."""
 import cgi
 content_type = response.getheader('content-type', 'text/plain')
 encoding = cgi.parse_header(content_type)[1].get('charset', 'ascii')
 return response.read().decode(encoding)

 def initialize_options(self):
 """Initialize options."""
 self.repository = None
 self.realm = None
 self.show_response = 0

 def finalize_options(self):
 """Finalizes options."""
 if self.repository is None:
 self.repository = self.DEFAULT_REPOSITORY
 if self.realm is None:
 self.realm = self.DEFAULT_REALM