User:Darien Caldwell/Advanced HTTP-DNS Code

dns.py File

<python>

import logging, urllib

import webapp2 from google.appengine.ext import db

class Service(db.Model):

 name = db.StringProperty(multiline=False)
 url = db.StringProperty(multiline=False,indexed=False)
 writepass = db.StringProperty(multiline=False,indexed=False)
 readpass = db.StringProperty(multiline = False,indexed=False)
 hidden = db.BooleanProperty(indexed=False)

class Main_DNS(webapp2.RequestHandler):

   def get(self):

       obj_name=self.request.headers.get('X-SecondLife-Object-Name')
       region=self.request.headers.get('X-SecondLife-Region')
       local_pos=self.request.headers.get('X-SecondLife-Local-Position')
       owner_name=self.request.headers.get('X-SecondLife-Owner-Name')
       owner_key=self.request.headers.get('X-SecondLife-Owner-Key')
       logging.info('DNS called from '+obj_name+' owned by '+owner_name+' ('+owner_key+') from Region:'+region+' Pos:'+local_pos)
       logging.info('POST Body was: '+self.request.body)

       admin_password = "adminpassword"        # This password enables you to have unrestricted access to the DNS
       # Putting "null" in password disables this feature.

       if self.request.get('type')=='add':    # Adding a new service to the DNS (You can also use Update but it won't tell you the service already exists)
           param2=self.request.get('name')   # the Name the service will be known by
           param3=self.request.get('url')    # the URL for the web service
           param4=self.request.get('wpass')  # the password for modifying the entry
           param5=self.request.get('rpass')  # the password for reading the entry
           param6=self.request.get('hidden') # enables hiding the entry from listing

           if admin_password != 'null' and admin_password == self.request.get('admin'):

               newrec = Service.get_by_key_name(param2)

               if newrec is None:  # the service doesn't exist, so add it.
                   if param2=="" or param3=="" :
                       self.response.out.write('Error2')
                   else:
                       if param6 == "1":
                           param6 = True
                       else:
                           param6 = False

                       newrec=Service(key_name=param2,name=param2,url=param3,writepass=param4,readpass=param5,hidden=param6)
                       newrec.put()
                       logging.info('Added Service: '+param2)
                       self.response.out.write('Added')
               else:
                   logging.info('Service: '+param2+' already found.')
                   self.response.out.write('Found')  # service already exists so announce that and do nothing
           else:
               self.response.set_status(401)
               logging.info('Rejected Service: '+param2+' Invalid admin password')
               self.response.out.write('Rejected')

       elif self.request.get('type')=='remove': #removing a service
           param2=self.request.get('name')     # the name the service is known by

           record = Service.get_by_key_name(param2)

           if record is None:
               self.response.set_status(200)
               self.response.out.write('None') # Service wasn't found
           elif record.writepass == "" or record.writepass == self.request.get('pass') or (admin_password != 'null' and admin_password == self.request.get('admin')):
               record.delete()  # remove
               logging.info('Removed Service: '+param2)
               self.response.out.write('Removed')
           else:
               self.response.set_status(401)
               logging.info('Rejected Service: '+param2+' remove. Not Found')
               self.response.out.write('Rejected')

       elif self.request.get('type')=='update':  # update an existing service. Note this creates a new service, or updates an existing one
           param2=self.request.get('name') #the Name the service will be known by         
           param3=self.request.get('url') # the URL for the web service
           param4=self.request.get('pass') # the password for modifying the entry

           if param2=="" :
               logging.info('Could not Update Blank Service')
               self.response.out.write('Error2')
               return

           record = Service.get_by_key_name(param2)

           if record is None:
               if param3 == "":
                   logging.info('Updating Service: '+param2+' failed. Blank URL')
                   self.response.out.write('Error3')
               else:
                   newhidden = False
                   if self.request.get('hidden') == '1':
                       newhidden = True
                   newrec=Service(key_name=param2,name=param2,url=param3,writepass=param4,readpass=param4, hidden=newhidden)
                   if self.request.get('wpass') != "":
                       newrec.writepass=self.request.get('wpass')
                   if self.request.get('rpass') != "":
                       newrec.readpass=self.request.get('rpass')
                   newrec.put()
                   logging.info('Added Service: '+param2+' (update add)')
                   self.response.out.write('Added')
           elif record.writepass == "" or record.writepass == param4 or (admin_password != 'null' and admin_password == self.request.get('admin')):

               # record already exists, update it                    
               newwpass = record.writepass
               newrpass = record.readpass
               newhidden = record.hidden

               if param3 == "":
                   param3 = record.url            
               if self.request.get('wpass') != "":
                   if self.request.get('wpass') == "null":
                       newwpass = ""
                   else:
                       newwpass = self.request.get('wpass')
               if self.request.get('rpass') != "":
                   if self.request.get('rpass') == "null":
                       newrpass = ""
                   else:
                       newrpass = self.request.get('rpass')
               if self.request.get('hidden') != "":
                   if self.request.get('hidden') == "1":
                       newhidden = True
                   elif self.request.get('hidden') == "0":
                       newhidden = False

               # add record, either replacing the deleted one, or adding a new one if it never existed
               record.url=param3
               record.writepass=newwpass
               record.readpass=newrpass
               record.hidden=newhidden

               record.put()
               logging.info('Updated Service: '+param2)
               self.response.out.write('Updated')
           else:
               self.response.set_status(401)
               logging.info('Update Service: '+param2+' failed. Invalid Password.')
               self.response.out.write('Rejected')

       elif self.request.get('type')=='retrieve': # get the current URL for a given service
           param2=self.request.get('name')     # the name the service is known by

           record = Service.get_by_key_name(param2)

           if record is None:
               self.response.set_status(200)
               logging.info('Retrieve Service: '+param2+' failed. Not Found.')
               self.response.out.write('None') # Service wasn't found
           elif record.readpass == "" or record.readpass == self.request.get('pass') or (admin_password != 'null' and admin_password == self.request.get('admin')):
               logging.info('Retrieved Service: '+param2)
               self.response.out.write(record.url) #print the URL
           else:
               self.response.set_status(401)
               logging.info('Retrieve Service: '+param2+' failed. Invalid Password.')
               self.response.out.write("Rejected")

       elif self.request.get('type')=='list': # List the existing services

           records = Service.all()

           if records is None:
               logging.info('Service List: Empty')
               self.response.out.write('Empty') # Services weren't found
           else:
               got_admin = False
               if admin_password != 'null' and admin_password == self.request.get('admin'):
                   got_admin = True;            

               for result in records: 
                   if result.hidden == False or got_admin == True:
                       self.response.out.write(result.name+',')
               logging.info('Service List: Success')
               self.response.out.write('END')  # Cap the list             

       else:
           logging.info('Invalid Service command type: '+self.request.get('type'))
           self.response.out.write('Error')
   def post(self):
       self.get()

