#! /usr/bin/lua

--[[ Copyright 2010 Myhailo Danylenko

This program is free software: you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation, either version 2 of the License, or

(at your option) any later version.

This program is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

GNU General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program.  If not, see <http://www.gnu.org/licenses/>. ]]

--

-- Simple mime type detector

--

local F = { }

local options = {

	database = '/etc/mime.types',

	default  = 'application/octet-stream',

}

local db = { }

local h = io.open ( options['database'], 'r' )

if not h then

	print ( ("mime-type: error loading database from %s"):format ( options['database'] ) )

else

	for l in h:lines () do

		local mtype, exts = l:match ( '^([%S]+)%s+(%S.*)$' )

		if mtype then

			for ext in exts:gmatch ( '%S+' ) do

				print ( ("Adding type %s for extension %s"):format ( mtype, ext ) )

				db[ext] = mtype

			end

		end

	end

	h:close ()

end

function F.type ( filename )

	local ext = filename:match ( "^.-%.([%w%d]+)$" )

	if ext then

		local mtype = db[ext]

		if mtype then

			return mtype

		end

	end

	return options['default']

end

return F

-- vim: se ts=4 sw=4: --