the ben tovim

# HTTP Server Shell
# Author: Barak Gonen
# Purpose: Provide a basis for Ex. 4.4
# Note: The code is written in a simple way, without classes, log files or
# other utilities, for educational purpose
# Usage: Fill the missing functions and constants

# TO DO: import modules

import os
# TO DO: set constants
import socket
from datetime import datetime
from time import mktime
from wsgiref.handlers import format_date_time

IP = '0.0.0.0'
PORT = 80
SOCKET_TIMEOUT = 5000  # 5000 ms
DEFAULT_URL = '/index.html'
REDIRECTION_DICTIONARY = []


def makeHTTPResponse(status='200 OK', server='bentovimerrrr server (noder)',
                     contentType='text/plain', content=''):
    # build a http requests
    return 'HTTP/1.1 %s\r\n' \
           'Date: %s\r\n' \
           'Server: %s\r\n' \
           'Content-Type: %s\r\n' \
           'Content-Length: %d\r\n' \
           '\r\n' \
           '%s' % (
               status, format_date_time(mktime(datetime.now().timetuple())),
               server, contentType, len(content), content)


def handeHTTPRequest(request):
    # break down http request into a dictionary
    request = request.split('\r\n')
    dic = {}
    title = request[0].split(' ')
    dic['request-type'] = title[0]
    dic['route'] = title[1]
    dic['http-version'] = title[2]
    for i in range(2, len(request) - 2):
        header = request[i].split(':', 1)
        dic[header[0]] = header[1]
    dic['body'] = request[len(request) - 1]
    return dic


def get_file_data(filename):
    # get file content
    file = open(filename, 'rb')
    content = file.read()
    file.close()
    # get file type
    fileType = filename.split('.')
    fileType = fileType[len(fileType) - 1]
    return content, fileType


def handle_client_request(request, client_socket):
    """ Check the required resource, generate proper HTTP response and send to
    client"""
    # TO DO : add code that given a resource (URL and parameters) generates the
    # proper response

    # get route
    resource = request['route']

    # set url
    if resource == '' or resource == '/':
        url = DEFAULT_URL
    else:
        url = resource

    # split to url, params
    try:
        url = url.split('?')
        try:
            url[1] = url[1].split('&')
        except:
            url[1] = [url[1]]
    except:
        pass

    params = {}
    if len(url) == 2:
        for param in url[1]:
            param = param.split('=')
            params[param[0]] = param[1]

    url = url[0]

    # convert route to absolute directory
    url = './webroot' + url
    if request['request-type'] != 'GET':
        client_socket.send(
            makeHTTPResponse(
                content='<h1>ERROR: 405 Method Not Allowed</h1> <br> try a' +
                        'different route or different request route',
                status=405,
                contentType='text/html'))
    elif os.path.isfile(url):
        fileContent, fileType = get_file_data(url)
        if fileType == 'html':
            client_socket.send(
                makeHTTPResponse(content=fileContent, contentType='text/html'))
        elif fileType == 'js':
            client_socket.send(
                makeHTTPResponse(
                    content=fileContent, contentType='text/javascript'))
        elif fileType == 'css':
            client_socket.send(
                makeHTTPResponse(content=fileContent, contentType='text/css'))
        elif fileType == 'jpg':
            client_socket.send(
                makeHTTPResponse(
                    content=fileContent, contentType='image/jpeg'))
        elif fileType == 'ico':
            client_socket.send(
                makeHTTPResponse(
                    content=fileContent, contentType='image/x-icon'))
        else:
            client_socket.send(
                makeHTTPResponse(
                    content=fileContent, contentType='text/plain'))

    client_socket.send(
        makeHTTPResponse(content='<h1>ERROR: 404 not found</h1> <br> try' +
                                 'a different route', status=404,
                         contentType='text/html'))


def validate_http_request(request):
    """ Check if request is a valid HTTP request and returns TRUE / FALSE and
    the requested URL """
    # TO DO: write function
    try:
        request = handeHTTPRequest(request)
        return True, request
    except:
        return False, ''


def handle_client(client_socket):
    """ Handles client requests: verifies client's requests are legal HTTP,
    calls function to handle the requests """
    print 'Client connected'
    while True:
        client_request = client_socket.recv(500000)
        valid_http, request = validate_http_request(client_request)
        if valid_http:
            print 'Got a valid HTTP request'
            handle_client_request(request, client_socket)
        else:
            print 'Error: Not a valid HTTP request'
            break
    print 'Closing connection'
    client_socket.close()


def main():
    # Open a socket and loop forever while waiting for clients
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind((IP, PORT))
    server_socket.listen(10)
    print "Listening for connections on port %d" % PORT

    while True:
        client_socket, client_address = server_socket.accept()
        print 'New connection received'
        client_socket.settimeout(SOCKET_TIMEOUT)
        handle_client(client_socket)


if __name__ == "__main__":
    main()