root/ff-luci/trunk/modules/rpc/luasrc/controller/rpc.lua @ 3131

--[[
LuCI - Lua Configuration Interface

Copyright 2008 Steven Barth <steven@midlink.org>
Copyright 2008 Jo-Philipp Wich <xm@leipzig.freifunk.net>

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

$Id$
]]--

local require = require
local pairs = pairs
local print = print
local pcall = pcall
local table = table

module "luci.controller.rpc"

function index()
    local function authenticator(validator, accs)
        local auth = luci.http.formvalue("auth", true)
        if auth then
            local user = luci.sauth.read(auth)
            if user and luci.util.contains(accs, user) then
                return user, auth
            end
        end
        luci.http.status(403, "Forbidden")
    end
    
    local rpc = node("rpc")
    rpc.sysauth = "root"
    rpc.sysauth_authenticator = authenticator
    rpc.notemplate = true
    
    entry({"rpc", "uci"}, call("rpc_uci"))
    entry({"rpc", "uvl"}, call("rpc_uvl"))
    entry({"rpc", "fs"}, call("rpc_fs"))
    entry({"rpc", "sys"}, call("rpc_sys"))
    entry({"rpc", "ipkg"}, call("rpc_ipkg"))
    entry({"rpc", "auth"}, call("rpc_auth")).sysauth = false
end

function rpc_auth()
    local jsonrpc = require "luci.jsonrpc"
    local sauth   = require "luci.sauth"
    local http    = require "luci.http"
    local sys     = require "luci.sys"
    local ltn12   = require "luci.ltn12"
    
    local loginstat
    
    local server = {}
    server.login = function(user, pass)
        local sid
        
        if sys.user.checkpasswd(user, pass) then
            sid = sys.uniqueid(16)
            http.header("Set-Cookie", "sysauth=" .. sid.."; path=/")
            sauth.write(sid, user)
        end
        
        return sid
    end
    
    http.prepare_content("application/json")
    ltn12.pump.all(jsonrpc.handle(server, http.source()), http.write)
end

function rpc_uci()
    if not pcall(require, "luci.model.uci") then
        luci.http.status(404, "Not Found")
        return nil
    end
    local uci     = require "luci.jsonrpcbind.uci"
    local jsonrpc = require "luci.jsonrpc"
    local http    = require "luci.http"
    local ltn12   = require "luci.ltn12"
    
    http.prepare_content("application/json")
    ltn12.pump.all(jsonrpc.handle(uci, http.source()), http.write)
end

function rpc_uvl()
    if not pcall(require, "luci.uvl") then
        luci.http.status(404, "Not Found")
        return nil
    end
    local uvl     = require "luci.jsonrpcbind.uvl"
    local jsonrpc = require "luci.jsonrpc"
    local http    = require "luci.http"
    local ltn12   = require "luci.ltn12"

    http.prepare_content("application/json")
    ltn12.pump.all(jsonrpc.handle(uvl, http.source()), http.write)
end

function rpc_fs()
    local util    = require "luci.util"
    local io      = require "io"
    local fs2     = util.clone(require "luci.fs")
    local jsonrpc = require "luci.jsonrpc"
    local http    = require "luci.http"
    local ltn12   = require "luci.ltn12"

    function fs2.readfile(filename)
        local stat, mime = pcall(require, "mime")
        if not stat then
            error("Base64 support not available. Please install LuaSocket.")
        end

        local fp = io.open(filename)
        if not fp then
            return nil
        end

        local output = {}
        local sink = ltn12.sink.table(output)
        local source = ltn12.source.chain(ltn12.source.file(fp), mime.encode("base64"))
        return ltn12.pump.all(source, sink) and table.concat(output)
    end
    
    function fs2.writefile(filename, data)
        local stat, mime = pcall(require, "mime")
        if not stat then
            error("Base64 support not available. Please install LuaSocket.")
        end

        local  file = io.open(filename, "w")
        local  sink = file and ltn12.sink.chain(mime.decode("base64"), ltn12.sink.file(file))
        return sink and ltn12.pump.all(ltn12.source.string(data), sink) or false
    end
    
    http.prepare_content("application/json")
    ltn12.pump.all(jsonrpc.handle(fs2, http.source()), http.write)
end

function rpc_sys()
    local sys     = require "luci.sys"
    local jsonrpc = require "luci.jsonrpc"
    local http    = require "luci.http"
    local ltn12   = require "luci.ltn12"
    
    http.prepare_content("application/json")
    ltn12.pump.all(jsonrpc.handle(sys, http.source()), http.write)
end

function rpc_ipkg()
    if not pcall(require, "luci.model.ipkg") then
        luci.http.status(404, "Not Found")
        return nil
    end
    local ipkg    = require "luci.model.ipkg"
    local jsonrpc = require "luci.jsonrpc"
    local http    = require "luci.http"
    local ltn12   = require "luci.ltn12"

    http.prepare_content("application/json")
    ltn12.pump.all(jsonrpc.handle(ipkg, http.source()), http.write)
end