Files
2026-02-21 16:45:37 +08:00

443 lines
7.7 KiB
C#

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Text;
namespace LitJson
{
public class JsonWriter
{
private static NumberFormatInfo number_format;
private WriterContext context;
private Stack<WriterContext> ctx_stack;
private bool has_reached_end;
private char[] hex_seq;
private int indentation;
private int indent_value;
private StringBuilder inst_string_builder;
private bool pretty_print;
private bool validate;
private TextWriter writer;
public int IndentValue
{
get
{
return indent_value;
}
set
{
indentation = indentation / indent_value * value;
indent_value = value;
}
}
public bool PrettyPrint
{
get
{
return pretty_print;
}
set
{
pretty_print = value;
}
}
public TextWriter TextWriter
{
get
{
return writer;
}
}
public bool Validate
{
get
{
return validate;
}
set
{
validate = value;
}
}
static JsonWriter()
{
number_format = NumberFormatInfo.InvariantInfo;
}
public JsonWriter()
{
inst_string_builder = new StringBuilder();
writer = new StringWriter(inst_string_builder);
Init();
}
public JsonWriter(StringBuilder sb)
: this(new StringWriter(sb))
{
}
public JsonWriter(TextWriter writer)
{
if (writer == null)
{
throw new ArgumentNullException("writer");
}
this.writer = writer;
Init();
}
private void DoValidation(Condition cond)
{
if (!context.ExpectingValue)
{
context.Count++;
}
if (!validate)
{
return;
}
if (has_reached_end)
{
throw new JsonException("A complete JSON symbol has already been written");
}
switch (cond)
{
case Condition.InArray:
if (!context.InArray)
{
throw new JsonException("Can't close an array here");
}
break;
case Condition.InObject:
if (!context.InObject || context.ExpectingValue)
{
throw new JsonException("Can't close an object here");
}
break;
case Condition.NotAProperty:
if (context.InObject && !context.ExpectingValue)
{
throw new JsonException("Expected a property");
}
break;
case Condition.Property:
if (!context.InObject || context.ExpectingValue)
{
throw new JsonException("Can't add a property here");
}
break;
case Condition.Value:
if (!context.InArray && (!context.InObject || !context.ExpectingValue))
{
throw new JsonException("Can't add a value here");
}
break;
}
}
private void Init()
{
has_reached_end = false;
hex_seq = new char[4];
indentation = 0;
indent_value = 4;
pretty_print = false;
validate = true;
ctx_stack = new Stack<WriterContext>();
context = new WriterContext();
ctx_stack.Push(context);
}
private static void IntToHex(int n, char[] hex)
{
for (int i = 0; i < 4; i++)
{
int num = n % 16;
if (num < 10)
{
hex[3 - i] = (char)(48 + num);
}
else
{
hex[3 - i] = (char)(65 + (num - 10));
}
n >>= 4;
}
}
private void Indent()
{
if (pretty_print)
{
indentation += indent_value;
}
}
private void Put(string str)
{
if (pretty_print && !context.ExpectingValue)
{
for (int i = 0; i < indentation; i++)
{
writer.Write(' ');
}
}
writer.Write(str);
}
private void PutNewline()
{
PutNewline(true);
}
private void PutNewline(bool add_comma)
{
if (add_comma && !context.ExpectingValue && context.Count > 1)
{
writer.Write(',');
}
if (pretty_print && !context.ExpectingValue)
{
writer.Write('\n');
}
}
private void PutString(string str)
{
Put(string.Empty);
writer.Write('"');
int length = str.Length;
for (int i = 0; i < length; i++)
{
switch (str[i])
{
case '\n':
writer.Write("\\n");
continue;
case '\r':
writer.Write("\\r");
continue;
case '\t':
writer.Write("\\t");
continue;
case '"':
case '\\':
writer.Write('\\');
writer.Write(str[i]);
continue;
case '\f':
writer.Write("\\f");
continue;
case '\b':
writer.Write("\\b");
continue;
}
if (str[i] >= ' ' && str[i] <= '~')
{
writer.Write(str[i]);
continue;
}
IntToHex(str[i], hex_seq);
writer.Write("\\u");
writer.Write(hex_seq);
}
writer.Write('"');
}
private void Unindent()
{
if (pretty_print)
{
indentation -= indent_value;
}
}
public override string ToString()
{
if (inst_string_builder == null)
{
return string.Empty;
}
return inst_string_builder.ToString();
}
public void Reset()
{
has_reached_end = false;
ctx_stack.Clear();
context = new WriterContext();
ctx_stack.Push(context);
if (inst_string_builder != null)
{
inst_string_builder.Remove(0, inst_string_builder.Length);
}
}
public void Write(bool boolean)
{
DoValidation(Condition.Value);
PutNewline();
Put((!boolean) ? "false" : "true");
context.ExpectingValue = false;
}
public void Write(decimal number)
{
DoValidation(Condition.Value);
PutNewline();
Put(Convert.ToString(number, number_format));
context.ExpectingValue = false;
}
public void Write(double number)
{
DoValidation(Condition.Value);
PutNewline();
string text = Convert.ToString(number, number_format);
Put(text);
if (text.IndexOf('.') == -1 && text.IndexOf('E') == -1)
{
writer.Write(".0");
}
context.ExpectingValue = false;
}
public void Write(int number)
{
DoValidation(Condition.Value);
PutNewline();
Put(Convert.ToString(number, number_format));
context.ExpectingValue = false;
}
public void Write(long number)
{
DoValidation(Condition.Value);
PutNewline();
Put(Convert.ToString(number, number_format));
context.ExpectingValue = false;
}
public void Write(string str)
{
DoValidation(Condition.Value);
PutNewline();
if (str == null)
{
Put("null");
}
else
{
PutString(str);
}
context.ExpectingValue = false;
}
public void Write(ulong number)
{
DoValidation(Condition.Value);
PutNewline();
Put(Convert.ToString(number, number_format));
context.ExpectingValue = false;
}
public void WriteArrayEnd()
{
DoValidation(Condition.InArray);
PutNewline(false);
ctx_stack.Pop();
if (ctx_stack.Count == 1)
{
has_reached_end = true;
}
else
{
context = ctx_stack.Peek();
context.ExpectingValue = false;
}
Unindent();
Put("]");
}
public void WriteArrayStart()
{
DoValidation(Condition.NotAProperty);
PutNewline();
Put("[");
context = new WriterContext();
context.InArray = true;
ctx_stack.Push(context);
Indent();
}
public void WriteObjectEnd()
{
DoValidation(Condition.InObject);
PutNewline(false);
ctx_stack.Pop();
if (ctx_stack.Count == 1)
{
has_reached_end = true;
}
else
{
context = ctx_stack.Peek();
context.ExpectingValue = false;
}
Unindent();
Put("}");
}
public void WriteObjectStart()
{
DoValidation(Condition.NotAProperty);
PutNewline();
Put("{");
context = new WriterContext();
context.InObject = true;
ctx_stack.Push(context);
Indent();
}
public void WritePropertyName(string property_name)
{
DoValidation(Condition.Property);
PutNewline();
PutString(property_name);
if (pretty_print)
{
if (property_name.Length > context.Padding)
{
context.Padding = property_name.Length;
}
for (int num = context.Padding - property_name.Length; num >= 0; num--)
{
writer.Write(' ');
}
writer.Write(": ");
}
else
{
writer.Write(':');
}
context.ExpectingValue = true;
}
}
}