/*
 * SDカード関連
 *
 * Copyright (c) 2016 Wakayama.rb Ruby Board developers
 *
 * This software is released under the MIT License.
 * https://github.com/wakayamarb/wrbb-v2lib-firm/blob/master/MITL
 *
 */
#include <arduino.h>
/*#include <SD.h>*/
/*#include <RTC.h>*/

#include "mruby.h"
#include "mruby/string.h"

#include "../llbruby.h"
#include "sSdCard.h"

File Fp[2];

/******************************************************/
/*ファイルが存在するかどうか調べる: SD_exists*/
/*	SD_exists( filename )*/
/*  filename: 調べるファイル名*/
/**/
/* 戻り値*/
/*	存在する: 1, 存在しない: 0*/
/******************************************************/
mrb_value mrb_sdcard_exists(mrb_state *mrb, mrb_value self)
{
	mrb_value value;
	char	*str;
	int ret = 0;

	mrb_get_args(mrb, "S", &value);

	str = RSTRING_PTR(value);

	if (SD_exists(str) == true) {
		ret = 1;
	}

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* ディレクトリを作成する: SD_mkdir*/
/*	SD_mkdir( dirname )*/
/*  dirname: 作成するディレクトリ名*/
/**/
/* 戻り値*/
/*	成功: 1, 失敗: 0*/
/******************************************************/
mrb_value mrb_sdcard_mkdir(mrb_state *mrb, mrb_value self)
{
	mrb_value value;
	char	*str;
	int ret = 0;

	mrb_get_args(mrb, "S", &value);

	str = RSTRING_PTR(value);

	if (SD_mkdir(str) == true) {
		ret = 1;
	}

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* ファイルを削除する: SD_remove*/
/*	SD_remove( filename )*/
/*  filename: 削除するファイル名*/
/**/
/* 戻り値*/
/*	成功: 1, 失敗: 0*/
/******************************************************/
mrb_value mrb_sdcard_remove(mrb_state *mrb, mrb_value self)
{
	mrb_value value;
	char	*str;
	int ret = 0;

	mrb_get_args(mrb, "S", &value);

	str = RSTRING_PTR(value);

	if (SD_remove(str) == true) {
		ret = 1;
	}

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* ファイルをコピーする: SD_copy*/
/*	SD_copy( srcfilename, distfilename )*/
/*  srcfilename: コピー元ファイル名*/
/*  distfilename: コピー先ファイル名*/
/**/
/* 戻り値*/
/*	成功: 1, 失敗: 0*/
/******************************************************/
mrb_value mrb_sdcard_copy(mrb_state *mrb, mrb_value self)
{
	mrb_value value1, value2;
	char	*str1;
	char	*str2;
	int ret = 0;
	int v;
	unsigned char chr[1];

	mrb_get_args(mrb, "SS", &value1, &value2);

	str1 = RSTRING_PTR(value1);
	str2 = RSTRING_PTR(value2);

	if (SD_exists(str2) == true) {
		SD_remove(str2);
	}

	/*既存のファイルをコピーする*/
	File *s = &Fp[0];
	File *d = &Fp[1];

	if (SD_open(s, str1, FILE_READ)) {
		if (SD_open(d, str2, FILE_WRITE)) {
			v = file_read(s);
			while (v >= 0) {
				chr[0] = (unsigned char)v;
				file_write(d, chr, 1);
				v = file_read(s);
			}
			ret = 1;
			file_flush(d);
			file_close(d);
		}
		file_close(s);
	}

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* ディレクトリを削除する: SD_rmdir*/
/*	SD_rmdir( dirname )*/
/*  dirname: 削除するディレクトリ名*/
/**/
/* 戻り値*/
/*	成功: 1, 失敗: 0*/
/******************************************************/
mrb_value mrb_sdcard_rmdir(mrb_state *mrb, mrb_value self)
{
	mrb_value value;
	char	*str;
	int ret = 0;

	mrb_get_args(mrb, "S", &value);

	str = RSTRING_PTR(value);

	if (SD_rmdir(str) == true) {
		ret = 1;
	}

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* ファイルをオープンします: SD_open*/
/*	SD_open( number, filename[, mode] )*/
/*	number: ファイル番号 0 または 1*/
/*	filename: ファイル名(8.3形式)*/
/*	mode: 0:Read, 1:Append, 2:New Create*/
/* 戻り値*/
/*	成功: 番号, 失敗: -1*/
/******************************************************/
mrb_value mrb_sdcard_open(mrb_state *mrb, mrb_value self)
{
	int	num;
	int mode;
	mrb_value value;
	char	*str;

	int n = mrb_get_args(mrb, "iS|i", &num, &value, &mode);

	str = RSTRING_PTR(value);

	if (n < 3) {
		mode = 0;
	}

	if (mode == 2) {
		/*新規書き込み*/
		if (SD_exists(str) == true) {
			SD_remove(str);
		}
	}

	int ret = -1;
	if (num < 0 || num > 1) { return mrb_fixnum_value(ret); }

	if (mode == 1 || mode == 2) {
		if (SD_open(&Fp[num], str, FILE_WRITE)) {
			file_seek(&Fp[num], file_size(&Fp[num]));
			ret = num;
		}
	}
	else {
		if (SD_open(&Fp[num], str, FILE_READ)) {
			ret = num;
		}
	}

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* openしたファイルをクローズします: SD_close*/
/*	SD_close( number )*/
/*	number: ファイル番号 0 または 1*/
/******************************************************/
mrb_value mrb_sdcard_close(mrb_state *mrb, mrb_value self)
{
	int	num;

	mrb_get_args(mrb, "i", &num);

	if (num >= 0 && num < 2) {
		file_close(&Fp[num]);
	}

	return mrb_nil_value();			/*戻り値は無しですよ。*/
}

/******************************************************/
/* openしたファイルから1バイト読み込みます: SD_read*/
/*	SD_read( number )*/
/*	number: ファイル番号 0 または 1*/
/* 戻り値*/
/*	0x00～0xFFが返る。ファイルの最後だったら-1が返る。*/
/******************************************************/
mrb_value mrb_sdcard_read(mrb_state *mrb, mrb_value self)
{
	int	num;

	mrb_get_args(mrb, "i", &num);

	int dat = -1;
	if (num < 0 || num > 1) { return mrb_fixnum_value(dat); }

	dat = file_read(&Fp[num]);

	return mrb_fixnum_value(dat);
}

/******************************************************/
/* openしたファイルの読み出し位置を移動する: SD_seek*/
/*	SD_seek( number, byte )*/
/*	number: ファイル番号 0 または 1*/
/*	byte: seekするバイト数(-1)でファイルの最後に移動する*/
/* 戻り値*/
/*	成功: 1, 失敗: 0*/
/******************************************************/
mrb_value mrb_sdcard_seek(mrb_state *mrb, mrb_value self)
{
	int	num;
	int size;
	int ret = 0;

	mrb_get_args(mrb, "ii", &num, &size);

	if (num < 0 || num > 1) { return mrb_fixnum_value(ret); }

	if (size == -1) {
		if (file_seek(&Fp[num], file_size(&Fp[num]))) {
			ret = 1;
		}
	}
	else {
		if (file_seek(&Fp[num], size)) {
			ret = 1;
		}
	}
	return mrb_fixnum_value(ret);
}

/******************************************************/
/* openしたファイルにバイナリデータを書き込む: SD_write*/
/*	SD_write( number, buf, len )*/
/*	number: ファイル番号 0 または 1*/
/*	buf: 書き込むデータ*/
/*	len: 書き込むデータサイズ*/
/* 戻り値*/
/*	実際に書いたバイト数*/
/******************************************************/
mrb_value mrb_sdcard_write(mrb_state *mrb, mrb_value self)
{
	int	num, len;
	mrb_value value;
	char	*str;

	mrb_get_args(mrb, "iSi", &num, &value, &len);

	str = RSTRING_PTR(value);

	int ret = 0;
	if (num < 0 || num > 1) { return mrb_fixnum_value(ret); }

	ret = file_write(&Fp[num], (unsigned char*)str, len);

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* openしたファイルの書き込みをフラッシュします: SD_flush*/
/*	SD_flush( number )*/
/*	number: ファイル番号 0 または 1*/
/******************************************************/
mrb_value mrb_sdcard_flush(mrb_state *mrb, mrb_value self)
{
	int	num;

	mrb_get_args(mrb, "i", &num);

	if (num >= 0 && num < 2) {
		file_flush(&Fp[num]);
	}

	return mrb_nil_value();			/*戻り値は無しですよ。*/
}

/******************************************************/
/* openしたファイルのサイズを取得します: SD_size*/
/*	SD_size( number )*/
/*	number: ファイル番号 0 または 1*/
/**/
/* 戻り値*/
/*	ファイルサイズ*/
/******************************************************/
mrb_value mrb_sdcard_size(mrb_state *mrb, mrb_value self)
{
	int	num;
	mrb_int ret = 0;

	mrb_get_args(mrb, "i", &num);

	if (num < 0 || num > 1) { return mrb_fixnum_value(ret); }

	ret = file_size(&Fp[num]);

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* openしたファイルのseek位置を取得します: SD_position*/
/*	SD_position( number )*/
/*	number: ファイル番号 0 または 1*/
/**/
/* 戻り値*/
/*	シーク位置*/
/******************************************************/
mrb_value mrb_sdcard_position(mrb_state *mrb, mrb_value self)
{
	int	num;
	mrb_int ret = 0;

	mrb_get_args(mrb, "i", &num);

	if (num < 0 || num > 1) { return mrb_fixnum_value(ret); }

	ret = file_position(&Fp[num]);

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* ライブラリを定義します*/
/******************************************************/
int sdcard_Init(mrb_state *mrb)
{
	/*日付と時刻を返す関数を登録*/
	/*SdFile::dateTimeCallback( &SD_DateTime );*/

	struct RClass *sdcardModule = mrb_define_module(mrb, "SD");

	mrb_define_module_function(mrb, sdcardModule, "exists", mrb_sdcard_exists, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "mkdir", mrb_sdcard_mkdir, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "remove", mrb_sdcard_remove, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "rename", mrb_sdcard_copy, MRB_ARGS_REQ(2));
	mrb_define_module_function(mrb, sdcardModule, "rmdir", mrb_sdcard_rmdir, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "open", mrb_sdcard_open, MRB_ARGS_REQ(2) | MRB_ARGS_OPT(1));

	mrb_define_module_function(mrb, sdcardModule, "close", mrb_sdcard_close, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "read", mrb_sdcard_read, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "seek", mrb_sdcard_seek, MRB_ARGS_REQ(2));
	mrb_define_module_function(mrb, sdcardModule, "write", mrb_sdcard_write, MRB_ARGS_REQ(3));
	mrb_define_module_function(mrb, sdcardModule, "flush", mrb_sdcard_flush, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "size", mrb_sdcard_size, MRB_ARGS_REQ(1));
	mrb_define_module_function(mrb, sdcardModule, "position", mrb_sdcard_position, MRB_ARGS_REQ(1));

	return 1;
}

/******************************************************/
/* 日付と時刻を返すコールバック関数*/
/******************************************************/
void SD_DateTime(uint16_t *date, uint16_t *time)
{
	uint16_t year = 2000;
	uint8_t month = 1, day = 1, hour = 0, minute = 0, second = 0;
	time_t timertc;
	struct tm *tm;

	timertc = rtc_read();
	tm = localtime(&timertc);

	year = tm->tm_year + 1900;
	month = tm->tm_mon;
	day = tm->tm_mday;
	hour = tm->tm_hour;
	minute = tm->tm_min;
	second = tm->tm_sec;

	/* FAT_DATEマクロでフィールドを埋めて日付を返す*/
	*date = (year - 1980) << 9 | month << 5 | day;

	/* FAT_TIMEマクロでフィールドを埋めて時間を返す*/
	*time = hour << 11 | minute << 5 | second >> 1;
}