class Redirector(webapp2.RequestHandler):

 def get(self):

   import urllib

   service_name=self.request.path

   if service_name[-1]=='/' :
     service_name=service_name[1:-1] #remove leading and trailing slash
   else:
     service_name=service_name[1:]  # remove leading slash only

   #un-escape just in case you're Kinki :p
   service_name=urllib.unquote(service_name)
   logging.debug('Looking up '+service_name+' (DNS)');

   record = Service.get_by_key_name(str(service_name))

   if record is None:
       logging.info('Redirect Failed: Not Found')   
       self.response.out.write('None') # Service wasn't found
   elif record.readpass != :
       self.response.set_status(403)
       logging.info('Redirect Failed: Read Protected. Use "retrieve" command type to get URL.') 
       self.response.out.write("<html><title>403 Forbidden</title><body>Cannot forward a read-protected entry.</body></html>")
   else:
       logging.info('Redirect Success') 
       if self.request.query_string !=  :
           self.redirect(urllib.unquote(record.url)+'/?'+self.request.query_string) # redirect to the HTTP-IN URL with arugments
       else:        
           self.redirect(urllib.unquote(record.url)) # redirect to the HTTP-IN URL    

app = webapp2.WSGIApplication( [('/', Main_DNS),

                               ('/.*',Redirector)], 
                                debug=True)

</python>

app.yaml File

<text> application: YOUR-APP-NAME-HERE version: 1 runtime: python27 api_version: 1 threadsafe: true

handlers: - url: /.*

 script: dns.app

</text>