Generating resource IDs

gistfile1.rb

header_subpath = "Resources.h"

resource_categories = {
    ".pdf" => "File",
    ".png" => "Image",
    ".jpg" => "Image"
}

def wikify(phrase)
    phrase = phrase.tr( "-", " " ).tr( "_", " " ).tr( "/", " " )
    phrase.gsub!(/^[a-z]|\s+[a-z]/) { |a| a.upcase }
    phrase.gsub!(/\s/, '')
    return phrase
end

def app_resources(categories)
    resources = []
    extensions = categories.keys
        
    Dir.glob( "**/*" ) do |filename|
        if extensions.include?( File.extname( filename ) )
            resources << filename
        end
    end

    resources
end

def write_header(path, resources, categories)        
    File.open( path, 'w' ) do |header|
        header.puts( "static NSString* __AppResources[] = {" )

        resources.each do |resource|
            header.puts( "\t@\"#{resource}\"," )
        end

        header.puts( "\tnil" )
        header.puts( "};" )
        header.puts

        index = 0
        resources.each do |resource|
            extension = File.extname(resource)
            identifier = wikify( resource.chomp( extension ) )
            header.puts( "static const NSUInteger k" + categories[ extension ] + identifier + " = #{index};" )
            index += 1
        end

        header.puts

        header.puts( "static __inline__ NSString* GetResourcePath( const NSUInteger identifier ) {" )
        header.puts( "\treturn ( identifier < #{resources.length} ) ? __AppResources[ identifier ] : nil;" )
        header.puts( "}" )

        header.puts

        header.puts( "static __inline__ UIImage* GetImageWithID( const NSUInteger identifier ) {" )
        header.puts( "\treturn [UIImage imageNamed:GetResourcePath( identifier )];" )
        header.puts( "}" )
    end
end

header_path = File.join( ENV['PROJECT_DIR'], header_subpath )
write_header( header_path, app_resources( resource_categories ), resource_categories )