#!/usr/bin/env python

# Copyright (C) 2010 dash@uberwall.org
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
# 3. All advertising materials mentioning features or use of this software
#    must display the following acknowledgement:
#      This product includes software developed by dash.
# 4. The name dash may not be used to endorse or promote
#    products derived from this software without specific prior written
#    permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY
# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.


import random
import curses
import time
import sys
import os
 
class cmate(object):
    def __init__(self):
	self.name="the hunt for 0days"
	self.lvl=0
	self.gold=0
	self.gcount=0
	self.collected=0
	self.eaten=0
	self.ms=0
	self.elive=0
	self.lives=3
	self.time=60
	self.gmap=[]
	self.mmap=[]
	self.gomap=[]
	self.win1=0
	self.win2=0
	self.sizex=25
	self.sizey=80
	self.winlist=[]
	self.stdscr=0
	self.msspeed=100

    def close_stdscr(self):
	""" at the end - we reset everything """
	curses.nocbreak()
	curses.echo()
	self.stdscr.keypad(0)
	curses.endwin()

    def create_stdscr(self):
	""" function for the stdscr """
	self.stdscr = curses.initscr()
	self.stdscr.border()
	self.stdscr.keypad(1)

	curses.cbreak()
	curses.noecho()
	curses.start_color()
	curses.curs_set(1)

    def create_colors(self):
	curses.init_pair(1,curses.COLOR_GREEN,curses.COLOR_BLACK)
	curses.init_pair(2,curses.COLOR_YELLOW,curses.COLOR_BLACK)
	curses.init_pair(3,curses.COLOR_RED,curses.COLOR_BLACK)
	curses.init_pair(4,curses.COLOR_WHITE,curses.COLOR_BLACK)
	curses.init_pair(5,curses.COLOR_WHITE,curses.COLOR_RED)
	curses.init_pair(6,curses.COLOR_YELLOW,curses.COLOR_BLUE)
	curses.init_pair(7,curses.COLOR_WHITE,curses.COLOR_BLUE)
	curses.init_pair(8,curses.COLOR_WHITE,curses.COLOR_WHITE)
	curses.init_pair(9,curses.COLOR_YELLOW,curses.COLOR_YELLOW)
	curses.init_pair(10,curses.COLOR_BLACK,curses.COLOR_BLACK)
	
    def check_windows(self):
	""" how many windows are in the list """
	wlen = len(self.winlist)
	return wlen

    def create_window(self,sizex,sizey,posx,posy):
	""" function for creating a new window """
	win = curses.newwin(sizex,sizey,posx,posy)
	self.winlist.append(win)
	return win

    def compute_strpos(self,win,pos,string):
	""" find relative positions for texts in the middle of the window"""
	sizex,sizey = win.getmaxyx()
	if pos == 1:
	    l = len(string)
	    spos = sizey - l
	    rpos = spos /2
	
	return rpos

    def write_window(self,win,posx,posy,text,attr):
	""" wrapper function for writing strings to a window """
	if posx!=-1 and posy!=-1 and attr !=-1:
	    win.addstr(posx,posy,text,attr)
	elif posx!=-1 and posy!=-1 and attr == -1:
	    win.addstr(posx,posy,text)
	elif posx==-1 and posy==-1 and attr == -1:
	    win.addstr(text)
	elif posx==-1 and posy==-1 and attr !=-1:
	    win.addstr(text,attr)
	
	win.refresh()

    def config_window(self,win,func):
	if func==1:
	    win.box()
	    win.refresh()
	elif func == 2:
	    win.erase()
	    win.refresh()
    
    def generate_map(self,x,y):
	fill = 35 # 35 is the fill element '#'
	gmap = []
	fields = x*y
	i = 0
	for cx in range(0,x):
	    for cy in range(0,y):
		gmap.append([cx,cy,32])
	    i+=1
	i+=1
	self.gmap = gmap
	
    def generate_monster(self,f,t):

	#gmap
	gmap = self.gmap
	#monsters
	ms = [f,t]
	ms = random.randint(ms[0],ms[1])
	l = len(gmap)-1
	mmap=[]

	for g in range(0,ms):
	    while 1:
		p = random.randint(0,l)
		if (gmap[p][0] != 0 and gmap[p][1] != 0) and (gmap[p][0] != 23 and gmap[p][1] != 79):
		    gmap[p][2] = 164
		    mmap.append(gmap[p])
		    break
	
	self.gmap = gmap
	self.mmap = mmap
	self.ms = ms

    def generate_gold(self,f,t):

	gmap = self.gmap

        #gold aka 0days
	gold = [f,t]
	gold = random.randint(gold[0],gold[1])
	l = len(gmap)-1

	for g in range(0,gold):
	    while 1:
		p = random.randint(0,l)
		if (gmap[p][0] != 0 and gmap[p][1] != 0) and (gmap[p][0] != 23 and gmap[p][1] != 79):
		    gmap[p][2] = 163
		    self.gomap.append(gmap[p])
		    break

	self.gold = gold
	self.gmap = gmap

    def generate_item(self,f,t,item):

	gmap = self.gmap
	#chill number 35
	f = [f,t]
	f = random.randint(f[0],f[1])
	l = len(gmap)-1
	for g in range(0,f):
	    while 1:
		p = random.randint(0,l)
		if (gmap[p][0] != 0 and gmap[p][1] != 0) and (gmap[p][0] != 23 and gmap[p][1] != 79) and gmap[p][2] != 163:
		    gmap[p][2] = item
		    break
	
	self.gmap = gmap

    def throw_items(self,win,rc):
	""" populate the map with the window with pre-generated items in gmap """
	gmap = self.gmap

	#general map
	for val in gmap:
	    if (val[0] != 0 and val[1] != 0) and (val[0] != 23 and val[1] != 79):
		if val[2] == 163:
		    win.addstr(val[0],val[1],chr(val[2]),curses.A_REVERSE)
		    win.refresh()
		elif val[2] == 164:
		    win.addstr(val[0],val[1],chr(val[2]),curses.color_pair(5))
		    win.refresh()
		else:
		    win.addstr(val[0],val[1],chr(val[2]),curses.color_pair(rc))
		    win.refresh()
	win.refresh()


	self.gmap = gmap

    def update_lifebar(self,win):
	""" show the player whats going on """

	#save my coordinates
	me = win.getyx()

	#which level
	collect = "lvl:%s" % (self.lvl)
	win.addstr(0,3,collect)
	
	#how much zerodays to grep
	gold = "%s / %s" % (self.gcount,self.gold)
	win.addstr(0,12,gold)

	#zeroday collection
	collect = "xploitz: %s" % (self.collected)
	win.addstr(0,20,collect)

	#lives
	lives = "live: %d" % (self.lives)
	win.addstr(0,35,lives)

	#knights name
	name = "%s" % (self.name)
	win.addstr(0,50,name)

	#time
	#time = "%s" % (self.time)
	#win.addstr(0,70,time)


	#how many fedz and whittezzzz
	fedz = "%s fedz" % (str(self.ms))
	win.addstr(23,3, fedz)

	#published by wh and eaten by fedz
	ate = "[%s]" % (str(self.eaten))
	win.addstr(23,20, ate)

	win.move(me[0],me[1])
	win.refresh()
	
    def check_object(self,win):
	""" check if a monster ate our zeroday """
