herolib/lib/data/atlas/atlas_save_test.v

module atlas

import incubaid.herolib.core.pathlib
import os

const test_dir = '/tmp/atlas_save_test'

fn testsuite_begin() {
	os.rmdir_all(test_dir) or {}
	os.mkdir_all(test_dir)!
}

fn testsuite_end() {
	os.rmdir_all(test_dir) or {}
}

fn test_save_and_load_basic() {
	// Create a collection with some content
	col_path := '${test_dir}/docs'
	os.mkdir_all(col_path)!

	mut cfile := pathlib.get_file(path: '${col_path}/.collection', create: true)!
	cfile.write('name:docs')!

	mut page1 := pathlib.get_file(path: '${col_path}/intro.md', create: true)!
	page1.write('# Introduction\n\nWelcome to the docs!')!

	mut page2 := pathlib.get_file(path: '${col_path}/guide.md', create: true)!
	page2.write('# Guide\n\nMore content here.')!

	// Create and scan atlas
	mut a := new(name: 'my_docs')!
	a.scan(path: test_dir)!

	assert a.collections.len == 1

	// Save all collections
	a.save()!
	assert os.exists('${col_path}/.collection.json')

	// Load in a new atlas
	mut a2 := new(name: 'loaded_docs')!
	a2.load_from_directory(test_dir)!

	assert a2.collections.len == 1

	// Access loaded data
	loaded_col := a2.get_collection('docs')!
	assert loaded_col.name == 'docs'
	assert loaded_col.pages.len == 2

	// Verify pages exist
	assert loaded_col.page_exists('intro')
	assert loaded_col.page_exists('guide')

	// Read page content
	mut intro_page := loaded_col.page_get('intro')!
	content := intro_page.read_content()!
	assert content.contains('# Introduction')
	assert content.contains('Welcome to the docs!')
}

fn test_save_and_load_with_includes() {
	col_path := '${test_dir}/docs_include'
	os.mkdir_all(col_path)!

	mut cfile := pathlib.get_file(path: '${col_path}/.collection', create: true)!
	cfile.write('name:docs')!

	mut page1 := pathlib.get_file(path: '${col_path}/intro.md', create: true)!
	page1.write('# Introduction\n\nWelcome to the docs!')!

	mut page2 := pathlib.get_file(path: '${col_path}/guide.md', create: true)!
	page2.write('# Guide\n\n!!include docs:intro\n\nMore content here.')!

	// Create and scan atlas
	mut a := new(name: 'my_docs')!
	a.scan(path: '${test_dir}/docs_include')!

	// Validate links (should find the include)
	a.validate_links()!

	col := a.get_collection('docs')!
	assert !col.has_errors()

	// Save
	a.save()!

	// Load
	mut a2 := new(name: 'loaded')!
	a2.load_from_directory('${test_dir}/docs_include')!

	loaded_col := a2.get_collection('docs')!
	assert loaded_col.pages.len == 2
	assert !loaded_col.has_errors()
}

fn test_save_and_load_with_errors() {
	col_path := '${test_dir}/docs_errors'
	os.mkdir_all(col_path)!

	mut cfile := pathlib.get_file(path: '${col_path}/.collection', create: true)!
	cfile.write('name:docs')!

	// Create page with broken link
	mut page1 := pathlib.get_file(path: '${col_path}/broken.md', create: true)!
	page1.write('[Broken link](nonexistent)')!

	// Create and scan atlas
	mut a := new(name: 'my_docs')!
	a.scan(path: '${test_dir}/docs_errors')!

	// Validate - will generate errors
	a.validate_links()!

	col := a.get_collection('docs')!
	assert col.has_errors()
	initial_error_count := col.errors.len

	// Save with errors
	a.save()!

	// Load
	mut a2 := new(name: 'loaded')!
	a2.load_from_directory('${test_dir}/docs_errors')!

	loaded_col := a2.get_collection('docs')!
	assert loaded_col.has_errors()
	assert loaded_col.errors.len == initial_error_count
	assert loaded_col.error_cache.len == initial_error_count
}

fn test_save_and_load_multiple_collections() {
	// Create multiple collections
	col1_path := '${test_dir}/multi/col1'
	col2_path := '${test_dir}/multi/col2'

	os.mkdir_all(col1_path)!
	os.mkdir_all(col2_path)!

	mut cfile1 := pathlib.get_file(path: '${col1_path}/.collection', create: true)!
	cfile1.write('name:col1')!

	mut cfile2 := pathlib.get_file(path: '${col2_path}/.collection', create: true)!
	cfile2.write('name:col2')!

	mut page1 := pathlib.get_file(path: '${col1_path}/page1.md', create: true)!
	page1.write('# Page 1')!

	mut page2 := pathlib.get_file(path: '${col2_path}/page2.md', create: true)!
	page2.write('# Page 2')!

	// Create and save
	mut a := new(name: 'multi')!
	a.scan(path: '${test_dir}/multi')!

	assert a.collections.len == 2

	a.save()!

	// Load from directory
	mut a2 := new(name: 'loaded')!
	a2.load_from_directory('${test_dir}/multi')!

	assert a2.collections.len == 2
	assert a2.get_collection('col1')!.page_exists('page1')
	assert a2.get_collection('col2')!.page_exists('page2')
}

fn test_save_and_load_with_images() {
	col_path := '${test_dir}/docs_images'
	os.mkdir_all(col_path)!
	os.mkdir_all('${col_path}/img')!

	mut cfile := pathlib.get_file(path: '${col_path}/.collection', create: true)!
	cfile.write('name:docs')!

	mut page := pathlib.get_file(path: '${col_path}/page.md', create: true)!
	page.write('# Page with image')!

	// Create a dummy image file
	mut img := pathlib.get_file(path: '${col_path}/img/test.png', create: true)!
	img.write('fake png data')!

	// Create and scan
	mut a := new(name: 'my_docs')!
	a.scan(path: '${test_dir}/docs_images')!

	col := a.get_collection('docs')!
	assert col.images.len == 1
	assert col.image_exists('test')

	// Save
	a.save()!

	// Load
	mut a2 := new(name: 'loaded')!
	a2.load_from_directory('${test_dir}/docs_images')!

	loaded_col := a2.get_collection('docs')!
	assert loaded_col.images.len == 1
	assert loaded_col.image_exists('test')

	img_file := loaded_col.image_get('test')!
	assert img_file.file_name() == 'test.png'
	assert img_file.is_image()
}