#!/usr/bin/env python

########################################################################

# This is transmission-twitter, program that removes and notifies      #

# about completed torrents in transmission-daemon                      #

#                                                                      #

# This program is free software: you can redistribute it and/or modify #

# it under the terms of the GNU General Public License as published by #

# the Free Software Foundation, either version 3 of the License, or    #

# (at your option) any later version.                                  #

#                                                                      #

# This program is distributed in the hope that it will be useful,      #

# but WITHOUT ANY WARRANTY; without even the implied warranty of       #

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        #

# GNU General Public License for more details:                         #

# http://www.gnu.org/licenses/gpl-3.0.txt                              #

########################################################################

import ConfigParser

import unittest

import os

class ConfigFile:

    def get_config(self):

        """ Returns the current configuration in a dictionary 

        The default config file is in .config/transmission-daemon/plugins/ directory

        This method reads the config file and returns the config information as a 

        dictionary. If configfile is not present or invalid default configuration is returned.

        """

        config_parser = ConfigParser.ConfigParser()

        configfile = os.getenv('HOME')+'/.config/transmission-daemon/plugins/transmission-scripts.conf'

        config_parser.read(

            ['transmission-scripts.conf',configfile]

            )

        config = dict()

        try:

            config["AUTOSTOP_FINISHED_TORRENTS"] = config_parser.getboolean(

                'Transmission',"AUTOSTOP_FINISHED_TORRENTS")

        except (ValueError,ConfigParser.NoSectionError,

                ConfigParser.NoOptionError):

            config["AUTOSTOP_FINISHED_TORRENTS"] = True

        try:

            config["TwitterIntegration"] = config_parser.getboolean(

                'TwitterIntegration','ENABLED')

        except(ValueError,ConfigParser.NoSectionError):

            config["TwitterIntegration"] = True

        if config["TwitterIntegration"]:

            try:

                config["CONSUMER_KEY"] = config_parser.get(

                    'TwitterIntegration',

                    'CONSUMER_KEY')

                config["CONSUMER_SECRET"] = config_parser.get(

                    'TwitterIntegration',

                    'CONSUMER_SECRET'

                    )                

            except(ValueError,ConfigParser.NoSectionError,

                   ConfigParser.NoOptionError):

                config["CONSUMER_KEY"] = "PbOGMLYLlsPtISZilEVeXw"

                config["CONSUMER_SECRET"] = "tNyHTMW0BfPW6Us1XjrLq1aU0ach42k8c8L7v6g2Ok"

            try:    

                config["ACCESS_TOKEN"] = config_parser.get(

                    'TwitterIntegration',

                    'ACCESS_TOKEN')

                config["ACCESS_TOKEN_SECRET"] = config_parser.get(

                    'TwitterIntegration',

                    'ACCESS_TOKEN_SECRET')

            except(ValueError,ConfigParser.NoSectionError,

                   ConfigParser.NoOptionError):

                #User has to authenticate this app to get the below keys

                config["ACCESS_TOKEN"] = ""

                config["ACCESS_TOKEN_SECRET"] = ""

            try:

                config["REQUEST_TOKEN_URL"] = config_parser.get(

                    'TwitterIntegration',

                    'REQUEST_TOKEN_URL')

                config["ACCESS_TOKEN_URL"] = config_parser.get(

                    'TwitterIntegration',

                    'ACCESS_TOKEN_URL')

                config["AUTHORIZE_URL"] = config_parser.get(

                    'TwitterIntegration',

                    'AUTHORIZE_URL')

                config["SCREEN_NAME"] = config_parser.get(

                    'TwitterIntegration',

                    'SCREEN_NAME')

            except(ValueError,ConfigParser.NoSectionError,

                   ConfigParser.NoOptionError):

                config["REQUEST_TOKEN_URL"] = 'https://api.twitter.com/oauth/request_token'

                config["ACCESS_TOKEN_URL"] = 'https://api.twitter.com/oauth/access_token'

                config["AUTHORIZE_URL"] = 'https://api.twitter.com/oauth/authorize'

                config["SCREEN_NAME"] = ""

        return config

    def write_config(self,section,key,value):

        """ writes key and value in specified section of configfile 

        It writes to the default config file /.config/transmission-daemon/plugins/transmission-scripts.conf

        Generally used to write the oauth tokens to the config file

        """

        config_parser = ConfigParser.ConfigParser()

        configfile = os.getenv('HOME')+'/.config/transmission-daemon/plugins/transmission-scripts.conf'

        config_parser.read(['transmission-scripts.conf',configfile])

        try:

            config_parser.set(section,key,value)

        except(ConfigParser.NoSectionError,ConfigParser.NoOptionError):

            #seems like the config file is not there

            return False

        try:

            with open(configfile,'w') as file:

                config_parser.write(file)

        except IOError:

            return False

        return True

class Testing(unittest.TestCase):

    """ Bunch of unit test cases """

    def test_get_config(self):

        """ Tests the get_config method """

        cf = ConfigFile()

        self.assertEquals({"AUTOSTOP_FINISHED_TORRENTS": True,

                           "TwitterIntegration": False},cf.get_config())

       # self.assertRaises(ConfigParser.NoSectionError,get_config,'transmission-scripts.conf')

    def test_add_config(self):

        cf = ConfigFile()

        self.assertEquals(1,cf.write_config('TwitterIntegration',

                                            'ACCESS_TOKEN','aebcd'))

if __name__ == '__main__':

    #unittest.main()

    cf = ConfigFile()

    config = cf.get_config()

    print config