#	f = open('logme.txt','a')
	mmap = self.mmap
	gomap = self.gomap
	for gold in gomap:
	    for monster in mmap:
		if gold[0] == monster[0] and gold[1] == monster[1]:
		    self.eaten=int(self.eaten) + 1
		    gomap.remove(gold)
		    #g = "%s\n" % (gold)
		    #f.write(g)
		    #m = "%s\n" % (monster)
		    #f.write(m)
		    self.update_lifebar(win)
	#f.close()
	self.gomap = gomap

    def check_position(self,win):
	""" checks position for gold, this function should be optimized as
	    by now the whole gmap is searched, which could be done simpler """

	gmap = self.gmap
	i=0
	me = win.getyx()
	for val in gmap:
	    if me[0] == val[0] and me[1] == val[1] and val[2] == 163:

		self.gcount = int(self.gcount) + 1
		self.elive = int(self.elive) + 1
		self.collected = int(self.collected) + 1
		self.update_lifebar(win)

		#changing the field - clearing space
		win.addch(me[0],me[1]," ")

		#clearing space in table
		gmap[i][2]=32

		#ok i am back at the same old place
		win.move(me[0],me[1])
		win.refresh()
		self.gmap = gmap
	    i+=1
    
    def check_collision(self,win):
	""" lets check if a monster is byting our ass """

	me = win.getyx()
	mmap = self.mmap

	for val in mmap:
	    if val[0] == me[0] and val[1] == me[1]: 
		self.lives = self.lives - 1
		return 1

	return -1

    def move_monster(self,win):
	gmap = self.gmap
	mmap = self.mmap
	i=0
	me = win.getyx()
	for val in mmap:
    
	    m = random.randint(1,4)

