root/SMTP/SMTPSession.lua

--------------------------------------------------------------------------------
-- Title:               SMTPSession.lua
-- Description:         Like a square peg in a round hole
-- Author:              Raphaël Szwarc http://alt.textdrive.com/lua/
-- Creation Date:       February 1, 2005
-- Legal:               Copyright (C) 2005 Raphaël Szwarc
--------------------------------------------------------------------------------

-- import dependencies
local LUObject = require( "LUObject" )
local LUList = require( "LUList" )
local LUString = require( "LUString" )
local LULog = require( "LULog" )
local SMTPCommand = require( "SMTPCommand" )
local socket = require( "socket" )

-- define the class
local super = LUObject
local self = super()

-- initialization method
function self:init( aServer, aReader, aWriter )
        self = super.init( self )

        self._server = aServer
        self._reader = aReader
        self._writer = aWriter
        
        return self
end

function self:server()
        return self._server
end

function self:reader()
        return self._reader
end

function self:writer()
        return self._writer
end

function self:delegate()
        if self._delegate == nil then
                self._delegate = self:server():delegate()
        end
        
        return self._delegate
end

function self:setDelegate( aValue )
        self._delegate = aValue
        
        return self
end

function self:domain()
        return self._domain
end

function self:setDomain( aValue )
        self._domain = aValue
        
        return self
end

function self:user()
        return self._user
end

function self:setUser( aValue )
        self._user = aValue
        
        self:setState( 1 )
        
        return self
end

function self:state()
        if self._state == nil then
                self._state = 0
        end

        return self._state
end

function self:setState( aValue )
        self._state = 0
        
        return self
end

function self:sender()
        return self._sender
end

function self:setSender( aValue )
        self._sender = aValue
        
        return self
end

function self:recipients()
        if self._recipients == nil then
                self._recipients = LUList()
        end
        
        return self._recipients
end

function self:reset()
        self:setSender( nil )
        self:recipients():clear()
        
        return self
end

function self:host()
        if self._host == nil then
                self._host = socket.dns.gethostname()
        end
        
        return self._host
end

function self:remoteHost()
        if self._remoteHost == nil then
                self._remoteHost = self:reader():socket():getpeername()
        end
        
        return self._remoteHost
end

local function catch( anException, anObject, aMethod, aSession, someTokens )
        LULog:warning( anObject, anException )

        aSession:writer():writeln( "500 Syntax error, command unrecognized" )
        
        return true
end

-- method to run this session
function self:run()
        local shouldContinue = true
        
        self:writer():writeln( "220 ESMTP Service ready" )
        
        while shouldContinue == true do
                local aLine = self:reader():read()
                
                if aLine ~= nil then
                        local someTokens = LUString:components( aLine, " " )
                        
                        LULog:debug( aLine )
                        
                        if someTokens:hasData() == true then
                                local aCommandName = someTokens:get()
                                local aCommand = SMTPCommand:commandWithName( aCommandName )
                                
                                someTokens:remove()
                                
                                shouldContinue = aCommand:try( "run", self, someTokens )( catch )
                        else
                                shouldContinue = false
                        end
                else
                        shouldContinue = false
                end
        end

        return self
end

return self