package my_format

import (
    "io"
    "github.com/nopdan/rose/model"
)

type MyFormat struct {
    model.BaseFormat
}

func New() *MyFormat {
    return &MyFormat{
        BaseFormat: model.BaseFormat{
            ID:        "my_format",       // 唯一标识，用于命令行和 API
            Name:      "我的格式",         // 显示名称
            Type:      model.FormatTypePinyin, // FormatTypePinyin / FormatTypeWubi / FormatTypeWords
            Extension: ".txt",            // 文件扩展名（含点号）
        },
    }
}

// 导入：从 Source 读取数据，返回 Entry 列表
func (f *MyFormat) Import(src model.Source) ([]*model.Entry, error) {
    // OpenTextReader 自动识别编码格式转为 utf8 reader
    // textReader, _, closeFn, err := model.OpenTextReader(src)
    // 获取 Reader
	r, err := model.NewReaderFromSource(src)
	if err != nil {
		return nil, err
	}
    // 解析 data，构建 Entry 列表
    // 每个 Entry 包含 Word（词组）、Code（编码）、Frequency（词频）等字段
    var entries []*model.Entry
    // ...解析逻辑
    return entries, nil
}

// 导出：将 Entry 列表写入 Writer
func (f *MyFormat) Export(entries []*model.Entry, w io.Writer) error {
    // 将 entries 按目标格式写入 w
    return nil
}

import "github.com/nopdan/rose/format/my_format"

// 在对应分组中添加
RegisterFormat(my_format.New())