#!/bin/ruby
#----------------------------------------------------------------------------
# Project:  TwlSDK
# File:     sumaker.rb
#
# Copyright 2007 Nintendo.  All rights reserved.
#
# These coded instructions, statements, and computer programs contain
# proprietary information of Nintendo of America Inc. and/or Nintendo
# Company Ltd., and are protected by Federal copyright law.  They may
# not be disclosed to third parties or copied or duplicated in any form,
# in whole or in part, without the prior written consent of Nintendo.
#
# $Date::            $
# $Rev$
# $Author$
#----------------------------------------------------------------------------

require "yaml"
require "fileutils"
require "tempfile"

DEFAULT_CONFIG = "default.yaml"
CUSTOM_CONFIG = "custom.yaml"

FILE_MATCH = /[a-zA-Z0-9\-_\.]+\z/

REGION = {
    "JPN" => "OS_TWL_REGION_JAPAN",
    "EUR" => "OS_TWL_REGION_EUROPE",
    "AUS" => "OS_TWL_REGION_AUSTRALIA",
    "USA" => "OS_TWL_REGION_AMERICA",
}

REGION_PATH = {
    "JPN" => "jp",
    "EUR" => "euro",
    "AUS" => "aus",
    "USA" => "usa",
}

# デフォルト設定の作成
def make_default_config
    config = {
        :TadFiles => [
        "HNAA.tad",
        "HNBA.tad",
        "HNCA.tad",
        "HNLA.tad",
        "HNHA.tad",
        ],
        :FontFile => "TWLFontTable.dat",
        :NandFirm => "menu_launcher.nand",
        :TargetPath => ".",
        :TemplatePath => "template",
        :RSF => "ARM9.TWL/main.rsf",
        :SRC => "ARM9.TWL/src/main.c",
        :DataPath => ".",
        :Region => "JPN",
    }
    write_config(DEFAULT_CONFIG, config)
    
end

# ファイル名にスクリプトを置いてあるディレクトリをくっつける
def calc_path(filename)
#    File.join(File.dirname(File.expand_path(__FILE__)), filename)
filename
end

# コンフィグファイルを読んで展開
def read_config(filename)
    YAML.load(File.read(calc_path(filename)))
end

# コンフィグのライト
def write_config(filename, data)
    File.open(calc_path(filename), "w") {|file|
        file.write data.to_yaml
    }
end

# データ内の文字列を置き換える
def replace_data(src, mark, data)
    while src.index(/%#{mark}%/) != nil do
        src = src.sub(/%#{mark}%/, data)
    end
    src
end


# ファイルを書き込む
def write_data(filename, data)
    temp = Tempfile.new("temp")
    temp.puts data
    temp.close
    FileUtils.cp(temp.path, calc_path(filename))
end

# main.rsfの書き換え
def make_rsf(config)
    tadlist = config[:TadFiles].dup
    tadlist << config[:FontFile] << config[:NandFirm]
    tadlist = tadlist.join(" ")
    src = File.read(File.join(config[:TemplatePath], config[:RSF]))
    #src = File.read(config[:TemplatePath] + "/" + config[:RSF])
    src = replace_data(src, "datapath", config[:DataPath])
    src = replace_data(src, "regionname", config[:Region])
    write_data(File.join(config[:TargetPath], config[:RSF]), replace_data(src, "filelist", tadlist))
    #write_data(config[:TargetPath] + "/" + config[:RSF], replace_data(src, "filelist", tadlist))
end

# main.c の書き換え
def make_main(config)
    filelist = config[:TadFiles]
    data = []
    for t in filelist do 
        data.push("\t\"rom:/data/" + t + "\", \n")
    end
    data = data.join
    #src = File.read(config[:TemplatePath] + "/" + config[:SRC])
    src = File.read(File.join(config[:TemplatePath], config[:SRC]))
    mod = replace_data(src, "tadlist", data)
    mod = replace_data(mod, "nandfirm", config[:NandFirm])
    mod = replace_data(mod, "fontfile", config[:FontFile])
    
    # region
	region_name = (config[:Region] != nil) ? config[:Region] : "JPN" 
    region = (REGION[region_name] != nil) ? REGION[region_name] : REGION["JPN"] 
    mod = replace_data(mod, "region", region)
    mod = replace_data(mod, "regionname", region_name)
    #write_data(config[:TargetPath] + "/" + config[:SRC], mod)    
    write_data(File.join(config[:TargetPath], config[:SRC]), mod)    

end

# SystemUpdaterを作成する
def make_updater(filename)
    config = read_config(filename)  
    make_rsf(config)
    make_main(config)
end

# ディレクトリ内のtadを検索
def pickup_files(target_dir, name)
    Dir.glob(target_dir + name)
end


# こっからメイン

#sumaker mode [filename]

if ARGV.size == 0
    p "Usage: sumaker mode [filename]"
    exit
end

mode = ARGV[0]

case mode 
when "default" then
    # デフォルト設定で作成
    make_default_config
    make_updater(DEFAULT_CONFIG)
when "custom" then
    if ARGV.size < 3 
        p "Usage: sumaker custom target_dir region"
        exit -1
    end
    # カスタム設定で作成
    config = "custom_base.yaml"
    region = ARGV[2]
    
    # カレントディレクトリ変更
    Dir.chdir(File.dirname(File.expand_path(__FILE__)))


    # ベースコンフィグに、dataディレクトリ内のtadとnandを追加
    config = read_config(config)
    if config[:TadFiles] == nil
        config[:TadFiles] = []
    end
    
    
    #config[:DataPath] += "/#{ARGV[1]}/#{REGION_PATH[region]}"
    config[:DataPath] = File.join(config[:DataPath], ARGV[1], REGION_PATH[region])
    config[:Region] = region
    tads = pickup_files(config[:DataPath], "/*.tad")
    nand = pickup_files(config[:DataPath], "/*.nand")
    font = pickup_files(config[:DataPath], "/*.dat")
# フォントファイル名変更
	system("mv #{font} #{File.join(config[:DataPath], "TWLFontTable")}.dat")
    font = pickup_files(config[:DataPath], "/*.dat")

    if nand.size > 0 
        config[:NandFirm] = nand[0].slice(FILE_MATCH)
    end
    if font.size > 0 
        config[:FontFile] = font[0].slice(FILE_MATCH)
    end
    for tad in tads do
        tad = tad.slice(FILE_MATCH)
        config[:TadFiles] << tad
    end
    config[:TadFiles] = config[:TadFiles].uniq
    write_config(CUSTOM_CONFIG, config)
    make_updater(CUSTOM_CONFIG)
    
    exit 0
end