#check what direction the random gen spitted
#check if monster has reached the cage

	    if m == 1: 
		if val[0]!=1:
		    win.addstr(val[0],val[1]," ")
		    x = val[0]
		    x = x - 1
		    mmap[i][0] = x
		    win.addstr(x,val[1],"O",curses.color_pair(5))
		    win.refresh()

	    elif m == 2: 
		if val[0]!=self.sizex-2:
		    win.addstr(val[0],val[1]," ")
		    x = val[0]	
		    x = x + 1
		    mmap[i][0] = x
		    win.addstr(x,val[1],"O",curses.color_pair(5))
		    win.refresh()

	    elif m == 3: 
		if val[1]!=1:
		    win.addstr(val[0],val[1]," ")
		    y = val[1]	
		    y = y - 1
		    mmap[i][1] = y
		    win.addstr(val[0],y,"O",curses.color_pair(5))
		    win.refresh()

	    elif m == 4: 
		if val[1]!=self.sizey-2:
		    win.addstr(val[0],val[1]," ")
		    y = val[1]	
		    y = y + 1
		    mmap[i][1] = y 
		    win.addstr(val[0],y,"O",curses.color_pair(5))
		    win.refresh()
#former monster field is empty now
	    i+=1

	win.move(me[0],me[1])
	win.refresh()
	self.mmap = mmap

    def move_cursor(self,win, x,y):
	""" for other useful and great animations """
	win.move(x,y)

    def delete_item(self, win, item, x,y):
	""" for deleting the littel things at the position xy """
	win.addch(x,y,item)

    def move_item(self, win, item, x,y,color):
	""" for moving "sprites" :) """
	win.move(x,y)
	self.write_window(win,x,y,item,color)
#	win.refresh()

    def move_player(self,action,win):

	a = action
	xy = win.getyx()
	x=xy[0]
	y=xy[1]

	win.addch(x,y," ")
	if a == 65:
	    if x!=1:
		win.move(x-1,y)
		win.refresh()
	    else:
		win.move(x,y)
		win.refresh()

	elif a == 66:
	    if x!=self.sizex-2:
		win.move(x+1,y)
		win.refresh()
	    else:
		win.move(x,y)
		win.refresh()

	elif a == 68:
	    if y!=1:
		win.move(x,y-1)
		win.refresh()
	    else:
		win.move(x,y)
		win.refresh()

	elif a == 67:
	    if y!=self.sizey-2:
		win.move(x,y+1)
		win.refresh()
	    else:
		win.move(x,y)
		win.refresh()