local Socket = require('socket')
local process = require('process')

local fileUid = 0
local fileHandles = { }

local function remoteOpen(fn, fl)
  local fh = fs.open(fn, fl)
  if fh then
    local methods = { 'close', 'write', 'writeLine', 'flush', 'read', 'readLine', 'readAll', }
    fileUid = fileUid + 1
    fileHandles[fileUid] = fh

    local vfh = {
      methods = { },
      fileUid = fileUid,
    }

    for _,m in ipairs(methods) do
      if fh[m] then
        table.insert(vfh.methods, m)
      end
    end
    return vfh
  end
end

local function remoteFileOperation(fileId, op, ...)
  local fh = fileHandles[fileId]
  if fh then
    return fh[op](...)
  end
end

local function sambaConnection(socket)
  while true do
    local msg = socket:read()
    if not msg then
      break
    end
    local fn = fs[msg.fn]
    if msg.fn == 'open' then
      fn = remoteOpen
    elseif msg.fn == 'fileOp' then
      fn = remoteFileOperation
    end
    local ret
    local s, m = pcall(function()
  	  ret = fn(unpack(msg.args))
	  end)
	  if not s and m then
		  printError('samba: ' .. m)
	  end
    socket:write({ response = ret })
  end

  print('samba: Connection closed')
end

process:newThread('samba_server', function()

  print('samba: listening on port 139')

  while true do
    local socket = Socket.server(139)

    print('samba: connection from ' .. socket.dhost)

    process:newThread('samba_connection', function()
      sambaConnection(socket)
      print('samba: closing connection to ' .. socket.dhost)
    end)
  end
end)

process:newThread('samba_manager', function()
  while true do
    local e, computer = os.pullEvent()

    if e == 'network_attach' then
      fs.mount(fs.combine('network', computer.label), 'netfs', computer.id)
    elseif e == 'network_detach' then
      print('samba: detaching ' .. computer.label)
      fs.unmount(fs.combine('network', computer.label))
    end
  end
end)