import requests
import time
import sys
import os
os.system("color 1e")

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=66, category='14', wait=10):
		page_num = starting_page
		while page_num < 999999:
			# gets asset ids of shirts
			params = {'CatalogContext': 66, '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
				while True:
					try:
						self.__download(asset['AssetId'])
						break
					except:
						print('Found an error. Retrying.')
						continue
				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()
		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
		asset_id = r.text.split('uploadedAssetId=')[-1].split('" />')[0]
		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='739254')
	# logs into Roblox
	bot.login(username='FUCKFUCKFUCK', password='FUCKFUCKFUCK')
	# starts collecting shirts on page one with a wait time of 10 seconds
	bot.get_shirts(starting_page=200, category='14', wait=0.5)