pagespeed/automatic/static_rewriter_main.cc

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */


#include <cstdio>

#include "net/instaweb/rewriter/public/process_context.h"
#include "net/instaweb/rewriter/public/rewrite_driver_factory.h"
#include "pagespeed/automatic/static_rewriter.h"
#include "pagespeed/kernel/base/file_system.h"
#include "pagespeed/kernel/base/string.h"
#include "pagespeed/kernel/base/string_util.h"
#include "pagespeed/kernel/base/string_writer.h"
#include "pagespeed/system/system_rewrite_options.h"

namespace net_instaweb {

class MessageHandler;

}  // namespace net_instaweb

// The purpose of this program is to help us test that pagespeed_automatic.a
// contains all that's needed to successfully link a rewriter using standard
// g++, without using the gyp flow.
int main(int argc, char** argv) {
  net_instaweb::ProcessContext process_context;
  net_instaweb::SystemRewriteOptions::Initialize();
  net_instaweb::RewriteDriverFactory::Initialize();
  net_instaweb::StaticRewriter static_rewriter(process_context, &argc, &argv);

  // Having stripped all the flags, there should be exactly 3
  // arguments remaining:
  //
  //   input_directory:   The directory where the origin web site is stored
  //   output_directory:  The directory where the rewritten web site is written
  //   URL:               The URL of HTML to rewrite.
  if (argc != 4) {
    fprintf(stderr, "Usage: [options] %s input_dir output_dir url.\n", argv[0]);
    fprintf(stderr, "Type '%s --help' to see the options\n", argv[0]);
    return 1;
  }
  const char* input_dir = argv[1];
  const char* output_dir = argv[2];
  const char* html_name = argv[3];

  GoogleString url = net_instaweb::StrCat("http://test.com/", html_name);
  GoogleString input_file_path = net_instaweb::StrCat(input_dir, "/",
                                                      html_name);
  GoogleString output_file_path = net_instaweb::StrCat(output_dir, "/",
                                                       html_name);
  GoogleString html_input_buffer, html_output_buffer;
  net_instaweb::FileSystem* file_system = static_rewriter.file_system();
  net_instaweb::MessageHandler* message_handler =
      static_rewriter.message_handler();
  net_instaweb::StringWriter writer(&html_output_buffer);

  int exit_status = 1;
  if (!file_system->ReadFile(input_file_path.c_str(), &html_input_buffer,
                             message_handler)) {
    fprintf(stderr, "failed to read file %s\n", input_file_path.c_str());
  } else if (!static_rewriter.ParseText(url, input_file_path, html_input_buffer,
                                        output_dir, &writer)) {
    fprintf(stderr, "StartParseId failed on url %s\n", url.c_str());
  } else if (!file_system->WriteFileAtomic(
      output_file_path, html_output_buffer, message_handler)) {
    fprintf(stderr, "failed to write file %s\n", output_file_path.c_str());
  } else {
    exit_status = 0;
  }

  // TODO(jmarantz): set up a file-based fetcher that will allow us to
  // rewrite resources in HTML files in this demonstration.

  net_instaweb::RewriteDriverFactory::Terminate();
  net_instaweb::SystemRewriteOptions::Terminate();
  return 0;
}