123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103 |
- using System.Collections.Generic;
- using System.IO;
- using System.Text;
- using System.Threading;
- namespace BepInEx.Logging
- {
- /// <summary>
- /// Logs entries using Unity specific outputs.
- /// </summary>
- public class DiskLogListener : ILogListener
- {
- /// <summary>
- /// Log levels to display.
- /// </summary>
- public LogLevel DisplayedLogLevel { get; set; }
- /// <summary>
- /// Writer for the disk log.
- /// </summary>
- public TextWriter LogWriter { get; protected set; }
- /// <summary>
- /// Timer for flushing the logs to a file.
- /// </summary>
- private Timer FlushTimer { get; set; }
- private bool InstantFlushing { get; set; }
- /// <summary>
- /// Creates a new disk log listener.
- /// </summary>
- /// <param name="localPath">Path to the log.</param>
- /// <param name="displayedLogLevel">Log levels to display.</param>
- /// <param name="appendLog">Whether to append logs to an already existing log file.</param>
- /// <param name="delayedFlushing">Whether to delay flushing to disk to improve performance. Useful to set this to false when debugging crashes.</param>
- /// <param name="fileLimit">Maximum amount of concurrently opened log files. Can help with infinite game boot loops.</param>
- public DiskLogListener(string localPath, LogLevel displayedLogLevel = LogLevel.Info, bool appendLog = false, bool delayedFlushing = true, int fileLimit = 5)
- {
- DisplayedLogLevel = displayedLogLevel;
- int counter = 1;
- FileStream fileStream;
- while (!Utility.TryOpenFileStream(Path.Combine(Paths.BepInExRootPath, localPath), appendLog ? FileMode.Append : FileMode.Create, out fileStream, share: FileShare.Read, access: FileAccess.Write))
- {
- if (counter == fileLimit)
- {
- Logger.LogError("Couldn't open a log file for writing. Skipping log file creation");
- return;
- }
- Logger.LogWarning($"Couldn't open log file '{localPath}' for writing, trying another...");
- localPath = $"LogOutput.{counter++}.log";
- }
- LogWriter = TextWriter.Synchronized(new StreamWriter(fileStream, Encoding.UTF8));
- if (delayedFlushing)
- {
- FlushTimer = new Timer(o => { LogWriter?.Flush(); }, null, 2000, 2000);
- }
- InstantFlushing = !delayedFlushing;
- }
- public static HashSet<string> BlacklistedSources = new HashSet<string>();
- /// <inheritdoc />
- public void LogEvent(object sender, LogEventArgs eventArgs)
- {
- if (LogWriter == null)
- return;
-
- if (BlacklistedSources.Contains(eventArgs.Source.SourceName))
- return;
- if ((eventArgs.Level & DisplayedLogLevel) == 0)
- return;
- LogWriter.WriteLine(eventArgs.ToString());
- if (InstantFlushing)
- LogWriter.Flush();
- }
- /// <inheritdoc />
- public void Dispose()
- {
- FlushTimer?.Dispose();
- LogWriter?.Flush();
- LogWriter?.Dispose();
- }
- ~DiskLogListener()
- {
- Dispose();
- }
- }
- }
|