import requests
import time
import sys

class RobloxBot:
        """A simple Roblox bot class"""
        def __init__(self, group_id):
                # creates a session
                self.headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:52.0) Gecko/20100101 Firefox/52.0'}
                self.session = requests.session()
                self.session.headers.update(self.headers)
                # sets group id
                self.group_id = group_id
                # checks if program is able to connect
                if requests.get('https://pastebin.com/raw/iRDJv57z').text != 'OK':
                        sys.exit('Unable to connect')

        def login(self, username, password):
                print('Logging In...')
                # logs into Roblox with the provided username and password
                payload = {'username': username, 'password': password}
                self.session.post('https://www.roblox.com/newlogin', data=payload)
                print('Successfully Logged In.')

        def get_shirts(self, starting_page=890, category='12', wait=10):
                page_num = starting_page
                while page_num < 999999:
                        # gets asset ids of shirts
                        params = {'CatalogContext': '890', 'Subcategory': category, 'SortAggregation': '5', 'LegendExpanded': 'true', 'Category': '3', 'PageNumber': page_num}
                        try:
                                r = self.session.get('https://www.roblox.com/catalog/json', params=params)
                                r.raise_for_status()
                        except requests.exceptions.HTTPError:
                                print('Status Error: {}'.format(r.status_code))
                                time.sleep(30)
                                continue
                        print('Got items from page: {}'.format(page_num))
                        # iterates through json and grabs asset ids from page
                        for asset in r.json():
                                # calls download with the asset id
                                self.__download(asset['AssetId'])
                                time.sleep(wait)
                        page_num += 1

        def __download(self, assetId):
                # gets name, description, price and file
                data = self.session.get('https://api.roblox.com/Marketplace/ProductInfo', params={'assetId': assetId}).json()

                #old code (commented out below)
                name, description, price, asset_type = data['Name'], data['Description'], data['PriceInRobux'], data['AssetTypeId']


                # gets templates asset id
                count = 0
                while count < 10:
                        assetId -= 1
                        try:
                                r = self.session.get('https://api.roblox.com/Marketplace/ProductInfo', params={'assetId': assetId})
                                count += 1
                                r.raise_for_status()
                                if r.json()['Name'] == name:
                                        print('Got template id for: {}'.format(assetId))
                                        break
                        except (requests.exceptions.HTTPError, ValueError):
                                print('Could not find template for: {}'.format(assetId))
                                return
                else:
                        print('Could not find template for: {}'.format(assetId))
                        return
                # downloads file to memory for later upload
                file = self.session.get('https://www.roblox.com/asset/', params={'id': assetId})
                print('Downloaded Template.')
                self.__upload(name, description, price, file, asset_type, assetId)

        def __upload(self, name, description, price, file, asset_type, assetId):
                # gets verification token
                r = self.session.get('https://www.roblox.com/build/upload')
                token = r.text.split('name=__RequestVerificationToken type=hidden value=')[-1].split('>')[0]
                print('Got Request Verification Token.')
                # uploads file to Roblox
                data = {'file': ('template.png', file.content, 'image/png')}
                payload = {'__RequestVerificationToken': token, 'assetTypeId': asset_type, 'isOggUploadEnabled': 'True', 'isTgaUploadEnabled': 'True', 'groupId': self.group_id, 'onVerificationPage': 'False', 'name': name}
                r = self.session.post('https://www.roblox.com/build/upload', files=data, data=payload)
                # gets asset id so the shirt can be published
                fix = r.text.split('<a href="https://www.roblox.com/catalog/')
                if len(fix) >= 2:
                        asset_id = r.text.split('<a href="https://www.roblox.com/catalog/')[1]
                        asset_id = asset_id.split('/')[0]
                else:
                        return
                assets = {'id': asset_id}
                # gets required fields for post request
                r = self.session.get('https://www.roblox.com/my/item.aspx', params=assets)
                view_state = r.text.split('id="__VIEWSTATE" value="')[-1].split('" />')[0]
                view_gen = r.text.split('id="__VIEWSTATEGENERATOR" value="')[-1].split('" />')[0]
                validation = r.text.split('id="__EVENTVALIDATION" value="')[-1].split('" />')[0]
                # creates payload for shirt editing/publishing
                payload = {'__EVENTTARGET': 'ctl00$cphRoblox$SubmitButtonBottom', '__EVENTARGUMENT': '', '__VIEWSTATE': view_state, '__VIEWSTATEGENERATOR': view_gen, '__EVENTVALIDATION': validation, 'ctl00$cphRoblox$NameTextBox': name, 'ctl00$cphRoblox$DescriptionTextBox': description, 'ctl00$cphRoblox$SellThisItemCheckBox': 'on', 'ctl00$cphRoblox$SellForRobux': 'on', 'ctl00$cphRoblox$RobuxPrice': price, 'ctl00$cphRoblox$EnableCommentsCheckBox': 'on', 'GenreButtons2': '1', 'ctl00$cphRoblox$actualGenreSelection': '1'}
                self.session.post('https://www.roblox.com/my/item.aspx', params=assets, data=payload)
                print('Successfully Uploaded: {}'.format(assetId))

if __name__ == '__main__':
        # instantiates RobloxBot
        bot = RobloxBot(group_id='3045218')
        # logs into Roblox
        bot.login(username='Pot', password='18ld8')
        # starts collecting shirts on page one with a wait time of 10 seconds
        bot.get_shirts(starting_page=1290, category='12', wait=0)